Probing Routing-Conditional Calibration in Attention-Residual Transformers
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 of within-bin permutation tests rejects the conditional-null at (only on one seed; not stable across seeds in that cell). AR-CondCal, a minimal -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 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.
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