Parallel Algorithms Align with Neural Execution
Valerie Engelmayer, Dobrik Georgiev, Petar Veličković
TL;DR
This work argues that neural algorithmic reasoning should leverage parallel computation, as parallel algorithms align with the intrinsic parallelism of neural architectures and can reduce training time and improve predictions. Using the CLRS neural framework, the paper analyzes parallel versus sequential implementations of searching, sorting, and strongly connected components, showing that parallel approaches yield shorter, less redundant computation trajectories and higher neural efficiency. The results demonstrate that parallel algorithms are easier to learn and execute in neural models, with significant training speedups and improved generalization, particularly for out-of-distribution data. The study connects neural computation with classical parallel models (Processor Arrays, PRAM) and highlights practical implications for designing neural learners that exploit parallelism in graph-based processing. Overall, adopting parallel algorithms enhances training efficiency and predictive performance, guiding future integration of parallel algorithmic reasoning in neural architectures.
Abstract
Neural algorithmic reasoners are parallel processors. Teaching them sequential algorithms contradicts this nature, rendering a significant share of their computations redundant. Parallel algorithms however may exploit their full computational power, therefore requiring fewer layers to be executed. This drastically reduces training times, as we observe when comparing parallel implementations of searching, sorting and finding strongly connected components to their sequential counterparts on the CLRS framework. Additionally, parallel versions achieve (often strongly) superior predictive performance.
