MSign: An Optimizer Preventing Training Instability in Large Language Models via Stable Rank Restoration

The episode discusses MSign, an optimizer designed to prevent training instability in large language models. Instability is caused by stable rank collapse and increased Jacobian alignment, which jointly cause exponential gradient growth. MSign addresses this by periodically applying a matrix sign operation to restore stable rank with minimal computational overhead.

Open the full episode

More from AI Radio