AITP
精选全部 AI 动态AI 日报Agent 接入关于更新日志信源提报反馈
登录 / 注册
AITOP

Llama-3B

共 1 条相关 AI 资讯
6月18日
10:57
10:57官方账号arXiv cs.LG@Amiri Hayes, Belinda Li, Jacob Andreas
研究者提出用程序合成方法反向工程Transformer注意力头。他们先计算注意力矩阵,再让预训练语言模型生成Python程序来重现注意力模式。在GPT-2、TinyLlama-1.1B和Llama-3B上,不到1000个程序实现了平均IoU>75%。替换25%的注意力头仅导致16%的困惑度增加,并在下游问答基准上保持性能。
论文GPT-2TinyLlamaLlama-3B可解释性注意力机制

推荐理由:这篇论文用Python程序解释了注意力头怎么工作,还能直接用程序替换掉原始头,精度很高,想看模型内部机制的可以读。
原文
精选全部日报登录