Scalable Rule Lists Learning with Sampling
Leonardo Pellegrina, Fabio Vandin
Abstract
Learning interpretable models has become a major focus of machine learning research, given the increasing prominence of machine learning in socially important decision-making. Among interpretable models, rule lists are among the best-known and easily interpretable ones. However, finding optimal rule lists is computationally challenging, and current approaches are impractical for large datasets.
We present a novel and scalable approach to learn nearly optimal rule lists from large datasets. Our algorithm uses sampling to efficiently obtain an approximation of the optimal rule list with rigorous guarantees on the quality of the approximation. In particular, our algorithm guarantees to find a rule list with accuracy very close to the optimal rule list when a rule list with high accuracy exists. Our algorithm builds on the VC-dimension of rule lists, for which we prove novel upper and lower bounds. Our experimental evaluation on large datasets shows that our algorithm identifies nearly optimal rule lists with a speed-up up to two orders of magnitude over state-of-the-art exact approaches. Moreover, our algorithm is as fast as, and sometimes faster than, recent heuristic approaches, while reporting higher quality rule lists. In addition, the rules reported by our algorithm are more similar to the rules in the optimal rule list than the rules from heuristic approaches.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on3
- Exploring the Whole Rashomon Set of Sparse Decision TreesRui Xin, Chudi Zhong, Zhi Chen, Takuya Takagi et al.NeurIPS 2022 · 117 citations
- Efficient Centrality Maximization with Rademacher AveragesLeonardo PellegrinaKDD 2023 · 9 citations
- MCRapper: Monte-Carlo Rademacher Averages for Poset Families and Approximate Pattern MiningLeonardo Pellegrina, Cyrus Cousins, Fabio Vandin, Matteo RiondatoKDD 2020 · 7 citations
Related papers
- Diverse Rule SetsGuangyi Zhang, Aristides GionisKDD 2020 · 18 citations
- Efficient Decision Rule List Learning via Unified Sequence Submodular OptimizationLinxiao Yang, Jingbang Yang, Liang SunKDD 2024
- Efficient Exploration of the Rashomon Set of Rule-Set ModelsMartino Ciaperoni, Han Xiao, Aristides GionisKDD 2024 · 3 citations
- A Scalable Two Stage Approach to Computing Optimal Decision SetsAlexey Ignatiev, Edward Lam, Peter J. Stuckey, João Marques-SilvaAAAI 2021 · 17 citations
- Learning Interpretable Decision Rule Sets: A Submodular Optimization ApproachFan Yang, Kai He, Linxiao Yang, Hongxia Du et al.NeurIPS 2021 · 35 citations
