Same softmax numerical bug also affects KL regularizers in RL, researcher notes

heghbalz · x · 2026-10-28

Brandon Damos points out that the near-agreement numerical issue from the softmax/cross-entropy bug paper extends beyond distillation: it appears anywhere a KL regularizer is used, such as RL with maxπ E[r(y)] − β KL(π, πref). He adds that reverse KL and JSD have similar issues with analogous fixes, with a summary thread.

Related event: Google Paper Reveals Two Numerical Bugs in All Softmax Implementations(3 posts)→

Original post →

More from Research

Research channel →