Accelerate Parallelizable Reasoning via Parallel Decoding within One Sequence

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autore principale: Yu, Yijiong
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911121295802368
author Yu, Yijiong
author_facet Yu, Yijiong
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2503_20533
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Accelerate Parallelizable Reasoning via Parallel Decoding within One Sequence
Yu, Yijiong
Computation and Language
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.
title Accelerate Parallelizable Reasoning via Parallel Decoding within One Sequence
topic Computation and Language
url https://arxiv.org/abs/2503.20533