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.