arXiv Computation and Language By Masoud Ataei

Conic Certification of Transformer Attention

Read the original on arXiv Computation and Language →

The paper introduces a conic programming framework for certifying transformer attention, providing guarantees on how far an attention head’s output can shift under input perturbations. It shows that the main challenge lies in the bilinear interaction between varying queries and keys, making certification NP-hard over boxes and inapproximable when logits can shift by a constant. The authors present an exponential cone moment relaxation that yields a quadratic error bound in the perturbation radius, and demonstrate that this approach outperforms linear bound propagation on GPT‑2, achieving tighter bounds and matching predicted second‑order accuracy.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computation and Language.

arXiv Machine Learning
Sep 30

Efficiently Approximating Attention Is Hard

arXiv:2609.37261v1 Announce Type: new Abstract: Softmax attention is ubiquitous in modern machine learning, but its quadratic scaling with sequence length makes it costly. To reduce this cost, attent...

By Lukas Haverbeck, Carmen Amo Alonso, Andres Felipe Posada-Moreno, Sebastian Trimpe, Marco Pavone
arXiv Machine Learning
1d ago

Softmax Attention with Any Temperature in Fixed Dimension: Matching the Best Known Maximum-Inner-Product Bounds

arXiv:2611.06055v1 Announce Type: cross Abstract: Attention is central to modern AI models for language, vision, and scientific modeling. However, direct evaluation compares every query with every ke...

By Zixuan Huang (State Key Laboratory of Multimedia Information Processing, School of Computer Science, Peking University, Beijing, China, School of Physics, Peking University, Beijing, China), Guozhang Chen (State Key Laboratory of Multimedia Information Processing, School of Computer Science, Peking University, Beijing, China, National Engineering Research Center of Visual Technology, School of Computer Science, Peking University, Beijing, China)
arXiv Machine Learning
Jun 9

Token Sample Complexity of Attention

arXiv:2512. 10656v3 Announce Type: replace Abstract: As context windows in large language models continue to expand, it is essential to characterize how attention behaves at extreme sequence lengths.

By L\'ea Bohbot, Cyril Letrouit, Gabriel Peyr\'e, Fran\c{c}ois-Xavier Vialard
arXiv Machine Learning
Oct 8

Attention via Black-Box Vector Search

arXiv:2610.10135v1 Announce Type: cross Abstract: Sparse attention mechanisms estimate attention over $n$ tokens using a small subset of keys. Many existing approaches use maximum inner product searc...

By Stepan Zharkov, Krish Singal, Ashwin Padaki, Alexandr Andoni