acceptodds
Under review as a conference paper at ICLR 2027

LBI: Parallel Scan Backpropagation via Latent Bounded Interfaces

Abstract

Backpropagation is sequential across depth, an -deep dependency chain that bottlenecks parallel training. Parallel-scan formulations reduce the depth to but compose Jacobians at per combine. We introduce Latent Bounded Interfaces (LBI), which restrict inter-region communication to an interface with , so the adjoint recursion becomes a suffix scan over Jacobians with the gradient exact under the bounded-interface model. The same restriction bounds what crosses a device boundary: per-sequence states and Jacobians in place of per-token activations, so a region-parallel training step's critical path is kilobytes per link regardless of width, and its one activation-sized transfer is a gradient that can be accumulated. Implemented as fused forward-mode kernels on four H100s, the step runs faster than pipeline parallelism at 1 Gbit/s and faster at 100 Mbit/s of link bandwidth, which extends exact-gradient training to multi-region and decentralized links. On language modeling, bounded-interface models train across Mamba-3, Transformer, and Hybrid backends at small interface ranks.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.