Skip to content
AI News HubLIVE
Original source2 min read

Toeplitz MLP Mixers are Low Complexity, Information-Rich Sequence Models

Summary

A new architecture called Toeplitz MLP Mixer (TMM) replaces attention with triangular-masked Toeplitz matrix multiplication, achieving sub-quadratic complexity (O(d n log n) time, O(d n) space) while retaining more input information and improving copying ability, information retrieval, and in-context learning accuracy over comparable architectures.

SourcearXiv Machine LearningAuthor: Benjamin L. Badger, Ethan Roland
Toeplitz MLP Mixers are Low Complexity, Information-Rich Sequence Models
Report an error

The correction channel is not available yet. You can copy the article reference below for later.

Correction instructions
Read article

[2605.06683] Toeplitz MLP Mixers are Low Complexity, Information-Rich Sequence Models

[Submitted on 24 Apr 2026]

Title:Toeplitz MLP Mixers are Low Complexity, Information-Rich Sequence Models

View a PDF of the paper titled Toeplitz MLP Mixers are Low Complexity, Information-Rich Sequence Models, by Benjamin L. Badger and Ethan Roland

View PDF HTML (experimental)

Abstract:Transformer-based large language models are in some respects limited by the quadratic time and space computational complexity of attention. We introduce the Toeplitz MLP Mixer (TMM), a transformer-like architecture that swaps attention for triangular-masked Toeplitz matrix multiplication over the sequence dimension resulting in $\mathcal{O} (dn \log n)$ time and $\mathcal O(dn)$ space complexity during training and $\mathcal O(dn)$ time and space at inference prefill. Despite the lack of sophisticated input modulation or state maintenance present in other sub-quadratic architectures, TMMs yield greater training efficiency in terms of loss achieved per compute and device memory. We demonstrate that TMMs are capable of retaining more input information resulting in improved copying ability, which we argue results from a lack of architectural biases. Consistent with higher input information retention, TMMs exhibit superior information retrieval and in-context learning benchmark accuracy compared to comparable architectures. We conclude with an analysis from the perspective of operator index theory and show that, counterintuitively, trained Toeplitz layers of causal non-invertible models are more likely to be invertible or nearly so than models that are actually invertible over their inputs.

Subjects:

Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computation and Language (cs.CL)

Cite as: arXiv:2605.06683 [cs.LG]

(or arXiv:2605.06683v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2605.06683

arXiv-issued DOI via DataCite

Submission history

From: Benjamin Badger [view email] [v1] Fri, 24 Apr 2026 20:37:21 UTC (11,702 KB)

Full-text links:

Access Paper:

View a PDF of the paper titled Toeplitz MLP Mixers are Low Complexity, Information-Rich Sequence Models, by Benjamin L. Badger and Ethan Roland

View PDF

HTML (experimental)

TeX Source

view license

Current browse context:

cs.LG

new | recent | 2026-05

Change to browse by:

cs cs.AI cs.CL

References & Citations

NASA ADS

Google Scholar

Semantic Scholar

Loading...

Data provided by:

Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle

Bibliographic Explorer (What is the Explorer?)

Connected Papers Toggle

Connected Papers (What is Connected Papers?)

Litmaps Toggle

Litmaps (What is Litmaps?)

scite.ai Toggle

scite Smart Citations (What are Smart Citations?)

Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle

alphaXiv (What is alphaXiv?)

Links to Code Toggle

CatalyzeX Code Finder for Papers (What is CatalyzeX?)

DagsHub Toggle

DagsHub (What is DagsHub?)

GotitPub Toggle

Gotit.pub (What is GotitPub?)

Huggingface Toggle

Hugging Face (What is Huggingface?)

ScienceCast Toggle

ScienceCast (What is ScienceCast?)

Demos

Demos

Replicate Toggle

Replicate (What is Replicate?)

Spaces Toggle

Hugging Face Spaces (What is Spaces?)

Spaces Toggle

TXYZ.AI (What is TXYZ.AI?)

Related Papers

Recommenders and Search Tools

Link to Influence Flower

Influence Flower (What are Influence Flowers?)

Core recommender toggle

CORE Recommender (What is CORE?)

IArxiv recommender toggle

IArxiv Recommender (What is IArxiv?)

Author

Venue

Institution

Topic

About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

Which authors of this paper are endorsers? | Disable MathJax (What is MathJax?)

Key points and analysis

Article intelligence

EngineersAdvanced

Key points

  • TMM uses Toeplitz matrices instead of attention, reducing complexity while maintaining performance.
  • It retains more input information, leading to better copying and in-context learning.
  • TMM achieves greater training efficiency per compute and memory.
  • Analysis with operator index theory suggests trained Toeplitz layers are nearly invertible.

Highlights and analysis are generated automatically and may contain errors. Check the original source.