English

Semantic Tube Prediction: Beating LLM Data Efficiency with JEPA

Machine Learning 2026-02-27 v1

Abstract

Large Language Models (LLMs) obey consistent scaling laws -- empirical power-law fits that predict how loss decreases with compute, data, and parameters. While predictive, these laws are descriptive rather than prescriptive: they characterize typical training, not optimal training. Surprisingly few works have successfully challenged the data-efficiency bounds implied by these laws -- which is our primary focus. To that end, we introduce the Geodesic Hypothesis, positing that token sequences trace geodesics on a smooth semantic manifold and are therefore locally linear. Building on this principle, we propose a novel Semantic Tube Prediction (STP) task, a JEPA-style regularizer that confines hidden-state trajectories to a tubular neighborhood of the geodesic. STP generalizes JEPA to language without requiring explicit multi-view augmentations. We show this constraint improves signal-to-noise ratio, and consequently preserves diversity by preventing trajectory collisions during inference. Empirically, STP allows LLMs to match baseline accuracy with 16×\times less training data on the NL-RX-SYNTH dataset, directly violating the data term of Chinchilla-style scaling laws and demonstrating that principled geometric priors can surpass brute-force scaling. Code is available at https://github.com/galilai-group/llm-jepa#stp.

Keywords

Cite

@article{arxiv.2602.22617,
  title  = {Semantic Tube Prediction: Beating LLM Data Efficiency with JEPA},
  author = {Hai Huang and Yann LeCun and Randall Balestriero},
  journal= {arXiv preprint arXiv:2602.22617},
  year   = {2026}
}

Comments

21 pages, 13 figures

R2 v1 2026-07-01T10:53:18.618Z