Attention Is All We Had — But Not What We Needed: Convergent State Machine for Iterative Energy-Based Language Generation

We introduce the Convergent State Machine (CSM), a novel architecture for language generation that replaces attention entirely with energy-based iterative state refinement. Three models trained (66M, 150M, 331M), all with zero attention layers. CSM 150M matches GPT-2 1.5B on MMLU within 0.4%, using 10x fewer parameters and 13x less data. Key finding: hard problems show 60% perplexity improvement with more iterations while easy problems degrade 60%. The model reasons deeper on harder problems — not repeated computation, but genuine difficulty-dependent reasoning. Iteration scaling confirmed across three model sizes: 66M converges at iter 15, 150M at iter 30, 331M at iter 40. Total training compute: under $100 of A100 GPU time.

Authors

Institutions

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-05-29
DOI
https://doi.org/10.5281/zenodo.20442828
Primary Topic
Ferroelectric and Negative Capacitance Devices
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Attention Is All We Had — But Not What We Needed: Convergent State Machine for Iterative Energy-Based Language Generation

Arunesh Dwivedi
Zenodo (CERN European Organization for Nuclear Research)
Ferroelectric and Negative Capacitance Devices
preprint

Attention Is All We Had — But Not What We Needed: Convergent State Machine for Iterative Energy-Based Language Generation

Arunesh Dwivedi
preprint en

Abstract

We introduce the Convergent State Machine (CSM), a novel architecture for language generation that replaces attention entirely with energy-based iterative state refinement. Three models trained (66M, 150M, 331M), all with zero attention layers. CSM 150M matches GPT-2 1.5B on MMLU within 0.4%, using 10x fewer parameters and 13x less data. Key finding: hard problems show 60% perplexity improvement with more iterations while easy problems degrade 60%. The model reasons deeper on harder problems — not repeated computation, but genuine difficulty-dependent reasoning. Iteration scaling confirmed across three model sizes: 66M converges at iter 15, 150M at iter 30, 331M at iter 40. Total training compute: under $100 of A100 GPU time.

Zenodo (CERN European Organization for Nuclear Research)
Ministry of Infrastructure and Water Management (NL)
Affordable and clean energy
Ferroelectric and Negative Capacitance Devices
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.