Estimating Instance-dependent Bayes-label Transition Matrix using a Deep Neural Network
Shuo YangErkun YangBo HanYang LiuMin XuGang NiuTongliang Liu
Proposes modeling instance-dependent label noise by estimating transitions from Bayes optimal labels to noisy labels using a deep neural network, shrinking the search space and improving classification accuracy on noisy datasets.
Modern machine learning applications rely heavily on massive datasets collected through web scraping, online queries, and crowdsourcing platforms. Because manual verification at scale is prohibitively expensive, these real-world datasets inevitably contain substantial amounts of mislabeled data. Deep neural networks tend to memorize these incorrect labels, which severely degrades model performance and reliability in deployment. A particularly challenging and realistic scenario is instance-dependent label noise, where the likelihood of a label error depends directly on the specific features or quality of an individual sample. Addressing this issue is critical for organizations deploying artificial intelligence systems where data quality cannot be manually guaranteed.
The article evaluates a new framework for training accurate deep learning classifiers on datasets corrupted by instance-dependent label noise. Specifically, it demonstrates that modeling the transition probabilities from optimal target categories (termed Bayes optimal labels) to observed noisy labels enables a deep neural network to directly estimate instance-specific noise patterns without relying on clean labels or rigid hand-crafted assumptions.
To achieve this, the approach identifies a subset of high-confidence training examples whose optimal labels can be mathematically inferred from the noisy data. A dedicated deep neural network is then trained on these distilled examples to estimate the transition behavior from optimal to noisy labels for any given input. Finally, this transition network is fixed and used to correct the loss function while training the main classification model across the entire noisy dataset. The authors validated this method using three benchmark image datasets corrupted with synthetic noise levels ranging from 10% to 50% across multiple runs, as well as one large-scale real-world dataset containing over one million noisily labeled clothing images.
The key findings demonstrate significant performance gains across all testing environments. First, the proposed method consistently outperformed leading baseline approaches across all noise levels on the synthetic benchmark datasets. Second, the performance advantage widened substantially as noise severity increased; for instance, on the CIFAR-10 image benchmark, the method achieved a 5.83 percentage point lead over the top-performing baseline at 10% noise, expanding to a 7.01 percentage point lead at 40% noise. Similar gains occurred on the SVHN digit dataset, where the performance margin grew from 2.60 percentage points at 10% noise to 7.14 percentage points at 50% noise. Third, on the real-world clothing dataset, the framework reached 73.39% test accuracy when combined with matrix revision, exceeding all baseline techniques.
These results indicate that estimating noise transitions based on optimal labels reduces mathematical uncertainty and provides better generalization compared to traditional methods that attempt to model clean distributions directly. For organizations, this approach mitigates the risk of model failure and reduces data cleaning costs by allowing systems to train effectively on imperfect, cheaply gathered datasets without requiring manual relabeling.
Organizations handling large, imperfectly labeled data should consider adopting parametric transition models and loss-correction techniques when training deep networks. To improve outcomes, technical teams can combine this method with existing matrix revision techniques to further boost accuracy. Prior to large-scale production deployment, engineering teams should conduct pilot tests on domain-specific data to tune the noise-bound threshold and verify stability across varying operational conditions.
The study operates under the assumption of bounded noise rates, meaning that error probabilities do not exceed a certain maximum threshold. While the method demonstrates high empirical reliability on standard image recognition benchmarks, caution is warranted when applying it to domains with severe label corruption exceeding theoretical bounds or data modalities outside standard image classification.
- Paper: Making Deep Neural Networks Robust to Label Noise: A Loss Correction Approach, Giorgio Patrini et al. (2016). Its forward loss correction uses a label-transition matrix to adjust training, the direct foundation this paper adapts from class-level noise rates to instance-specific transitions.
- Paper: Learning with Noisy Labels, Nagarajan Natarajan et al. (2013). Its unbiased-loss and label-dependent-cost corrections establish the earlier theory for learning under noisy labels that contextualizes the source’s transition-based loss correction.
No sufficiently relevant recommendations were found.
