English

LoRe: Adaptive Interaction-Evaluation Routing with Per-Step Interaction Budgets for Iterative Graph Solvers

Machine Learning 2026-05-29 v1 Artificial Intelligence

Abstract

Diffusion-based neural solvers for combinatorial optimization repeatedly re-evaluate dense edge/factor interactions, making inference expensive in wall-clock time and often memory-bound at scale. Inspired by the computational methodologies of many-body physics, we introduce LoRe, a training-free, inference-time drop-in wrapper that enforces per-step interaction-evaluation budgeting: at each iteration, it evaluates only a fixed fraction of interactions by dynamically routing computation to high-conflict or high-uncertainty interactions, instead of using a fixed sparsification (e.g., static kNN graphs or static masks). Under fully inclusive end-to-end wall-clock accounting, LoRe substantially improves scalability on the Maximum Independent Set (MIS) problem, extending feasible inference more than 3×3\times beyond the baseline's out-of-memory limit, delivering a 8×\sim 8\times speedup and a 12×\sim 12\times peak-memory reduction, with solution quality preserved in this regime. Demonstrating cross-task generality on the large-scale Traveling Salesperson Problem (TSP) and zero-shot robustness to topology shifts, LoRe achieves a 15×\sim 15\times speedup at n=1000n=1000 with a 44×44\times memory reduction and competitive tour quality.

Keywords

Cite

@article{arxiv.2605.29005,
  title  = {LoRe: Adaptive Interaction-Evaluation Routing with Per-Step Interaction Budgets for Iterative Graph Solvers},
  author = {Jintao Li and Yong-Yi Wang and Zheng-An Wang and Heng Fan},
  journal= {arXiv preprint arXiv:2605.29005},
  year   = {2026}
}

Comments

Accepted at ICML 2026