Towards Full Pipeline FP8 Reinforcement Learning for LLMs
A paper on Hugging Face's daily papers describes FP8 reinforcement learning challenges and proposes a solution to fix training instability in LLMs.
This paper reveals challenges in full-pipeline FP8 reinforcement learning for large language models, causing instability and abnormal outputs, and proposes a solution to address these issues.
- reveals full-pipeline FP8 RL instability issue
- proposes Calibrated Clipping method to fix instability
- experiments show performance comparable to BF16 baseline
Full article184 words · extracted from huggingface.co · click to collapse
Reinforcement learning (RL) has become a key technique for improving the reasoning and agentic abilities of large language models (LLMs). Although FP8 quantization can accelerate RL training, maintaining stability throughout an FP8 RL pipeline remains challenging. While previous works have focused on resolving train-inference mismatches using correction techniques like TIS, we reveal that full-pipeline FP8 RL still suffers from severe training instability, manifesting as anomalous mid-training entropy surges and garbled outputs. We trace this instability to a previously overlooked cause: compounded FP8 quantization noise distorts the importance ratio, disproportionately pushing negative-advantage tokens outside the trust region and erroneously zeroing out their gradients. As a result, pathological outputs are not properly penalized and accumulate over the course of training. To address this, we propose Calibrated Clipping, a dynamic method that aligns the FP8 clipping bounds with high-precision BF16 distributions by matching the lower-bound clipping quantile and rebalancing the upper bound accordingly. Extensive experiments across GRPO and DAPO algorithms, model scales from 8B to 32B, and multiple FP8 scaling granularities demonstrate that our approach successfully eliminates entropy surges and restores performance comparable to the BF16 baseline.
Text extracted automatically; images, tables and formatting may be missing. Original: https://huggingface.co/papers/2609.22870