Deep Thinking by Markov Chain of Continuous Thoughts

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liu, Jiayu, Huang, Zhenya, Yang, Xuan, Ji, Tianyun, Sims, Anya, Xu, Hao, Chen, Enhong, Teh, Yee Whye, Miao, Ning
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915972932173824
author Liu, Jiayu
Huang, Zhenya
Yang, Xuan
Ji, Tianyun
Sims, Anya
Xu, Hao
Chen, Enhong
Teh, Yee Whye
Miao, Ning
author_facet Liu, Jiayu
Huang, Zhenya
Yang, Xuan
Ji, Tianyun
Sims, Anya
Xu, Hao
Chen, Enhong
Teh, Yee Whye
Miao, Ning
contents Transformer-based models can perform complicated reasoning by generating reasoning paths token by token. While effective, this approach often requires generating thousands of tokens to solve a single problem, which can be slow and computationally expensive. More importantly, it involves a discrete sampling operation at the end of each time step, creating an information bottleneck across time steps. In this work, we propose MarCos, an improvement of the transformer structure that allows fully continuous reasoning at the thought level. Unlike traditional transformer layers, which focus on refining token predictions at each time step, layers in MarCos map a continuous representation of a stepwise thought to the distribution of the next thought. This enables us to achieve multi-step reasoning in a single pass of MarCos. Preliminary experimental results on synthetic and real-world math tasks show the great potential of MarCos. Notably, we observe that the increased information bandwidth of MarCos elicits the ability of parallel thinking, in contrast to single-threaded thinking in traditional transformers. Meanwhile, in real-world math tasks, MarCos achieves more than $10\times$ speedup in wall-clock time with the same level of accuracy. Our code is available at https://github.com/Ljyustc/MarCos.
format Preprint
id arxiv_https___arxiv_org_abs_2509_25020
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Deep Thinking by Markov Chain of Continuous Thoughts
Liu, Jiayu
Huang, Zhenya
Yang, Xuan
Ji, Tianyun
Sims, Anya
Xu, Hao
Chen, Enhong
Teh, Yee Whye
Miao, Ning
Machine Learning
Transformer-based models can perform complicated reasoning by generating reasoning paths token by token. While effective, this approach often requires generating thousands of tokens to solve a single problem, which can be slow and computationally expensive. More importantly, it involves a discrete sampling operation at the end of each time step, creating an information bottleneck across time steps. In this work, we propose MarCos, an improvement of the transformer structure that allows fully continuous reasoning at the thought level. Unlike traditional transformer layers, which focus on refining token predictions at each time step, layers in MarCos map a continuous representation of a stepwise thought to the distribution of the next thought. This enables us to achieve multi-step reasoning in a single pass of MarCos. Preliminary experimental results on synthetic and real-world math tasks show the great potential of MarCos. Notably, we observe that the increased information bandwidth of MarCos elicits the ability of parallel thinking, in contrast to single-threaded thinking in traditional transformers. Meanwhile, in real-world math tasks, MarCos achieves more than $10\times$ speedup in wall-clock time with the same level of accuracy. Our code is available at https://github.com/Ljyustc/MarCos.
title Deep Thinking by Markov Chain of Continuous Thoughts
topic Machine Learning
url https://arxiv.org/abs/2509.25020