English

JEPA-Reasoner: Decoupling Latent Reasoning from Token Generation

Computation and Language 2026-01-29 v3

Abstract

Current autoregressive language models couple high-level reasoning and low-level token generation into a single sequential process, making the reasoning trajectory vulnerable to compounding expression errors. We propose JEPA-Reasoner, a novel architectural paradigm that decouples these tasks using a Joint-Embedding Predictive Architecture (JEPA) for pure latent-space reasoning and a separate Talker module for linguistic reconstruction. By isolating the reasoning engine from the discrete token-sampling process, our architecture enables: (1) Error Containment, where token-level failures cannot propagate into the latent reasoning chain; (2) Continuous Guidance, providing the generator with access to the entire lossless reasoning trajectory; and (3) Representation of Uncertainty, allowing the model to maintain multiple hypotheses via mixed latent vectors. Controlled experiments on synthetic and natural language tasks demonstrate that this decoupling enables a 0.9B model to achieve a 149.5\% improvement in 8-shot GSM8K accuracy over a coupled Transformer baseline trained on identical data. This work shifts the focus from scaling coupled models to investigating decoupled architectures as a more robust foundation for complex reasoning.

Keywords

Cite

@article{arxiv.2512.19171,
  title  = {JEPA-Reasoner: Decoupling Latent Reasoning from Token Generation},
  author = {Bingyang Kelvin Liu and Ziyu Patrick Chen and David P. Woodruff},
  journal= {arXiv preprint arXiv:2512.19171},
  year   = {2026}
}
R2 v1 2026-07-01T08:36:29.287Z