MHLA

用 token 级多头,让线性注意力恢复全量表达力

开源 🏷️完全免费 ✅ 目前可打开 图像生成 🍴 复刻 6 🐛 待处理 8 👀 关注 1
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 160
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,提供在线项目页面,无收费信息,完全免费。

项目预览

MHLA 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队DAGroup-PKU
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,提供在线项目页面,无收费信息,完全免费。
访问状态✅ 目前可打开
是否开源是
开源协议MIT
主要语言Python
技术栈/模型dit,image-classification,image-generation,imagenet,language-modeling,linear-attention,sana,video-generation
GitHub 星标★ 160
上线时间2025-12-28 00:00:00
代码更新2026-05-19
仓库体积145.1 MB
本站数据更新2026-11-11
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

MHLA 是一个面向图像生成与语言建模的线性注意力机制改进项目,旨在解决线性注意力在表达力上的不足。传统线性注意力通过核函数近似 softmax 注意力,虽降低了计算复杂度,但牺牲了模型对复杂 token 关系的建模能力。MHLA 提出在 token 级别引入多头机制,将每个 token 的注意力计算拆分为多个子空间,从而在不显著增加计算成本的前提下恢复线性注意力的表达力。项目基于 PyTorch 实现,支持图像分类(ImageNet)、图像生成(如 Sana 架构)、视频生成和语言建模等任务,提供了与 DiT 等主流架构的集成示例。核心能力包括:高效的线性注意力实现、token 级多头分解策略、以及跨模态任务的验证。该项目为 ICLR 2026 投稿,代码已开源,适合研究高效注意力机制和长序列建模的开发者。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 160

核心亮点

  • 提出 token 级多头分解,直接提升线性注意力对细粒度关系的建模能力
  • 支持图像生成、分类、视频、语言多任务,验证场景广泛
  • 基于 Sana/DiT 集成,易于在现有扩散模型上替换使用

不足之处

  • 早期项目,文档和社区支持待观察
  • 未提供与标准 softmax 注意力的详细性能对比基准

适用场景

  • 长序列图像生成(如高分辨率扩散模型)
  • 视频生成中的时空注意力建模
  • 高效语言模型的长上下文推理

替代项目

FlashAttention、Linear Attention (Linformer)、Mamba

上一篇:WaDi

下一篇:flow-matching

同类项目推荐

LLMNodeBed 开源

统一基准,对比LLM节点分类算法,复现ICML论文结果。

[ICML 2025] Official implementation for paper "A Comprehensive Analysis on LLM-based···

★ 76 2026-08-09
modelkit 开源

让 ML 模型开发到上线更省心

Toolkit for developing and maintaining ML models

★ 149 2026-08-09
EvoControl 开源

让AI自己优化代码,还能按你的要求来

Official implementation of Controlled Self-Evolution for Algorithmic Code Optimizati···

★ 135 2026-08-09
verbalized-sampling 开源

让LLM输出带概率的多样答案,告别重复文本

Verbalized Sampling, a training-free prompting strategy to mitigate mode collapse in···

★ 824 2026-08-09
create-llm 开源

一条命令生成可训练的 PyTorch 大模型工程

The fastest way to build and start training your own LLM. CLI tool that scaffolds pr···

★ 373 2026-09-16
autodistill 开源

用基础模型自动标注,无人工标注训练视觉模型

Images to inference with no labeling (use foundation models to train supervised mode···

★ 2798 2026-08-09
Scaling-Diffusion-Transformers-m··· 开源

让扩散模型从小规模到大规模训练,超参数无需重调,省时省力。

[NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers···

★ 101 2026-08-09
RPG 开源

给大模型 RL 训练加上安全护栏,稳定提升性能

[ICLR 2026] RPG: KL-Regularized Policy Gradient (https://arxiv.org/abs/2505.17508)

★ 78 2026-08-09