Search papers, labs, and topics across Lattice.
The paper identifies a stability vs. effectiveness trade-off in Looped Language Models (LoopLMs) that limits their ability to scale reasoning at test time. They propose STARS, a training framework that uses Jacobian Spectral Radius Regularization with random loop sampling to constrain latent states to approach asymptotically stable fixed points, thereby maximizing effectiveness while ensuring stability. Experiments on arithmetic and mathematical reasoning tasks demonstrate that STARS enables LoopLMs to scale more reliably at test time, mitigating performance degradation as recurrence depth increases and improving peak performance.
LoopLMs don't reliably scale at test time because of an inherent stability vs. effectiveness trade-off, but a new training method can fix that.
Looped Language Models (LoopLMs) enable efficient latent reasoning through depth recurrence, yet exhibit unreliable test-time scaling behavior: performance often peaks at a certain iteration depth and then collapses with further recurrence. Through latent dynamics analysis, we find an inherent trade-off between stability and effectiveness in existing architectures and strategies. By conceptualizing reasoning as uncertainty reduction, we propose that convergence toward stable fixed points while preserving effectiveness represents a promising way. To this end, we propose STARS (STAbility-driven Recurrent Scaling), a training framework that constrains latent states to approach asymptotically stable fixed points. This is realized via efficient Jacobian Spectral Radius Regularization with random loop sampling, enabling STARS to maximize effectiveness while ensuring rigorous stability. Experiments on arithmetic tasks show that STARS achieves reliable test-time scaling, and on complex mathematical reasoning it substantially mitigates performance degradation as recurrence depth increases while also improving peak performance.
