THE RESEARCH ATLAS

FlashAttention:具备 IO 感知能力的快速、内存高效精确注意力

通过减少内存访问开销,在保持精确注意力计算的同时提高效率。

FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness

Tri Dao et al.
2022-05 · Academic · 研究论文
来源核查: 2026-10-24署名机构 · Stanford University / University at Buffalo 🇺🇸 美国
机构与地域来源

地域依据是署名机构或发布机构所在地,不代表作者国籍或完整合作名单。

arxiv.org ↗

为什么值得读

理解为什么硬件感知算法与模型架构同样重要。

阅读原始来源 ↗原始 PDF ↗

引用

@misc{dao2022flashattention,
  title = {FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness},
  author = {Tri Dao and Daniel Y. Fu and Stefano Ermon and Atri Rudra and Christopher Ré},
  year = {2022},
  eprint = {2205.14135},
  archivePrefix = {arXiv},
  primaryClass = {cs.LG},
  url = {https://arxiv.org/abs/2205.14135}
}

论文与代码版权归原作者;引用及许可请以原始来源为准。

相关研究

2012 · Academic · 研究论文署名机构 · University of Toronto 🇨🇦 加拿大

基于深度卷积神经网络的 ImageNet 分类

ImageNet Classification with Deep Convolutional Neural Networks

深度卷积网络将数据、GPU 计算与表示学习结合,为现代视觉研究提供了重要起点。

2017-06 · Google · 研究论文署名机构 · Google / University of Toronto 🇺🇸 美国 · 🇨🇦 加拿大

注意力即全部

Attention Is All You Need

Transformer 用注意力机制构建序列模型,成为后续语言与多模态模型的重要基础。

2018-10 · Google · 研究论文署名机构 · Google AI Language 🇺🇸 美国

BERT:面向语言理解的深度双向 Transformer 预训练

BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

通过掩码语言预训练学习双向表示,再迁移到语言理解任务。