paper-with-me

Papers

Hybrid Latent Reasoning via Reinforcement Learning

2025-05-24 · Zhenrui Yue, Bowen Jin, Huimin Zeng, Honglei Zhuang, Zhen Qin, Jinsung Yoon, Lanyu Shang, Jiawei Han, Dong Wang

Recent advances in large language models (LLMs) have introduced latent reasoning as a promising alternative to autoregressive reasoning. By performing internal computation with hidden states from previous steps, latent reasoning benefit from more informative features rather than sampling a discrete chain-of-thought (CoT) path. Yet latent reasoning approaches are often incompatible with LLMs, as their continuous paradigm conflicts with the discrete nature of autoregressive generation. Moreover, these methods rely on CoT traces for training and thus fail to exploit the inherent reasoning patterns of LLMs. In this work, we explore latent reasoning by leveraging the intrinsic capabilities of LLMs via reinforcement learning (RL). To this end, we introduce hybrid reasoning policy optimization (HRPO), an RL-based hybrid latent reasoning approach that (1) integrates prior hidden states into sampled tokens with a learnable gating mechanism, and (2) initializes training with predominantly token embeddings while progressively incorporating more hidden features. This design maintains LLMs' generative capabilities and incentivizes hybrid reasoning using both discrete and continuous representations. In addition, the hybrid HRPO introduces stochasticity into latent reasoning via token sampling, thereby enabling RL-based optimization without requiring CoT trajectories. Extensive evaluations across diverse benchmarks show that HRPO outperforms prior methods in both knowledge- and reasoning-intensive tasks. Furthermore, HRPO-trained LLMs remain interpretable and exhibit intriguing behaviors like cross-lingual patterns and shorter completion lengths, highlighting the potential of our RL-based approach and offer insights for future work in latent reasoning.

📄 PDF Abstract BibTeX arXiv:2505.18454

Code (1)

yueeeeeeee/hrpo 공식 구현 jax

Tasks

reinforcement-learningReinforcement LearningReinforcement Learning (RL)

Similar Papers 제목 키워드 기반

HyLaR: Hybrid Latent Reasoning with Decoupled Policy Optimization

2026-04-22 · Tao Cheng, Shi-Zhe Chen, Hao Zhang, Yixin Qin 외 arxiv

Chain-of-Thought (CoT) reasoning significantly elevates the complex problem-solving capabilities of multimodal large language models (MLLMs). However, adapting CoT to vision typically discretizes signals to fit LLM input…

Reinforcement LearningText Generation

Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoning

2025-02-05 · DiJia Su, Hanlin Zhu, Yingchen Xu, Jiantao Jiao 외

Large Language Models (LLMs) excel at reasoning and planning when trained on chainof-thought (CoT) data, where the step-by-step thought process is explicitly outlined by text tokens. However, this results in lengthy inpu…

Language ModelingLanguage ModellingMathematical Reasoning

Learning When to Stop: Adaptive Latent Reasoning via Reinforcement Learning

2025-11-26 · Alex Ning, Yen-Ling Kuo, Gabe Gomes arxiv

Latent reasoning represents a new development in Transformer language models that has shown potential in compressing reasoning lengths compared to chain-of-thought reasoning. By directly passing the information-rich prev…

Reinforcement LearningKnowledge Distillation

Nemotron 3 Ultra: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoning

2026-06-12 · NVIDIA, :, Aaron Blakeman, Aaron Thomas 외 arxiv

We introduce Nemotron 3 Ultra, a 550 billion total and 55 billion active parameter Mixture-of-Experts Hybrid Mamba-Attention language model. We pre-trained Nemotron 3 Ultra on 20 trillion text tokens, then extended the c…

Reinforcement Learning

LoopRPT: Reinforcement Pre-Training for Looped Language Models

2026-03-20 · Guo Tang, Shixin Jiang, Heng Chang, Nuo Chen 외 arxiv

Looped language models (LoopLMs) perform iterative latent computation to refine internal representations, offering a promising alternative to explicit chain-of-thought (CoT) reasoning. However, existing reinforcement lea…

Reinforcement Learning