註冊並分享邀請連結,可獲得影片播放與邀請獎勵。

Nanbeige
@nanbeige
Nanbeige LLM Lab, BOSS Zhipin (Kanzhun Limited)
加入 December 2023
436 正在關注    5.7K 粉絲
We introduce Depth Hybrid Attention (DHA), which incorporates depth-wise value mixing into hybrid architectures with very little additional compute and cache overhead. By allowing layers to mix information across depth, DHA broadens each layer’s view and increases the model’s expressive capacity. Our DHA method adapts to multiple hybrid architectures, including Depth Window Attention (DWA) for GQA + SWA and Depth Delta Attention (DDA) for MLA + KDA. Across loss curves, short-context benchmark results, and long-context retrieval evaluations, both variants outperform their corresponding full-attention and hybrid-attention baselines, while retaining hybrid attention’s low KV-cache footprint. Project overview & code:
顯示更多
0
11
218
31
轉發到社區