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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | CIFAR-100 (test) | Accuracy90.4 | 63 | |
| Image Classification | CIFAR-10-C (test) | ECE2.8 | 37 | |
| Image Classification | Camelyon17 OOD (test) | Accuracy91.46 | 14 | |
| Image Classification | Camelyon17 ID (val) | Accuracy90.68 | 14 | |
| Image Classification | CIFAR-100-C averaged (test) | Accuracy74.15 | 9 | |
| Image Classification Calibration | CAMELYON 17 | Accuracy89.19 | 9 | |
| Image Classification Calibration | CIFAR-10-C | Accuracy75.12 | 9 | |
| Image Classification Calibration | CIFAR-100-C | Accuracy52.66 | 9 | |
| Image Classification Calibration | iWILDCam | Accuracy76.11 | 9 | |
| Image Classification Calibration | Tiny-ImageNet-C | Accuracy24.03 | 9 |