AlphaToken: Decoupling Adaptation and Stability for Path-Aware Response Token Valuation in LLM Post-Training
Not on the map. Why?
Abstract (arXiv)
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, 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.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 43 of its 59 references that are in the map. First posted on arXiv 1 Jun 2026.
Builds on
43 of 59 references.
References · 43
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2016Learning without Forgettingcited by 2,465
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2023Enhancing Chat Language Models by Scaling High-quality Instructional Conversationscited by 825
- 2021Robust fine-tuning of zero-shot modelscited by 785
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
31 more
- 2021Deep Learning on a Data Diet: Finding Important Examples Early in Trainingcited by 529
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2019Towards Efficient Data Valuation Based on the Shapley Valuecited by 279
- 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 269
- 2020Influence Functions in Deep Learning Are Fragilecited by 247
- 2019Limitations of the Empirical Fisher Approximation for Natural Gradient Descentcited by 213
- 2021Beta Shapley: a Unified and Noise-reduced Data Valuation Framework for Machine Learningcited by 162
- 2019Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithmscited by 147
- 2020A Principled Approach to Data Valuation for Federated Learningcited by 127
- 2022Data Banzhaf: A Robust Data Valuation Framework for Machine Learningcited by 126
- 2021Sampling Permutations for Shapley Value Estimationcited by 101
- 2020A Distributional Framework for Data Valuationcited by 79
- 2024Data Shapley in One Training Runcited by 76
- 2024TokenSHAP: Interpreting Large Language Models with Monte Carlo Shapley Value Estimationcited by 41
- 2025Token Cleaning: Fine-Grained Data Selection for LLM Supervised Fine-Tuningcited by 39
- 2024SelectIT: Selective Instruction Tuning for LLMs via Uncertainty-Aware Self-Reflectioncited by 34
- 2024Capturing the Temporal Dependence of Training Data Influencecited by 34
- 2020A Multilinear Sampling Algorithm to Estimate Shapley Valuescited by 33
- 2024Selective Preference Optimization via Token-Level Reward Function Estimationcited by 25
- 2025SFT Doesn't Always Hurt General Capabilities: Revisiting Domain-Specific Fine-Tuning in LLMscited by 23
- 2025Upweighting Easy Samples in Fine-Tuning Mitigates Forgettingcited by 16
- 2026ProFit: Leveraging High-Value Signals in SFT via Probability-Guided Token Selectioncited by 16
- 2025TokenShapley: Token Level Context Attribution with Shapley Valuecited by 14
- 2025Mitigating Forgetting in LLM Fine-Tuning via Low-Perplexity Token Learningcited by 12
- 2025ConfPO: Exploiting Policy Model Confidence for Critical Token Selection in Preference Optimizationcited by 10
- 2024SparsePO: Controlling Preference Alignment of LLMs via Sparse Token Maskscited by 9
- 2025Winning the Pruning Gamble: A Unified Approach to Joint Sample and Token Pruning for Efficient Supervised Fine-Tuningcited by 8
- 2025Rethinking Supervised Fine-Tuning: Emphasizing Key Answer Tokens for Improved LLM Accuracycited by 4
- 2025TRIM: Token-wise Attention-Derived Saliency for Data-Efficient Instruction Tuningcited by 3
- 2026Explainable Token-level Noise Filtering for LLM Fine-tuning Datasetscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Decoupling Token Roles in Autoregressive Pretrainingcited by 0
- 2026Decoupling Token Roles in Autoregressive Pretraining27 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL
- 2026JoyAI-LLM Flash: Advancing Mid-Scale LLMs with Token Efficiency
- 2025Optimal Transport-Based Token Weighting scheme for Enhanced Preference Optimization
- 2026GMTS: Gradient Magnitude-based Token Selection Improves RLVR Training for LLM Reasoning
- 2026Consolidating Rewarded Perturbations for LLM Post-Training
- 2026Which Tokens Matter? Adaptive Token Selection for RLVR with the Relative Surprisal Index
- 2026InfoSFT: Learn More and Forget Less with Information-Aware Token Weighting
- 2026Training-Trajectory-Aware Token Selection