English

Higher-order Linear Attention

Machine Learning 2026-05-15 v3 Artificial Intelligence Computation and Language

Abstract

The quadratic cost of scaled dot-product attention is a central obstacle to scaling autoregressive language models to long contexts. Linear-time attention and State Space Models (SSMs) provide scalable alternatives but are typically restricted to first-order or kernel-based approximations, which can limit expressivity. We introduce Higher-order Linear Attention (HLA), a causal, streaming mechanism that realizes higher interactions via compact prefix sufficient statistics. In the second-order case, HLA maintains a constant-size state and computes per-token outputs in linear time without materializing any n×nn \times n matrices. We give closed-form streaming identities, a strictly causal masked variant using two additional summaries, and a chunk-parallel training scheme based on associative scans that reproduces the activations of a serial recurrence exactly. We further outline extensions to third and higher orders. Collectively, these results position HLA as a principled, scalable building block that combines attention-like, data-dependent mixing with the efficiency of modern recurrent architectures.

Keywords

Cite

@article{arxiv.2510.27258,
  title  = {Higher-order Linear Attention},
  author = {Yifan Zhang and Zhen Qin and Mengdi Wang and Quanquan Gu},
  journal= {arXiv preprint arXiv:2510.27258},
  year   = {2026}
}

Comments

Project Page: https://github.com/yifanzhang-pro/HLA

R2 v1 2026-07-01T07:15:14.850Z