English

QuAIL: Quality-Aware Inertial Learning for Robust Training under Data Corruption

Machine Learning 2026-02-04 v1 Artificial Intelligence

Abstract

Tabular machine learning systems are frequently trained on data affected by non-uniform corruption, including noisy measurements, missing entries, and feature-specific biases. In practice, these defects are often documented only through column-level reliability indicators rather than instance-wise quality annotations, limiting the applicability of many robustness and cleaning techniques. We present QuAIL, a quality-informed training mechanism that incorporates feature reliability priors directly into the learning process. QuAIL augments existing models with a learnable feature-modulation layer whose updates are selectively constrained by a quality-dependent proximal regularizer, thereby inducing controlled adaptation across features of varying trustworthiness. This stabilizes optimization under structured corruption without explicit data repair or sample-level reweighting. Empirical evaluation across 50 classification and regression datasets demonstrates that QuAIL consistently improves average performance over neural baselines under both random and value-dependent corruption, with especially robust behavior in low-data and systematically biased settings. These results suggest that incorporating feature reliability information directly into optimization dynamics is a practical and effective approach for resilient tabular learning.

Keywords

Cite

@article{arxiv.2602.03686,
  title  = {QuAIL: Quality-Aware Inertial Learning for Robust Training under Data Corruption},
  author = {Mattia Sabella and Alberto Archetti and Pietro Pinoli and Matteo Matteucci and Cinzia Cappiello},
  journal= {arXiv preprint arXiv:2602.03686},
  year   = {2026}
}
R2 v1 2026-07-01T09:34:33.353Z