Kimi Team proposes Attention Residuals, replacing fixed residual accumulation with learned attention

burny_tech · x · 2026-10-27

Kimi Team's new paper Attention Residuals argues that standard PreNorm residual connections aggregate all layer outputs with fixed unit weights, causing uncontrolled hidden-state growth that dilutes each layer's contribution. AttnRes instead lets each layer selectively aggregate earlier representations via softmax attention with learned, input-dependent weights. To cut memory/communication overhead at scale, Block AttnRes partitions layers into blocks and attends over block-level representations, working as a practical drop-in replacement. Scaling-law runs show consistent gains across sizes. Developer burnytech reports adding AttnRes to the General Physics Transformer: a reproduced table shows Block AttnRes cutting radiative-layer 12-step NMSE from 0.39277 to 0.28917, with faster loss convergence.

Original post →

More from Models

Models channel →