TRIAGE: Direction-Aware Mismatch Stabilization of Native NVFP4 Reinforcement Learning
cs.LG, cs.AI
Submitted: 2026-10-05
Updated: 2026-10-09
Code: https://github.com/InfiXAI/TRIAGE
License: http://creativecommons.org/licenses/by/4.0/
The gist: Low-precision execution can substantially accelerate reinforcement learning (RL) for large language models, but discrepancies between learner and sampler execution can destabilize policy optimization.
Terminology
Abstract
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.
Sources
- Scaling Law for Quantization-Aware Training
- Training Verifiers to Solve Math Word Problems
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- ACRL: Adaptive Control of Training-Inference Discrepancy for Stable Reinforcement Learning
- QaRL: Rollout-Aligned Quantization-Aware RL for Fast and Stable Training under Training--Inference Mismatch
- QeRL: Beyond Efficiency -- Quantization-enhanced Reinforcement Learning for LLMs
- QuRL: Efficient Reinforcement Learning with Quantized Rollout
- Quantization Meets Reasoning: Exploring LLM Low-Bit Quantization Degradation for Mathematical Reasoning
- Let's Verify Step by Step
- Pretraining Large Language Models with NVFP4
- Quartet II: Accurate LLM Pre-Training in NVFP4 by Improved Unbiased Gradient Estimation
- FP8-RL: A Practical and Stable Low-Precision Stack for LLM Reinforcement Learning
- Composer 2 Technical Report
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism
- InfiR : Crafting Effective Small Language Models and Multimodal Small Language Models in Reasoning
- SSPO: Subsentence-level Policy Optimization
- Diagnosing Training Inference Mismatch in LLM Reinforcement Learning via a Zero-Mismatch Reference
- QUADS: Stabilizing NVFP4 Reinforcement Learning for MoE via QUantization-error Alignment across Dual Sides
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks