arXiv Machine Learning By Shutian Liang, Yue Song

Understanding Mamba-3 Through Oscillatory Dynamics

Read the original on arXiv Machine Learning →

Mamba-3 incorporates input‑dependent rotations into its recurrent state updates, which the authors interpret as underdamped second‑order dynamics. Experiments show that these oscillatory dynamics outperform non‑oscillatory regimes on state‑tracking tasks and improve long‑context prediction and length generalization in language‑model pretraining. Further interventions confirm that the learned phase is crucial, suggesting that rotation controls how retained information combines with new updates, thereby supporting long‑context processing.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv AI
Oct 16

Learning to Read Out: Unembedding Dynamics in Language Model Pretraining

The paper investigates when a language model actually learns task information versus when its readout (unembedding matrix) learns to express that information. Using a new parameter‑trajectory cross‑coding method, the authors find that hidden states can encode task data early in training, but the readout reorganizes only in a narrow window (0.4–1.1% of Pythia’s pretraining). By varying the readout learning rate, they show that faster readout learning reduces the gap between hidden‑state encoding and output expression without affecting final performance.

By Matteo He, William F. Shen, Alex Iacob, Andrej Jovanovic, Xinchi Qiu, Nicholas D. Lane