English

Parametrized Multi-Agent Routing via Deep Attention Models

Machine Learning 2025-07-31 v1

Abstract

We propose a scalable deep learning framework for parametrized sequential decision-making (ParaSDM), where multiple agents jointly optimize discrete action policies and shared continuous parameters. A key subclass of this setting arises in Facility-Location and Path Optimization (FLPO), where multi-agent systems must simultaneously determine optimal routes and facility locations, aiming to minimize the cumulative transportation cost within the network. FLPO problems are NP-hard due to their mixed discrete-continuous structure and highly non-convex objective. To address this, we integrate the Maximum Entropy Principle (MEP) with a neural policy model called the Shortest Path Network (SPN)-a permutation-invariant encoder-decoder that approximates the MEP solution while enabling efficient gradient-based optimization over shared parameters. The SPN achieves up to 100×\times speedup in policy inference and gradient computation compared to MEP baselines, with an average optimality gap of approximately 6% across a wide range of problem sizes. Our FLPO approach yields over 10×\times lower cost than metaheuristic baselines while running significantly faster, and matches Gurobi's optimal cost with annealing at a 1500×\times speedup-establishing a new state of the art for ParaSDM problems. These results highlight the power of structured deep models for solving large-scale mixed-integer optimization tasks.

Keywords

Cite

@article{arxiv.2507.22338,
  title  = {Parametrized Multi-Agent Routing via Deep Attention Models},
  author = {Salar Basiri and Dhananjay Tiwari and Srinivasa M. Salapaka},
  journal= {arXiv preprint arXiv:2507.22338},
  year   = {2025}
}

Comments

This work is under submission to AAAI 2026. Please cite the arXiv version until the final version is published

R2 v1 2026-07-01T04:25:16.384Z