更多项目
项目深入

一直以来,在Transformer架构中,注意力(Attention)与前馈网络(Feed-Forward Networ…

AI 导读

北京邮电大学团队的 MiTA(Mixture-of-Top-k Attention)工作的官方 PyTorch 实现仓库。帖子内容指出,把标准注意力公式展开后可视为一种权重由输入生成、隐藏层宽度随序列长度增长的 FFN,并由此重新理解长上下文开销及线性注意力、Top-k 注意力等技术路线;MiTA 将快权重(Fast Weights)视角发展为五维分类框架,并设计可变形、可复用的 Top-k 专家。材料称该工作已被 NeurIPS 2026 接收,代码开源,并提供论文与代码链接,README 中列出同作者此前 DEPICT 等工作。

分级
值得看
是否付费
免费公开仓库
预估学习时长
约 20 分钟模型按 README 估算
收录时间
2026-11-04

热度

有热度
7 天新增 star
+9
30 天新增 star
+62
Hacker News
没有
社区讨论
1 帖

项目信息

语言
Python
许可
MIT
最近提交
2026-10-07
Star
68
Fork
4

上手方式

代码
# make sure that flash-attn==2.6.3 is installed before using MiTA
from mita import MiTA_Attention
attention_mita = MiTA_Attention(dim=384, num_heads=6)
x = torch.randn(1, 256, 384)
x = block(x)

摘自 README(新标签页打开)