返回全部动态

非对称注意力头:Transformer注意力的结构化头级上下文分配

原标题:Asymmetric Attention Heads: Structured Head-Wise Context Allocation for Transformer Attention

arXiv cs.CL一手来源研究质量 83

AI 摘要

本文提出非对称注意力头(AAH)框架,允许Transformer各头或头组使用不同长度的因果上下文窗口,同时保持标准多头注意力输出接口。在4096 token的实验中,多种AAH局部分配变体取得了比纯全注意力更低的验证损失。作者强调AAH主要作为质量与可解释性的结构化上下文分配机制,当前实现未证明GPU FLOPs减少。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

Asymmetric Attention Heads: Structured Head-Wise Context Allocation for Transformer Attention Abstract Standard multi-head attention (MHA) gives every head the same full causal context span, although heads can serve different contextual roles. Some heads may rely mainly on nearby lexical or syntactic context, while others may depend on longer-range relations such as entity interactions, discourse links, or state changes. We present Asymmetric Attention Heads (AAH), a head-wise context-allocation


发布时间:2026-08-21 12:00
抓取时间:2026-08-21 12:00
来源机构:arXiv
阅读原文arxiv.org