English

Probing Routing-Conditional Calibration in Attention-Residual Transformers

Computer Vision and Pattern Recognition 2026-05-12 v1 Artificial Intelligence

Abstract

Post-hoc calibration is usually evaluated as a function of logits or softmax confidence alone, even as routing-augmented architectures increasingly accompany predictions with sample-specific internal routing traces and pair them with claims of calibration-relevant uncertainty. We ask a basic question: do these traces provide stable routing-specific evidence for post-hoc calibration beyond confidence? We study this in Attention-Residual transformers (Kimi Team, 2026) through a matched-confidence diagnostic suite that stratifies examples by routing-derived state, compares subgroup gaps against within-bin routing-permutation nulls, and evaluates matched post-hoc probes differing only in their auxiliary feature. Across our completed AR runs, scalar routing summaries do not provide stable evidence of routing-conditional miscalibration: weighted gaps remain small or seed-sensitive, and only 11 of 3030 within-bin permutation tests rejects the conditional-null at α=0.05\alpha=0.05 (only on one seed; not stable across seeds in that cell). AR-CondCal, a minimal 22-D Nadaraya--Watson probe on confidence and routing-depth variance, lies within the seed-variance band of matched confidence-only and predictive-entropy controls and does not reliably improve worst-routing-tertile ECE; bandwidth-sensitivity checks (Scott multiples, CV-NLL, global-ECE oracle) do not change this. A full-vector MLP over (c,H1,,HL)(c, H_1, \ldots, H_L) can appear to improve over a linear confidence baseline, but the apparent gain disappears once a capacity-matched confidence-only MLP is included as a control, and shuffled routing profiles achieve comparable performance. Apparent routing-aware calibration gains in this AR setting should not be read as internal-state calibration until matched-confidence, bandwidth, capacity, and permutation controls rule out common confounds.

Keywords

Cite

@article{arxiv.2605.09850,
  title  = {Probing Routing-Conditional Calibration in Attention-Residual Transformers},
  author = {Wenhao Liang and Lin Yue and Wei Emma Zhang and Miao Xu and Mingyu Guo and Olaf Maennel and Weitong Chen},
  journal= {arXiv preprint arXiv:2605.09850},
  year   = {2026}
}

Comments

Under reviewing

R2 v1 2026-07-22T07:02:55.710Z