KOLM-Alpha β€” Fully-Oscillatory Kuramoto Language Model

Delon Swartz β€” AI Researcher | Engineer

KOLM-Alpha is, to our knowledge, the first language model with no transformer computation anywhere β€” both routing between tokens and processing within them are performed by networks of coupled Kuramoto oscillators. It is the successor to KOLM-Hybrid-1, which kept standard attention and made only the feed-forward slot oscillatory. KOLM-Alpha replaces attention itself with synchronization routing.

Naming note: the name KOLM-Alpha previously labelled the hybrid twin study, now published as KOLM-Hybrid-1. Under the project naming registry (2026-07), KOLM-Alpha denotes this fully-oscillatory line (working name "KOLM-True").

The mechanism: KuramotoRouter A

Each layer's routing is a competitive synchronization process, not static attention:

  • Softmax competition over the causal past β€” sharp selection, not blurry averaging.
  • RoPE on the routing scores β€” position is in the routing, not just the moved content.
  • Dynamic query modulation β€” routing weights are recomputed at every settling step, modulated by the evolving oscillator state. This is genuinely dynamic routing that static attention cannot express.

The routed value drives each token's oscillators, which settle via Kuramoto dynamics under trained couplings, natural rotations, and (optionally) frustrated (phase-offset) coupling. Normalization is SphereNorm. Every block is an exact identity at initialization.

The result

Under the same controlled protocol as KOLM-Hybrid-1 (identical tokenizer, data, order, budget, seed; 11.26M TinyStories tokens; single laptop):

Model Params Val loss Perplexity
KOLM-Alpha (Router A) 17.06M 2.3170 10.14
KOLM-Hybrid-1 (attention + osc-FFN) 16.92M 2.6946 14.80
Transformer twin (TMT) 17.54M 2.7188 15.16

KOLM-Alpha wins decisively β€” 0.38 nats below the hybrid, 0.40 below the transformer twin β€” while doing zero transformer computation. It crossed the hybrid's final score at 5M tokens (41% through training).

An honest negative precedes this win: Router v1 (sigmoid-gated, position-blind, routing frozen across settle steps) lost at 3.0273. The three diagnosed failures motivated Router A's softmax + RoPE + dynamic modulation.

Scaling to 42M from scratch (the flagship)

KOLM-Alpha-42M ("W42") is the same architecture at 16 layers, d=384, H=320, frustrated coupling, randomized settling depth K ~ U[1,4], trained from scratch on 188M tokens (32k vocabulary, TinyStories + Simple English Wikipedia):

  • Val loss 1.7141 / perplexity 5.55 β€” from scratch, matching function-preserving-transfer-warmed comparables (K-sweep K1 1.7162 / K2 1.6848 / K4 1.7044).
  • Chat, via staged SFT: broad chat mixture β†’ single-turn discipline (Alpaca). The staged order beats Alpaca-from-base by 0.35 nats (alpaca-val 3.19 vs 3.55). Turn discipline is clean (answers once, stops at <|end|>); simple facts land ("The capital of France is Paris"). Entity binding and counting remain beyond 42M β€” capacity limits, not recipe bugs.

Why it matters

The hybrid showed oscillators can process as well as an MLP. KOLM-Alpha shows they can route better than attention β€” the whole layer is now synchronization dynamics, with a settling-depth compute dial that adds a memory-free axis of scale. Randomized-depth training makes that dial extrapolate past its trained depth.

Files (weights to follow)

File Purpose
kolm_true.py full-oscillator model + trainer (Router A)
kuramoto_torch.py Kuramoto block (parity-tested to 2.2e-16 vs NumPy)
chat_true.py REPL / sampler with the /think K settling dial
native_true_A.pt KOLM-Alpha 17M weights (Router A)
native_w42.pt KOLM-Alpha-42M pretrain weights
native_w42chat_alp.pt 42M chat model (staged SFT)
tiny32k.json 32k tokenizer

Roadmap

  • KOLM-Beta-T (repo) β€” transplanting pretrained transformer weights into this architecture to reach billion-parameter capability without pretraining.
  • Larger from-scratch KOLM-Alpha rungs exposing the settling dial as a user-facing "think-harder" control.

Status & limitations

Research preview. Headline comparisons are single-seed; the 17M Router-A run used a settling/lr protocol that differs slightly from the twin runs (a matched rerun is queued before external citation). Corpora are easy and all budgets are far below Chinchilla-optimal; factual recall at 42M is limited.

License

Apache-2.0.

Acknowledgements

Builds on the Kuramoto-oscillator formulation of AKOrN (Miyato et al., ICLR 2025), the frustrated-synchronization principle of FSN (arXiv:2606.18694), and the TinyStories corpus (Eldan & Li, 2023).

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Paper for DelonSwartz/KOLM-Alpha