arXiv · 2503.20533
Accelerate Parallelizable Reasoning via Parallel Decoding within One Sequence
Abstract
Recent advances in reasoning models have demonstrated significant improvements in accuracy by employing detailed and comprehensive reasoning processes. However, generating these lengthy reasoning sequences is computationally expensive and time-consuming. To address this inefficiency, we leverage the inherent parallelizability of certain tasks to accelerate the reasoning process. Specifically, when multiple parallel reasoning steps exist, we decode multiple tokens per forward pass via a tree-like attention mask within a single sequence, avoiding additional memory usage. Experimental results show that our method achieves up to nearly 100\% speedup in decoding while basically maintaining the answer quality.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Yijiong Yu. 2025-03-26. Accelerate Parallelizable Reasoning via Parallel Decoding within One Sequence. https://arxiv.org/abs/2503.20533
Cite the original work for its findings. Save a collection to share your selection of sources.