cs.CLJun 1, 2026

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

Authors: Liu QingOu WuYi Du

Organizations: Hangzhou Institute for Advanced Study

Abstract

Token selection is pivotal for effective LLM post-training. However, existing methods mostly rely on local heuristics and rarely formulate token selection as a principled valuation of individual response tokens. We introduce AlphaToken\textbf{AlphaToken}, a response token valuation framework that decouples valuation into adaptation\textbf{adaptation} (promoting target-task learning) and stability\textbf{stability} (preserving pre-trained capabilities), and makes each objective path-aware\textbf{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\textbf{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.

Explore similar work

CardsList