mlsanityData to 14 Oct 2026

Too Sharp, Too Sure: When Calibration Follows Curvature

arXiv 2604.20614 ↗First posted 22 Apr 2026cs.LG math.DS math.OCcited by 3

Not on the map. Why?

Abstract (arXiv)

Modern neural networks can achieve high accuracy while remaining poorly calibrated, producing confidence estimates that do not match empirical correctness. Yet calibration is often treated as a post-hoc attribute. We take a different perspective: we study calibration as a training-time phenomenon on small vision tasks, and ask whether calibrated solutions can be obtained reliably by intervening on the training procedure. We identify a tight coupling between calibration, curvature, and margins during training of deep networks under multiple gradient-based methods. Empirically, Expected Calibration Error (ECE) closely tracks curvature-based sharpness throughout optimization. Mathematically, we show that both ECE and Gauss--Newton curvature are controlled, up to problem-specific constants, by the same margin-dependent exponential tail functional along the trajectory. Guided by this mechanism, we introduce a margin-aware training objective that explicitly targets robust-margin tails and local smoothness, yielding improved out-of-sample calibration across optimizers without sacrificing accuracy.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 42 of its 56 references that are in the map. First posted on arXiv 22 Apr 2026.

Builds on

42 of 56 references.

References · 42

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2017On Calibration of Modern Neural Networkscited by 4,960
  3. 2019Theoretically Principled Trade-off between Robustness and Accuracycited by 1,517
  4. 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
  5. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
  6. 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
  7. 2019When Does Label Smoothing Help?cited by 988
  8. 2017Spectrally-normalized margin bounds for neural networkscited by 686
  9. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
  10. 2017Exploring Generalization in Deep Learningcited by 600
  11. 2019Fantastic Generalization Measures and Where to Find Themcited by 513
  12. 2017Sharp Minima Can Generalize For Deep Netscited by 466
30 more
  1. 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 464
  2. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
  3. 2019Beyond temperature scaling: Obtaining well-calibrated multiclass probabilities with Dirichlet calibrationcited by 364
  4. 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 359
  5. 2020Calibrating Deep Neural Networks using Focal Losscited by 357
  6. 2019On Mixup Training: Improved Calibration and Predictive Uncertainty for Deep Neural Networkscited by 305
  7. 2017Train longer, generalize better: closing the generalization gap in large batch training of neural networkscited by 279
  8. 2020Mix-n-Match: Ensemble and Compositional Methods for Uncertainty Calibration in Deep Learningcited by 190
  9. 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 101
  10. 2017Fisher-Rao Metric, Geometry, and Complexity of Neural Networkscited by 100
  11. 2021The Devil is in the Margin: Margin-based Label Smoothing for Network Calibrationcited by 91
  12. 2018On the Relation Between the Sharpest Directions of DNN Loss and the SGD Step Lengthcited by 88
  13. 2018A Walk with SGDcited by 78
  14. 2020Regularizing Neural Networks via Adversarial Model Perturbationcited by 78
  15. 2019Normalized Flat Minima: Exploring Scale Invariant Definition of Flat Minima for Neural Networks using PAC-Bayesian Analysiscited by 59
  16. 2021Relating Adversarially Robust Generalization to Flat Minimacited by 57
  17. 2019Deterministic PAC-Bayesian generalization bounds for deep networks via generalizing noise-resiliencecited by 51
  18. 2020Rethinking Parameter Counting in Deep Models: Effective Dimensionality Revisitedcited by 51
  19. 2022SAM as an Optimal Relaxation of Bayescited by 41
  20. 2024Does SGD really happen in tiny subspaces?cited by 36
  21. 2024Edge of Stochastic Stability: Revisiting the Edge of Stability for SGDcited by 30
  22. 2021Meta-Calibration: Learning of Model Calibration Using Differentiable Expected Calibration Errorcited by 29
  23. 2020Improving Calibration through the Relationship with Adversarial Robustnesscited by 26
  24. 2024Sharpness-Aware Minimization Efficiently Selects Flatter Minima Late in Trainingcited by 24
  25. 2022The Calibration Generalization Gapcited by 18
  26. 2025Rethinking Early Stopping: Refine, Then Calibratecited by 17
  27. 2023Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimizationcited by 13
  28. 2025Benefits of Early Stopping in Gradient Descent for Overparameterized Logistic Regressioncited by 9
  29. 2025Optimal and Provable Calibration in High-Dimensional Binary Classification: Angular Calibration and Platt Scalingcited by 2
  30. 2025Towards Understanding The Calibration Benefits of Sharpness-Aware Minimizationcited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.