arXiv:2502.12188cs.LGcs.AI2025-02AAAI被引 3

无需训练即可让扩散模型跨问题、跨规模求解组合优化难题

Boosting Cross-problem Generalization in Diffusion-Based Neural Combinatorial Solver via Inference Time Adaptation

  • 推理时通过自适应调整实现零样本跨问题迁移
  • 仅在TSP上训练的模型可零样本解决PCTSP和定向旅行商问题
  • 理论分析支持跨问题泛化能力,适合高效部署于新场景

基于扩散的神经组合优化(NCO)通过学习离散扩散模型生成解,无需手工设计领域知识,已证明在求解NP完全(NPC)问题上的有效性。然而,现有方法在跨尺度与跨问题泛化方面仍存在显著挑战,且训练成本高于传统求解器。尽管扩散模型近期提出无需训练的引导方法,利用预定义引导函数实现条件生成,但此类方法在组合优化中尚未充分探索。为此,我们提出一种无需训练的推理时自适应框架(DIFU-Ada),使扩散型NCO求解器具备零样本跨问题迁移与跨尺度泛化能力,无需额外训练。我们提供理论分析以理解跨问题迁移机制。实验表明,仅在旅行商问题(TSP)上训练的扩散求解器,可通过推理时自适应,在不同规模的TSP变体(如带奖励收集的旅行商问题PCTSP和定向旅行商问题OP)上实现具有竞争力的零样本性能。

原文摘要 · Abstract (English)

Diffusion-based Neural Combinatorial Optimization (NCO) has demonstrated effectiveness in solving NP-complete (NPC) problems by learning discrete diffusion models for solution generation, eliminating hand-crafted domain knowledge. Despite their success, existing NCO methods face significant challenges in both cross-scale and cross-problem generalization, and high training costs compared to traditional solvers. While recent studies on diffusion models have introduced training-free guidance approaches that leverage pre-defined guidance functions for conditional generation, such methodologies have not been extensively explored in combinatorial optimization. To bridge this gap, we propose a training-free inference time adaptation framework (DIFU-Ada) that enables both the zero-shot cross-problem transfer and cross-scale generalization capabilities of diffusion-based NCO solvers without requiring additional training. We provide theoretical analysis that helps understanding the cross-problem transfer capability. Our experimental results demonstrate that a diffusion solver, trained exclusively on the Traveling Salesman Problem (TSP), can achieve competitive zero-shot transfer performance across different problem scales on TSP variants, such as Prize Collecting TSP (PCTSP) and the Orienteering Problem (OP), through inference time adaptation.

扩散模型组合优化零样本迁移推理优化

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