Accelerating Machine Learning Inference With Probabilistic Predicates

Yao Lu,Aakanksha Chowdhery,Srikanth Kandula,Surajit Chaudhuri
DOI: https://doi.org/10.1145/3183713.3183751
2018-01-01
Abstract:Classic query optimization techniques, including predicate push-down, are of limited use for machine learning inference queries, because the user-defined functions (UDFs) which extract relational columns from unstructured inputs are often very expensive; query predicates will remain stuck behind these UDFs if they happen to require relational columns that are generated by the UDFs. In this work, we demonstrate constructing and applying probabilistic predicates to filter data blobs that do not satisfy the query predicate; such filtering is parametrized to different target accuracies. Furthermore, to support complex predicates and to avoid per-query training, we augment a cost-based query optimizer to choose plans with appropriate combinations of simpler probabilistic predicates. Experiments with several machine learning workloads on a big-data cluster show that query processing improves by as much as 10x.
What problem does this paper attempt to address?