arXiv:2411.00003cs.AIcs.LG2024-11被引 3

无需专家搜索,用扩散模型直接生成最优解

Unsupervised Training of Diffusion Models for Feasible Solution Generation in Neural Combinatorial Optimization

  • 自监督训练扩散模型,直接优化解的代价
  • 在PMSP和ATSP上达到当前最佳性能
  • 适用于两类物品的组合优化问题

神经组合优化(NCO)方法近年来在不依赖人工启发式算法的情况下生成近似最优解方面表现出色。然而,这些方法的高性能通常依赖于问题特定的后期搜索过程,限制了其在常见组合优化问题(如旅行商问题)中的应用。本文提出IC/DC,一种从零开始无监督训练的组合优化框架,通过自监督方式最小化解的代价并满足问题约束。该框架专为涉及两类物品的组合优化问题设计,无需额外的搜索过程即可生成有效解。其新颖的架构能捕捉物品间的复杂关系,在复杂场景中实现高效优化。在平行机调度问题(PMSP)和非对称旅行商问题(ATSP)上,IC/DC的表现优于现有NCO方法。

原文摘要 · Abstract (English)

Recent advancements in neural combinatorial optimization (NCO) methods have shown promising results in generating near-optimal solutions without the need for expert-crafted heuristics. However, high performance of these approaches often rely on problem-specific human-expertise-based search after generating candidate solutions, limiting their applicability to commonly solved CO problems such as Traveling Salesman Problem (TSP). In this paper, we present IC/DC, an unsupervised CO framework that directly trains a diffusion model from scratch. We train our model in a self-supervised way to minimize the cost of the solution while adhering to the problem-specific constraints. IC/DC is specialized in addressing CO problems involving two distinct sets of items, and it does not need problem-specific search processes to generate valid solutions. IC/DC employs a novel architecture capable of capturing the intricate relationships between items, and thereby enabling effective optimization in challenging CO scenarios. IC/DC achieves state-of-the-art performance relative to existing NCO methods on the Parallel Machine Scheduling Problem (PMSP) and Asymmetric Traveling Salesman Problem (ATSP).

组合优化扩散模型无监督学习

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