Explainable and Local Correction of Classification Models Using Decision Trees
Hirofumi Suzuki, Hiroaki Iwashita, Takuya Takagi, Keisuke Goto, Yuta Fujishige, Satoshi Hara
Abstract
In practical machine learning, models are frequently updated, or corrected, to adapt to new datasets. In this study, we pose two challenges to model correction. First, the effects of corrections to the end-users need to be described explicitly, similar to standard software where the corrections are described as release notes. Second, the amount of corrections need to be small so that the corrected models perform similarly to the old models. In this study, we propose the first model correction method for classification models that resolves these two challenges. Our idea is to use an additional decision tree to correct the output of the old models. Thanks to the explainability of decision trees, the corrections are describable to the end-users, which resolves the first challenge. We resolve the second challenge by incorporating the amount of corrections when training the additional decision tree so that the effects of corrections to be small. Experiments on real data confirm the effectiveness of the proposed method compared to existing correction methods.
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 c599efd1-5f48-4773-871f-705a60fc9ff1Builds on1
Related papers
- Personal Insights for Altering Decisions of Tree-based Ensembles over TimeNave Frost, Naama Boer, Daniel Deutch, Tova MiloVLDB 2020 · 7 citations
- Backward Compatibility in Tree-Based Explanations and Enhanced CART AlgorithmHirofumi SuzukiKDD 2026
- Editing a classifier by rewriting its prediction rulesShibani Santurkar, Dimitris Tsipras, Mahalaxmi Elango, David Bau et al.NeurIPS 2021 · 105 citations
- Clarify: Improving Model Robustness With Natural Language CorrectionsYoonho Lee, Michelle S. Lam, Helena Vasconcelos, Michael S. Bernstein et al.UIST 2024 · 3 citations
- AIMEE: An Exploratory Study of How Rules Support AI Developers to Explain and Edit ModelsDavid Piorkowski, Inge Vejsbjerg, Owen Cornec, Elizabeth M. Daly et al.CSCW 2023 · 8 citations
