TorchQL: A Programming Framework for Integrity Constraints in Machine Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Naik, Aaditya, Stein, Adam, Wu, Yinjun, Naik, Mayur, Wong, Eric
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916441108774912
author Naik, Aaditya
Stein, Adam
Wu, Yinjun
Naik, Mayur
Wong, Eric
author_facet Naik, Aaditya
Stein, Adam
Wu, Yinjun
Naik, Mayur
Wong, Eric
contents Finding errors in machine learning applications requires a thorough exploration of their behavior over data. Existing approaches used by practitioners are often ad-hoc and lack the abstractions needed to scale this process. We present TorchQL, a programming framework to evaluate and improve the correctness of machine learning applications. TorchQL allows users to write queries to specify and check integrity constraints over machine learning models and datasets. It seamlessly integrates relational algebra with functional programming to allow for highly expressive queries using only eight intuitive operators. We evaluate TorchQL on diverse use-cases including finding critical temporal inconsistencies in objects detected across video frames in autonomous driving, finding data imputation errors in time-series medical records, finding data labeling errors in real-world images, and evaluating biases and constraining outputs of language models. Our experiments show that TorchQL enables up to 13x faster query executions than baselines like Pandas and MongoDB, and up to 40% shorter queries than native Python. We also conduct a user study and find that TorchQL is natural enough for developers familiar with Python to specify complex integrity constraints.
format Preprint
id arxiv_https___arxiv_org_abs_2308_06686
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle TorchQL: A Programming Framework for Integrity Constraints in Machine Learning
Naik, Aaditya
Stein, Adam
Wu, Yinjun
Naik, Mayur
Wong, Eric
Databases
Machine Learning
Software Engineering
Finding errors in machine learning applications requires a thorough exploration of their behavior over data. Existing approaches used by practitioners are often ad-hoc and lack the abstractions needed to scale this process. We present TorchQL, a programming framework to evaluate and improve the correctness of machine learning applications. TorchQL allows users to write queries to specify and check integrity constraints over machine learning models and datasets. It seamlessly integrates relational algebra with functional programming to allow for highly expressive queries using only eight intuitive operators. We evaluate TorchQL on diverse use-cases including finding critical temporal inconsistencies in objects detected across video frames in autonomous driving, finding data imputation errors in time-series medical records, finding data labeling errors in real-world images, and evaluating biases and constraining outputs of language models. Our experiments show that TorchQL enables up to 13x faster query executions than baselines like Pandas and MongoDB, and up to 40% shorter queries than native Python. We also conduct a user study and find that TorchQL is natural enough for developers familiar with Python to specify complex integrity constraints.
title TorchQL: A Programming Framework for Integrity Constraints in Machine Learning
topic Databases
Machine Learning
Software Engineering
url https://arxiv.org/abs/2308.06686