English

Restoring Exploration after Post-Training: Latent Exploration Decoding for Large Reasoning Models

Computation and Language 2026-05-12 v3 Machine Learning

Abstract

Large Reasoning Models (LRMs) have recently achieved strong mathematical and code reasoning performance through Reinforcement Learning (RL) post-training. However, we show that modern reasoning post-training induces an unintended exploration collapse: temperature-based sampling no longer increases pass@nn accuracy. Empirically, the final-layer posterior of post-trained LRMs exhibit sharply reduced entropy, while the entropy of intermediate layers remains relatively high. Motivated by this entropy asymmetry, we propose Latent Exploration Decoding (LED), a depth-conditioned decoding strategy. LED aggregates intermediate posteriors via cumulative sum and selects depth configurations with maximal entropy as exploration candidates. Without additional training or parameters, LED consistently improves pass@1 and pass@16 accuracy by 0.61 and 1.03 percentage points across multiple reasoning benchmarks and models. Furthermore, integrating LED into reinforcement learning, e.g., using GRPO as the rollout strategy, yields faster reward improvement and higher final performance, due to the efficient exploration capability of LED. Project page: https://github.com/AlbertTan404/LED.

Keywords

Cite

@article{arxiv.2602.01698,
  title  = {Restoring Exploration after Post-Training: Latent Exploration Decoding for Large Reasoning Models},
  author = {Wenhui Tan and Fiorenzo Parascandolo and Enver Sangineto and Jianzhong Ju and Zhenbo Luo and Qian Cao and Rita Cucchiara and Ruihua Song and Jian Luan},
  journal= {arXiv preprint arXiv:2602.01698},
  year   = {2026}
}

Comments

Project Page: https://github.com/AlbertTan404/LED

R2 v1 2026-07-01T09:31:01.704Z