Paper · · arXiv

Exact Learning Dynamics of In-Context Learning in Linear Transformers and Its Application

We derive exact stochastic gradient dynamics for a linear transformer, which reveal staged learning and conservation laws, and then use the same measurements to study nonlinear models.