Researchers have identified and fixed two numerical bugs in standard implementations of softmax, log-softmax, and cross-entropy. The first bug, Softmax Collapse, causes the target logit's gradient to round to zero when one logit is significantly larger than others, disrupting the gradient sum. The second bug, Softmax Cancellation, produces non-zero gradients even when student and teacher logits are identical, occurring when these logits are close. The proposed solutions eliminate the "Slingshot" phenomenon, accelerate grokking with weight decay, and prevent accuracy collapse without it. AI
IMPACT Fixes may improve training stability and efficiency for models using these common functions.
RANK_REASON The cluster contains an academic paper detailing a technical fix for common machine learning functions. [lever_c_demoted from research: ic=1 ai=1.0]
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →