AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Training

Qing Liu, Ou Wu, Yi Du


Abstract
Token selection is pivotal for effective LLM post-training. However, existing methods mostly rely on local token-level heuristics, while response-token selection in post-training lacks an optimization-guided valuation proxy. We introduce AlphaToken, a response token valuation framework that decouples valuation into adaptation (promoting target-task learning) and stability (preserving pre-trained capabilities), and makes each objective path-aware by combining the direct-path signal from local token gradients with the downstream causal-path signal in autoregressive generation. Since retention data are typically unavailable, AlphaToken approximates stability via a Fisher-drift proxy anchored at the pre-trained reference model. For efficient computation, we extend Ghost Dot-Product to token-level valuation. AlphaToken masks low-value response tokens during fine-tuning and preference optimization, concentrating training signals on more valuable positions. Experiments show that AlphaToken improves post-training performance and mitigates catastrophic forgetting. Code at https://github.com/luciabcd/Alphatoken.
Anthology ID:
2026.emnlp-main.866
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:
20028–20054
Language:
URL:
https://aclanthology.org/2026.emnlp-main.866/
DOI:
Bibkey:
Cite (ACL):
Qing Liu, Ou Wu, and Yi Du. 2026. AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Training. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 20028–20054, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Training (Liu et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.866.pdf