LaneRoPE: Positional Encoding for Collaborative Parallel Reasoning and Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cesa, Gabriele, Hehn, Thomas, Torres-Camps, Aleix, Casellas, Àlex Batlle, Ros-Giralt, Jordi, Behboodi, Arash, Orekondy, Tribhuvanesh
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914606584168448
author Cesa, Gabriele
Hehn, Thomas
Torres-Camps, Aleix
Casellas, Àlex Batlle
Ros-Giralt, Jordi
Behboodi, Arash
Orekondy, Tribhuvanesh
author_facet Cesa, Gabriele
Hehn, Thomas
Torres-Camps, Aleix
Casellas, Àlex Batlle
Ros-Giralt, Jordi
Behboodi, Arash
Orekondy, Tribhuvanesh
contents Parallel LLM test-time scaling techniques (e.g., best-of-$N$) require drawing $N>1$ sequences conditioned on the same input prompt. These methods boost accuracy while exploiting the computational efficiency of batching $N$ generations. However, each sequence in the batch is traditionally generated independently and hence does not reuse intermediate generations, computations, or observations from other sequences. In this paper, we propose LaneRoPE to enable coordination and collaboration among $N>1$ sequences at generation time. LaneRoPE involves two key ideas: (a) an inter-sequence attention mask to make sampling of sequences dependent on one another; and (b) a RoPE extension that injects positional information that captures relative positions between tokens, both within and outside a particular sequence. We evaluate our approach on mathematical reasoning tasks and find promising results: LaneRoPE enables collaboration among sequences, yielding additional accuracy gains under limited generated sequence length. Importantly, since LaneRoPE enables coordination with minimal changes to the underlying LLM architecture and introduces a negligible overhead at inference time, it is appealing to rapidly incorporate parallel reasoning into existing LLM inference pipelines.
format Preprint
id arxiv_https___arxiv_org_abs_2605_27570
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle LaneRoPE: Positional Encoding for Collaborative Parallel Reasoning and Generation
Cesa, Gabriele
Hehn, Thomas
Torres-Camps, Aleix
Casellas, Àlex Batlle
Ros-Giralt, Jordi
Behboodi, Arash
Orekondy, Tribhuvanesh
Artificial Intelligence
Parallel LLM test-time scaling techniques (e.g., best-of-$N$) require drawing $N>1$ sequences conditioned on the same input prompt. These methods boost accuracy while exploiting the computational efficiency of batching $N$ generations. However, each sequence in the batch is traditionally generated independently and hence does not reuse intermediate generations, computations, or observations from other sequences. In this paper, we propose LaneRoPE to enable coordination and collaboration among $N>1$ sequences at generation time. LaneRoPE involves two key ideas: (a) an inter-sequence attention mask to make sampling of sequences dependent on one another; and (b) a RoPE extension that injects positional information that captures relative positions between tokens, both within and outside a particular sequence. We evaluate our approach on mathematical reasoning tasks and find promising results: LaneRoPE enables collaboration among sequences, yielding additional accuracy gains under limited generated sequence length. Importantly, since LaneRoPE enables coordination with minimal changes to the underlying LLM architecture and introduces a negligible overhead at inference time, it is appealing to rapidly incorporate parallel reasoning into existing LLM inference pipelines.
title LaneRoPE: Positional Encoding for Collaborative Parallel Reasoning and Generation
topic Artificial Intelligence
url https://arxiv.org/abs/2605.27570