Deeper is Not Always Better: Mitigating the Alignment Tax via Confident Layer Decoding

Sining Zhoubian, Xuanming Zhang, Yuxuan Chen, Tianyi Tang, An Yang, Sean Du, Chujie Zheng, Fei Huang, Dayiheng Liu, Gao Huang, Jingren Zhou


Abstract
Autoregressive generation in large language models (LLMs) conventionally decodes from the final layer, assuming that deeper representations yield more reliable next-token predictions. We revisit this assumption by revealing a recurring **Guess–Refine–Perturb** dynamic: early layers form coarse guesses, intermediate layers refine reasoning-relevant semantics, and final layers can perturb these refined predictions toward generic or alignment-preferred tokens. We introduce **Confident Decoding**, a training-free decoding strategy that dynamically selects a near-final layer through entropy-guided conservative backward search. We further formulate layer selection as an optimal stopping problem and show that, under explicit local-monotonicity and bounded-probe-noise conditions, the search rule filters terminal perturbations while bounding unfinished refinement. Experiments across dense and Mixture-of-Experts LLMs show that gains concentrate on challenging reasoning settings, including GPQA-Diamond, Omni-MATH, and HLE, while saturated tasks remain broadly stable, with no additional KV-cache memory and less than 2% latency increase. These results suggest dynamically bypassing final-layer perturbations can recover stronger reasoning behavior from aligned LLMs.
Anthology ID:
2026.emnlp-main.1086
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
24944–24970
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1086/
DOI:
Bibkey:
Cite (ACL):
Sining Zhoubian, Xuanming Zhang, Yuxuan Chen, Tianyi Tang, An Yang, Sean Du, Chujie Zheng, Fei Huang, Dayiheng Liu, Gao Huang, and Jingren Zhou. 2026. Deeper is Not Always Better: Mitigating the Alignment Tax via Confident Layer Decoding. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 24944–24970, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Deeper is Not Always Better: Mitigating the Alignment Tax via Confident Layer Decoding (Zhoubian et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1086.pdf