Latent Thoughts, Certain Gains: Bayesian Optimization for LLM Reasoning

Xin Xiao, Jiang Zhong


Abstract
Recent advances in Large Language Models (LLMs) have shifted toward latent reasoning, where intermediate processes are represented as continuous vectors rather than text tokens. While this paradigm reduces inference cost, existing approaches often suffer from instability on complex, out-of-distribution tasks where robust reasoning is critical. To address this, we propose Bayesian Latent Thought Optimization (BLTO), a training-free framework that enhances frozen LLMs entirely at test time. BLTO treats latent thought tokens as stochastic variables and formulates reasoning as a Bayesian inference problem. We introduce a variational posterior over latent vectors and optimize it by maximizing a reward-weighted objective that balances reasoning quality with KL regularization toward a standard Gaussian prior. The reward is derived from the model’s predictive consistency via output entropy, enabling fully self-supervised optimization without external supervision. BLTO employs reparameterized sampling and reward-weighted updates within a variational inference framework, together with an adaptive controller that dynamically determines inference steps based on reward improvement and uncertainty convergence. Extensive experiments on challenging reasoning benchmarks demonstrate that BLTO consistently improves frozen LLM performance.
Anthology ID:
2026.emnlp-main.1128
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
25844–25853
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1128/
DOI:
Bibkey:
Cite (ACL):
Xin Xiao and Jiang Zhong. 2026. Latent Thoughts, Certain Gains: Bayesian Optimization for LLM Reasoning. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 25844–25853, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Latent Thoughts, Certain Gains: Bayesian Optimization for LLM Reasoning (Xiao & Zhong, EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1128.pdf