DeMix: Debugging Training Data with Mixed Data Error Types by Investigating Influence Vectors
Not on the map. Why?
Abstract (arXiv)
High-quality training data is essential for the success of machine learning models. However, real-world datasets often contain mixed types of errors arising from systematic flaws in data preparation pipelines, including label errors, feature errors, and spurious correlations. Effective debugging of training data requires both detecting erroneous samples and identifying their specific error types to enable targeted repair, yet existing data cleaning and attribution methods fail to adequately address this dual requirement. In this paper, we propose DeMix, a novel framework that simultaneously diagnoses erroneous samples and their error types. Our key insight is that different error types produce distinct patterns on model behavior. DeMix captures such error-specific patterns by influence vectors that characterize how each training sample affects model predictions across all validation samples. We formulate training data debugging as a multi-label classification problem where a classifier is developed to predict error types directly from influence vectors. We further introduce an intervention-based learning strategy that guides the classifier to capture invariant rationales specific to each error type, ensuring the learned classifier generalizes effectively. Empirical evaluations on 11 tasks across tabular data prediction, recommendation systems, and LLM alignment demonstrate that DeMix significantly outperforms state-of-the-art approaches, achieving a 22.61% improvement in data debugging F1-score and a 9.32% gain in task model performance after data repair. Code is available at: https://github.com/SJTU-DMTai/DeMix.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 21 of its 44 references that are in the map. First posted on arXiv 10 Jun 2026.
Builds on
21 of 44 references.
References · 21
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
- 2016Deep Variational Information Bottleneckcited by 930
- 2017Deep Interest Network for Click-Through Rate Predictioncited by 544
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 532
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 523
- 2018Deep Interest Evolution Network for Click-Through Rate Predictioncited by 338
- 2022Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanismcited by 171
- 2024ODIN: Disentangled Reward Mitigates Hacking in RLHFcited by 136
- 2022Training Data Influence Analysis and Estimation: A Surveycited by 135
- 2023Discover and Cure: Concept-aware Mitigation of Spurious Correlationcited by 71
9 more
- 2024From Lists to Emojis: How Format Bias Affects Model Alignmentcited by 49
- 2024Dataset Regeneration for Sequential Recommendationcited by 19
- 2025DataMan: Data Manager for Pre-training Large Language Modelscited by 17
- 2022Back to the Basics: Revisiting Out-of-Distribution Detection Baselinescited by 16
- 2025A Comprehensive Survey on Imbalanced Data Learningcited by 10
- 2023Error Discovery by Clustering Influence Embeddingscited by 8
- 2025Influence Guided Context Selection for Effective Retrieval-Augmented Generationcited by 7
- 2025Shapley Value-driven Data Pruning for Recommender Systemscited by 3
- 2025Improving Group Robustness on Spurious Correlation via Evidential Alignmentcited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024Delta-Influence: Unlearning Poisons via Influence Functions
- 2026When Offline Selectors Cannot Beat the Best Single Model: A Diagnostic Study on edX Dropout Prediction
- 2025ZeroED: Hybrid Zero-shot Error Detection through Large Language Model Reasoning
- 2024DECIDER: Leveraging Foundation Model Priors for Improved Model Failure Detection and Explanation
- 2025Exploring LLM Agents for Cleaning Tabular Machine Learning Datasets
- 2025Rectified Decoupled Dataset Distillation: A Closer Look for Fair and Comprehensive Evaluation
- 2022Distilling Model Failures as Directions in Latent Space
- 2025Conformal Unlearning: A New Paradigm for Unlearning in Conformal Predictors