Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Target-Agnostic Calibration under Distribution Shift with Frequency-Aware Gradient Rectification

About

Real-world model deployments inevitably encounter distribution shifts, rendering the confidence estimates of deep neural networks highly unreliable, posing severe risks in safety-critical applications. Existing methods improve calibration via training-time regularization or post-hoc adjustment, but often rely on access to (or simulation of) target domains, limiting practicality. We propose Frequency-aware Gradient Rectification (FGR), a target-agnostic training framework for robust calibration. From a frequency perspective, FGR applies low-pass filtering to a subset of training images to diminish spurious high-frequency cues and encourage the learning of domain-invariant features. However, the associated information loss can degrade In-Distribution (ID) calibration. To resolve this trade-off, FGR treats ID calibration as a hard constraint and rectifies conflicting parameter updates via geometric projection. This ensures a first-order non-increase in the ID calibration objective without introducing an additional loss-balancing coefficient. Extensive experiments on synthetic, real-world, and semantic shift datasets demonstrate that FGR significantly improves calibration under diverse shifts while preserving ID performance, and it remains compatible with post-hoc calibration methods. Our code is available at https://github.com/YilinZhang107/FGR-Calib.

Yilin Zhang, Cai Xu, You Wu, Ziyu Guan, Wei Zhao• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)
Accuracy90.4
63
Image ClassificationCIFAR-10-C (test)
ECE2.8
37
Image ClassificationCamelyon17 OOD (test)
Accuracy91.46
14
Image ClassificationCamelyon17 ID (val)
Accuracy90.68
14
Image ClassificationCIFAR-100-C averaged (test)
Accuracy74.15
9
Image Classification CalibrationCAMELYON 17
Accuracy89.19
9
Image Classification CalibrationCIFAR-10-C
Accuracy75.12
9
Image Classification CalibrationCIFAR-100-C
Accuracy52.66
9
Image Classification CalibrationiWILDCam
Accuracy76.11
9
Image Classification CalibrationTiny-ImageNet-C
Accuracy24.03
9
Showing 10 of 13 rows

Other info

Follow for update