arXiv AI

Rethinking the Test-Time Prompt Tuning Objective from the Perspective of Calibration

arXiv Machine Learning
Aug 27

Sample Margin-Aware Recalibration of Temperature Scaling

The paper introduces SMART, a lightweight recalibration technique that adjusts logits based on the margin between the top two logits, called the logit gap. It uses a soft-binned Expected Calibration Error objective to balance bias and variance, enabling stable updates even with limited calibration data. Experiments across various datasets and architectures show SMART achieves state‑of‑the‑art calibration with fewer parameters than existing methods.

By Haolan Guo, Linwei Tao, Haoyang Luo, Minjing Dong, Chang Xu
Hugging Face Trending Papers
Jun 21

Reliability-Guided Adaptive Ensembling for Robust Test-Time Adaptation

Test-time adaptation (TTA) can mitigate domain shift without source data, but it is highly brittle under adversarially contaminated test streams, where corrupted inputs also destabilize online updates. We study robust test-time adaptation (RTTA) in the adversarial-stream setting, which remains comparatively underexplored relative to standard TTA, and propose SAFER (Stochastic Augmentation Framework for Enhanced Robustness), a training-free reliability-guided augmentation wrapper for RTTA.

arXiv Machine Learning
Oct 12

Optimizable Model Averaging for Ensembling Heterogeneous Model Outputs

Optimizable Model Averaging (OMA) is a black‑box method that assigns nonnegative weights to each candidate model’s predictive probabilities, keeping all model parameters frozen. By minimizing regularized predictive entropy, OMA learns a single scalar coefficient per model to average outputs from heterogeneous architectures. The authors show that appropriate regularization reduces expected calibration error and improves the trade‑off between performance and calibration compared to simple output averaging.

By Mijung Park
arXiv Computer Vision
Oct 26

MAGIC: Margin-gated Contrastive Debiasing with Evidence Restoration for VLM Test-time Robustness Scaling

The paper introduces MAGIC, a method for improving test-time robustness of large vision‑language models by mitigating language bias. MAGIC evaluates decision margins to selectively activate textual and visual perturbation branches, integrates logits from multiple perturbed views, and restores the most informative corrupted views for ambiguous samples. It also uses a calibration dataset to prune ineffective branches, reducing inference cost while enhancing performance on benchmark datasets.

By Shunshun Gu, Enze Zhang, Ming Pei, Hang Zhou, Xiao Luo
arXiv Machine Learning
Jul 10

Prediction-Powered Active Testing

arXiv:2607. 08347v1 Announce Type: cross Abstract: Active testing provides a label--efficient approach to risk estimation by adaptively selecting which test points should be labelled.

By Kianoosh Ashouritaklimi, Valentin Kilian, Daolang Huang, Tom Rainforth, Fran\c{c}ois Caron
arXiv Computer Vision
Oct 9

Do Not Train Away Uncertainty: Early Uncertainty Anchored Calibration

The paper introduces EUA‑Cal, a calibration method that preserves the early‑training uncertainty of deep neural networks. By regularizing predictions and feature prototypes based on the early model, EUA‑Cal mitigates overconfidence while maintaining accuracy. Experiments on image classification and multiple‑choice question answering across eight models show EUA‑Cal outperforms existing calibration techniques.

By Yutong Xie, Jiawei Tang, Zhenglin Hua, Yuxiang Ma, Si Qin, Yaxin Hou, Hui Liu, Junhui Hou, Yuheng Jia
arXiv Computer Vision
Oct 23

Residual Calibration via Local Feature-Space Refinement

The paper introduces Local Adaptive Probability Estimation (LAPE), a post‑hoc residual calibration framework that refines global calibrator outputs using non‑parametric local evidence from neighboring samples in feature space. LAPE serves as a plug‑and‑play module that leverages feature‑space geometry and reuses calibration data, improving calibration quality for various baselines. The authors also extend the approach to multi‑view inputs with Cascade Feature Augmentation Fusion (CFAF) and demonstrate its effectiveness through extensive experiments.

By Xihao Wang, Chengyao Yu, Ruixing Ming