TRIAGE: Direction-Aware Mismatch Stabilization of Native NVFP4 Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Low-precision execution can substantially accelerate reinforcement learning (RL) for large language models, but discrepancies between learner and sampler execution can destabilize policy optimization. In this paper, we characterize the interaction between mismatch and the policy-gradient direction, distinguishing locally amplifying from contracting update contributions that mismatch magnitude alone cannot identify. In native NVFP4 runs, we observe an early imbalance between the two amplifying regions, favoring negative-advantage, negative-gap updates. Their tail tokens become concentrated in a small fraction of response segments before mismatch spreads globally. Motivated by these findings, we introduce TRIAGE, a direction-aware stabilization method that uses segment-level diagnosis to selectively rebalance policy-gradient updates and applies bounded repair to residual severe mismatch. TRIAGE modifies the optimization objective while retaining native NVFP4 weight-and activation 4-bit (W4A4) forward execution on both the sampler and learner. Experiments on Qwen3-4B and Qwen3-30B-A3B show stable optimization throughout the evaluated training horizon and achieve full precision level performance across five mathematical reasoning benchmarks, while native NVFP4 with TRIAGE provides up to 2.3x higher rollout throughput than BF16.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). First posted on arXiv 5 Oct 2026.
Builds on
No reference list yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026QUADS: Stabilizing NVFP4 Reinforcement Learning for MoE via QUantization-error Alignment across Dual Sides
- 2026GAC: Stabilizing Asynchronous RL Training for LLMs via Gradient Alignment Control
- 2026COPC: Coupled Off-Policy Correction for Asynchronous LLM Reinforcement Learning
- 2025Stabilizing Reinforcement Learning with LLMs: Formulation and Practices
- 2026RTPO: Reverse-Turn Policy Optimization for Stabilizing Agentic RL Training
- 2026ASymPO: Asymmetric-Scale Policy Optimization for Asynchronous LLM Post-Training Without Behavior Information
- 2026BRACE: Anchored Bellman-Residual Correction for Stale Critics in Asynchronous RL
- 2025COPO: Consistency-Aware Policy Optimization