Attention-Sink

看清注意力汇聚如何产生,优化你的语言模型训练与推理。

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 5 🐛 待处理 1 👀 关注 4
GitHub ↗
GitHub 星标 ★ 164
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,需自行部署使用。

项目预览

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

项目数据

分类开源模型
开发团队sail-sg
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,需自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型attention-mechanism,attention-sink,language-model,large-language-models
GitHub 星标★ 164
上线时间2024-10-13 00:00:00
代码更新2025-07-08
仓库体积6.2 MB
本站数据更新2026-11-11
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

Attention-Sink 是 ICLR 2025 的 Spotlight 论文官方实现,从实证角度研究语言模型中注意力汇聚(attention sink)现象的出现机制。该项目通过系列实验揭示,在训练过程中,某些特殊 token(如句首标记)会吸引大量注意力权重,形成所谓的'注意力汇聚',并分析其与模型性能、训练动态的关系。核心能力包括:提供可复现的实验代码、预训练模型检查点、以及用于分析注意力分布的可视化工具。它帮助研究者深入理解 transformer 内部运作原理,为优化模型架构(如流式生成、长上下文处理)提供理论依据。项目代码基于 PyTorch,支持自定义实验配置,适合学术研究与模型调试。

Star 增长趋势

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

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

核心亮点

  • 提供 ICLR 2025 官方实证代码,可复现论文关键实验
  • 包含注意力分布可视化工具,直观展示汇聚现象
  • 基于 PyTorch,易于扩展和自定义实验配置

不足之处

  • 文档/社区待观察
  • 主要面向研究场景,对工程应用支持有限

适用场景

  • 学术研究:分析注意力机制内部行为
  • 模型优化:调整训练策略以利用或抑制注意力汇聚
  • 教学演示:展示 transformer 训练动态

替代项目

FlashAttention、Longformer、Sparse Transformer

上一篇:DeSTA2

下一篇:LLaVA-Mini

同类项目推荐

HPT 开源

用预训练Transformer统一异构机器人策略,提升泛化与训练效率

Heterogeneous Pre-trained Transformer (HPT) as Scalable Policy Learner.

★ 547 2026-08-09
LLaVAR 开源

让AI看懂图片里的文字,并和你对话

Code/Data for the paper: "LLaVAR: Enhanced Visual Instruction Tuning for Text-Rich I···

★ 269 2026-08-09
gemma 开源

谷歌开放权重模型,本地单卡即可跑推理和微调

Gemma open-weight LLM library, from Google DeepMind

★ 5839 2026-09-16
Janus 开源

一个模型搞定看图说话和画图,多模态全能选手

Janus-Series: Unified Multimodal Understanding and Generation Models

★ 17761 2026-08-09
moment 开源

开箱即用的时间序列基础模型,零样本预测与微调一把梭

MOMENT: A Family of Open Time-series Foundation Models, ICML'24

★ 870 2026-08-09
awesome-vla-for-ad 开源

一榜看尽VLA自动驾驶研究脉络,快速入门与追踪前沿

Vision-Language-Action Models for Autonomous Driving: Past, Present, and Future

★ 524 2026-08-09
General-Physics-Transformer 开源

一个模型搞定多种物理系统模拟,告别重复建模

A foundation model to learn multiple physical systems at once

★ 98 2026-08-09