Amortized-Precision Quantization for Early-Exit Vision Transformers
For practitioners deploying efficient ViTs, this provides a principled method to stabilize early-exit inference under quantization, achieving superior accuracy-efficiency trade-offs.
APQ addresses instability in low-precision early-exit ViTs by modeling layer-wise quantization noise exposure, and MAQEE jointly optimizes exit thresholds and bit-widths, reducing BOPs by up to 95% while maintaining accuracy and outperforming baselines by up to 20% across multiple vision tasks.
Vision Transformers (ViTs) achieve strong performance across vision tasks, yet their deployment with low-precision early exiting remains fragile. Existing quantization methods assume static full-depth execution, making them unstable when exit decisions are perturbed by quantization noise, which can amplify errors along dynamic inference paths. In this paper, we introduce Amortized-Precision Quantization (APQ), a utilization-aware formulation that accounts for layer-wise stochastic exposure to quantization noise and reveals depth-precision trade-offs. Building on APQ, we propose Mutual Adaptive Quantization with Early Exiting (MAQEE), a bi-level framework that jointly optimizes exit thresholds and bit-widths under explicit risk control to improve inference stability. MAQEE establishes a superior Pareto frontier in the accuracy-efficiency trade-off, reducing BOPs by up to 95% while maintaining accuracy and outperforming strong baselines by up to 20\% across classification, detection, and segmentation tasks.