arXiv:2604.20109cs.LGcs.AI2026-04

用冷启动马尔可夫链优化神经网络,提升图匹配求解精度。

Learning to Solve the Quadratic Assignment Problem with Warm-Started MCMC Finetuning

  • 基于能量模型设计快速2交换采样,支持实时优化
  • 在QAPLIB上平均最优性差距接近零,对困难实例鲁棒性强
  • 适合需要高精度求解的组合优化场景

二次分配问题(QAP)是基础的NP难任务,传统启发式与学习型求解器均难以在结构多样的真实实例上保持稳定性能。为此,我们提出PLMA,一种创新的排列学习框架。该框架采用高效的冷启动马尔可夫链微调机制,在部署时利用短马尔可夫链锚定此前探索过的优良区域,实现快速适应。为在排列空间中高效进行马尔可夫链采样,我们设计了一种加性能量模型(EBM),支持O(1)时间复杂度的2-交换梅特罗波利斯-哈斯金采样。同时,用于参数化该能量模型的神经网络引入可扩展、灵活的跨图注意力机制,以建模设施与位置间的交互关系。大量实验表明,PLMA在多个基准测试中持续优于现有先进基线:在QAPLIB上实现近零平均最优性差距,对著名难题Taixxeyy实例表现出显著更强鲁棒性,并可在带宽最小化任务中作为有效求解器使用。

原文摘要 · Abstract (English)

The quadratic assignment problem (QAP) is a fundamental NP-hard task that poses significant challenges for both traditional heuristics and modern learning-based solvers. Existing QAP solvers still struggle to achieve consistently competitive performance across structurally diverse real-world instances. To bridge this performance gap, we propose PLMA, an innovative permutation learning framework. PLMA features an efficient warm-started MCMC finetuning procedure to enhance deployment-time performance, leveraging short Markov chains to anchor the adaptation to the promising regions previously explored. For rapid exploration via MCMC over the permutation space, we design an additive energy-based model (EBM) that enables an $O(1)$-time 2-swap Metropolis-Hastings sampling step. Moreover, the neural network used to parameterize the EBM incorporates a scalable and flexible cross-graph attention mechanism to model interactions between facilities and locations in the QAP. Extensive experiments demonstrate that PLMA consistently outperforms state-of-the-art baselines across various benchmarks. In particular, PLMA achieves a near-zero average optimality gap on QAPLIB, exhibits remarkably superior robustness on the notoriously difficult Taixxeyy instances, and also serves as an effective QAP solver in bandwidth minimization.

组合优化能量模型马尔可夫链排列学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。