arXiv Computation and Language By Joshua Nunley

TANGO: Token-Aggregated Nonlinear Gating Operators for Natural and Formal Language Modeling

Read the original on arXiv Computation and Language →

The paper introduces TANGO, a new Transformer variant that replaces the standard self‑attention and feed‑forward sublayers with a single cross‑token gated residual update. Each token generates a SwiGLU gate, and query‑key similarities weight these gates to rescale destination features, yielding a quadratic‑time model. A windowed variant, WANGO, limits the gated interactions to a recent window and uses prefix statistics for older tokens, achieving linear‑time complexity while maintaining competitive performance.

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
1d ago

TANGO: Treating Tokens as Operators

The paper introduces TANGO, a Token‑Aggregated Nonlinear Gating Operator that blends cross‑token mixing and token‑wise transformation in transformer architectures. By computing a nonlinear gate per source token and averaging these gates for each destination, TANGO forms a source‑conditioned linear operator that improves predictive performance. Experiments on web text, formal mathematics, and code show that full‑prefix TANGO achieves the lowest test negative log‑likelihood across 16 settings, while a narrower variant offers substantial throughput gains with only a modest increase in loss.

By Joshua Nunley
arXiv AI
Jun 4

MesaNet: Sequence Modeling by Locally Optimal Test-Time Training

arXiv:2506. 05233v2 Announce Type: replace-cross Abstract: Sequence modeling is currently dominated by causal transformer architectures that use softmax self-attention.

By Johannes von Oswald, Nino Scherrer, Seijin Kobayashi, Luca Versari, Songlin Yang, Sarthak Mittal, Maximilian Schlegel, Kaitlin Maile, Yanick Schimpf, Oliver Sieberling, Alexander Meulemans, Rif A. Saurous, Guillaume Lajoie, Charlotte Frenkel, Razvan Pascanu, Blaise Ag\"uera y Arcas, Jo\~ao Sacramento