mlsanityData to 14 Oct 2026

RiLM: Parameter-Efficient Language Modeling via Geodesic Decoding

arXiv 2609.10305 ↗First posted 9 Sep 2026cs.CL cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Language models under one million parameters matter for edge deployment, domain adaptation, and reproducible research, yet a two-layer LSTM or Transformer at embedding width d = 128 still spends roughly one third of its capacity on the output matrix W_out in R^(d x |V|). We propose Riemannian Language Models (RiLM), which remove that layer entirely: context unfolds as a trajectory on a Riemannian manifold, and next-token probabilities arise from squared geodesic distance between the current state and vocabulary embeddings. The same embedding map serves input and output -- decoding is geometry. We instantiate the framework on flat R^d (Flat RiLM) and the Poincare ball H^d (HypRiLM) with a shared MLP composition map phi (~290k parameters, d = 128, |V| = 2000). Across five seeds on WikiText-2, HypRiLM reaches 54.2 +/- 0.2 validation perplexity versus 87.6 +/- 0.6 for Flat RiLM; tied and matched LSTM, Transformer, and SSM controls remain at 113-147 PPL on WT-2 -- HypRiLM leads by roughly 2x over the strongest tied recurrent baseline (SSM, 113.0 +/- 3.8). Penn Treebank and a 10k-vocabulary stress test confirm that geodesic decoding transfers across corpora and larger |V|, while hyperbolic curvature helps selectively. We also characterize boundary collapse in naive hyperbolic recurrence and show how Mobius stabilization restores trainability. Claims are scoped to controlled small-model comparisons, not full-vocabulary state of the art.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 10 of its 11 references that are in the map. First posted on arXiv 9 Sep 2026.

Builds on

10 of 11 references.

References · 10

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  4. 2018Neural Ordinary Differential Equationscited by 3,779
  5. 2015Variational Inference with Normalizing Flowscited by 2,170
  6. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
  7. 2017Poincar'e Embeddings for Learning Hierarchical Representationscited by 684
  8. 2018Hyperbolic Neural Networkscited by 402
  9. 2016Using the Output Embedding to Improve Language Modelscited by 272
  10. 2017Regularizing and Optimizing LSTM Language Modelscited by 205

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025HELM: Hyperbolic Large Language Models via Mixture-of-Curvature Experts
  2. 2025Shared Global and Local Geometry of Language Model Embeddings
  3. 2026Geometric Properties of the Voronoi Tessellation in Latent Semantic Manifolds of Large Language Models
  4. 2026OrBIT: Structure-Guided Embedding Compression
  5. 2026SIGMA: Scalable Spectral Insights for LLM Model Collapse
  6. 2026A Geometric Perspective on Next-Token Prediction in Large Language Models: Three Emerging Phases
  7. 2026On the Geometric Structure of Layer Updates in Deep Language Models
  8. 2026A Single-Layer Model Can Do Language Modeling