When fine-tuning large language models with reinforcement learning, BF16 saves memory but is less precise than FP16. At low learning rates, small weight updates get lost in rounding, slowing convergence.
The researchers solved this by using stochastic rounding for BF16 weights: each weight is rounded up or down based on a probability, so even small updates affect the expected weight.
This matches FP16's convergence speed without increasing model drift from its original behavior, improving training efficiency while preserving learned capabilities.
