FP4 Reinforcement Learning Post-Training for Large Language Models
Researchers present the first end-to-end FP4 reinforcement learning post-training for large language models, where both rollout and training policies operate at 4-bit precision. A systematic study identifies rollout activation quantization as the primary source of degradation, not training-side quantization. Outliers in activations cause underflow to zero under FP4. Restoring training policy to higher precision while keeping rollout in FP4 worsens accuracy, revealing rollout-training mismatch as the main failure mode. To address this, they propose Rollout Residual Quantization (Rollout-ResQ), a single residual correction term with a hardware-friendly sparsity pattern added only to the FP4 rollout matrix multiplication. This lightweight correction recovers performance.
Key facts
- First end-to-end FP4 RL post-training for LLMs
- Both rollout and training policies operate at 4-bit precision
- Rollout activation quantization is the dominant source of degradation
- Outliers cause activation values to underflow to zero under FP4
- Restoring training policy to higher precision worsens accuracy
- Rollout-training mismatch is the principal failure mode
- Rollout-ResQ adds a single residual correction term
- Correction is constrained to a hardware-friendly sparsity pattern
Entities
—