AI RESEARCH

Adaptive Head Budgeting for Efficient Multi-Head Attention

arXiv CS.LG

ArXi:2604.22583v1 Announce Type: new Transformers have become the dominant architecture across a wide range of domains, largely due to the effectiveness of multi-head attention in capturing diverse representation subspaces. However, standard multi-head attention activates all heads uniformly for every input, regardless of task requirements or input complexity. In many scenarios, particularly for coarse-grained tasks such as text classification, the relevant information is often global and does not require the full diversity of attention heads.