Jianliang He, Xintian Pan, Siyu Chen, Zhuoran Yang: In-Context Linear Regression Demystified: Training Dynamics and Mechanistic Interpretability of Multi-Head Softmax Attention. ICML 2025