Do Machine Learning Models Learn Statistical Rules Inferred from Data?
Aaditya Naik, Yinjun Wu, Mayur Naik, Eric Wong
Abstract
Machine learning models can make critical errors that are easily hidden within vast amounts of data. Such errors often run counter to rules based on human intuition. However, rules based on human knowledge are challenging to scale or to even formalize. We thereby seek to infer statistical rules from the data and quantify the extent to which a model has learned them. We propose a framework SQRL that integrates logic-based methods with statistical inference to derive these rules from a model's training data without supervision. We further show how to adapt models at test time to reduce rule violations and produce more coherent predictions. SQRL generates up to 300K rules over datasets from vision, tabular, and language settings. We uncover up to 158K violations of those rules by state-of-the-art models for classification, object detection, and data imputation. Test-time adaptation reduces these violations by up to 68.7% with relative performance improvement up to 32%. SQRL is available at https://github.com/DebugML/sqrl .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 759998a2-98b4-44f1-9cfa-85f9e9da7778Cited by top-tier papers2
- DISCRET: Synthesizing Faithful Explanations For Treatment Effect EstimationYinjun Wu, Mayank Keoliya, Kan Chen, Neelay Velingker et al.ICML 2024 · 3 citations
- TorchQL: A Programming Framework for Integrity Constraints in Machine LearningAaditya Naik, Adam Stein, Yinjun Wu, Mayur Naik et al.OOPSLA 2024
Builds on10
- Revisiting Deep Learning Models for Tabular DataYury Gorishniy, Ivan Rubachev, Valentin Khrulkov, Artem BabenkoNeurIPS 2021 · 1,847 citations
- Tent: Fully Test-Time Adaptation by Entropy MinimizationDequan Wang, Evan Shelhamer, Shaoteng Liu, Bruno A. Olshausen et al.ICLR 2021 · 1,731 citations
- Contrastive Test-Time AdaptationDian Chen, Dequan Wang, Trevor Darrell, Sayna EbrahimiCVPR 2022 · 219 citations
- Test Time Adaptation via Conjugate Pseudo-labelsSachin Goyal, Mingjie Sun, Aditi Raghunathan, J. Zico KolterNeurIPS 2022 · 152 citations
- Divide and Contrast: Source-free Domain Adaptation via Adaptive Contrastive LearningZiyi Zhang, Weikai Chen, Hui Cheng, Zhen Li et al.NeurIPS 2022 · 112 citations
Related papers
- TabLog: Test-Time Adaptation for Tabular Data Using Logic RulesWeijieying Ren, Xiaoting Li, Huiyuan Chen, Vineeth Rakesh et al.ICML 2024 · 6 citations
- A Unified Framework for Rule Learning: Integrating Commonsense Knowledge from LLMs with Structured Knowledge from Knowledge GraphsQirui Hao, Kewei Cheng, Tongze Zhang, Hongyuan Liu et al.WWW 2026
- Beyond Static Pipelines: Learning Dynamic Workflows for Text-to-SQLYihan Wang, Peiyu Liu, Runyu Chen, Wei XuICML 2026 · 1 citation
- Self-explaining deep models with logic rule reasoningSeungeon Lee, Xiting Wang, Sungwon Han, Xiaoyuan Yi et al.NeurIPS 2022 · 27 citations
- Can LLMs Reason with Rules? Logic Scaffolding for Stress-Testing and Improving LLMsSiyuan Wang, Zhongyu Wei, Yejin Choi, Xiang RenACL 2024
