Balancing Two Classifiers via A Simplex ETF Structure for Model Calibration
Not on the map. Why?
Abstract (arXiv)
In recent years, deep neural networks (DNNs) have demonstrated state-of-the-art performance across various domains. However, despite their success, they often face calibration issues, particularly in safety-critical applications such as autonomous driving and healthcare, where unreliable predictions can have serious consequences. Recent research has started to improve model calibration from the view of the classifier. However, the exploration of designing the classifier to solve the model calibration problem is insufficient. Let alone most of the existing methods ignore the calibration errors arising from underconfidence. In this work, we propose a novel method by balancing learnable and ETF classifiers to solve the overconfidence or underconfidence problem for model Calibration named BalCAL. By introducing a confidence-tunable module and a dynamic adjustment method, we ensure better alignment between model confidence and its true accuracy. Extensive experimental validation shows that ours significantly improves model calibration performance while maintaining high predictive accuracy, outperforming existing techniques. This provides a novel solution to the calibration challenges commonly encountered in deep learning.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 37 of its 54 references that are in the map. First posted on arXiv 14 Apr 2025.
Builds on
37 of 54 references.
References · 37
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2017Mask R-CNNcited by 7,277
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2016Wide Residual Networkscited by 3,070
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
- 2018YOLOv3: An Incremental Improvementcited by 2,659
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
25 more
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 989
- 2019When Does Label Smoothing Help?cited by 988
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
- 2020Calibrating Deep Neural Networks using Focal Losscited by 357
- 2019On Mixup Training: Improved Calibration and Predictive Uncertainty for Deep Neural Networkscited by 305
- 2023Neural Collapse Inspired Feature-Classifier Alignment for Few-Shot Class Incremental Learningcited by 116
- 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
- 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?cited by 100
- 2021The Devil is in the Margin: Margin-based Label Smoothing for Network Calibrationcited by 91
- 2017End-to-end Training for Whole Image Breast Cancer Diagnosis using An All Convolutional Designcited by 88
- 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
- 2015Posterior calibration and exploratory analysis for natural language processing modelscited by 76
- 2022Neural Collapse Inspired Attraction-Repulsion-Balanced Loss for Imbalanced Learningcited by 43
- 2020Sequential Deep Learning for Credit Risk Monitoring with Tabular Financial Datacited by 39
- 2023Dual Focal Loss for Calibrationcited by 33
- 2022AdaFocal: Calibration-aware Adaptive Focal Losscited by 32
- 2021Machine learning based disease diagnosis: A comprehensive reviewcited by 30
- 2022Understanding and Improving Transfer Learning of Deep Models via Neural Collapsecited by 24
- 2023A Neural Collapse Perspective on Feature Evolution in Graph Neural Networkscited by 19
- 2023ACLS: Adaptive and Conditional Label Smoothing for Network Calibrationcited by 15
- 2023Generalizing and Decoupling Neural Collapse via Hyperspherical Uniformity Gapcited by 14
- 2023Are Neurons Actually Collapsed? On the Fine-Grained Structure in Neural Representationscited by 10
- 2023Calibrating a Deep Neural Network with Its Predecessorscited by 9
- 2023Quantifying the Variability Collapse of Neural Networkscited by 9
- 2024Decoupling Feature Extraction and Classification Layers for Calibrated Neural Networkscited by 5
Built on by
Citing papers per year: 2025: 2; 2026: 1.
3 papers.