Moonshot AI
Attention Residuals
The paper proposes Attention Residuals (AttnRes), replacing fixed unit-weight residual connections in LLMs with learned, input-dependent softmax attention over preceding layer outputs. This addresses PreNorm dilution, where hidden-state magnitudes grow with depth, diluting layer contributions. A scalable variant, Block AttnRes, partitions layers into…
Kimi Team, Guangyu Chen, Yu Zhang, Jianlin Su, et al.- Published
- Mar 2026
- Citations
- 42
- Code
- 3.5K stars
