Hsun-Yu Kuo

Stabilizing Extrapolation in Looped Transformers via Learned Stochastic Stopping

This work suggests that "when to stop" should be treated as a training-time design choice, not merely an inference-time computation-allocation rule.