An Efficient Diffusion-based Non-Autoregressive Solver for Traveling Salesman Problem

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Mingzhao, Zhou, You, Cao, Zhiguang, Xiao, Yubin, Wu, Xuan, Pang, Wei, Jiang, Yuan, Yang, Hui, Zhao, Peng, Li, Yuanshu
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917900885950464
author Wang, Mingzhao
Zhou, You
Cao, Zhiguang
Xiao, Yubin
Wu, Xuan
Pang, Wei
Jiang, Yuan
Yang, Hui
Zhao, Peng
Li, Yuanshu
author_facet Wang, Mingzhao
Zhou, You
Cao, Zhiguang
Xiao, Yubin
Wu, Xuan
Pang, Wei
Jiang, Yuan
Yang, Hui
Zhao, Peng
Li, Yuanshu
contents Recent advances in neural models have shown considerable promise in solving Traveling Salesman Problems (TSPs) without relying on much hand-crafted engineering. However, while non-autoregressive (NAR) approaches benefit from faster inference through parallelism, they typically deliver solutions of inferior quality compared to autoregressive ones. To enhance the solution quality while maintaining fast inference, we propose DEITSP, a diffusion model with efficient iterations tailored for TSP that operates in a NAR manner. Firstly, we introduce a one-step diffusion model that integrates the controlled discrete noise addition process with self-consistency enhancement, enabling optimal solution prediction through simultaneous denoising of multiple solutions. Secondly, we design a dual-modality graph transformer to bolster the extraction and fusion of features from node and edge modalities, while further accelerating the inference with fewer layers. Thirdly, we develop an efficient iterative strategy that alternates between adding and removing noise to improve exploration compared to previous diffusion methods. Additionally, we devise a scheduling framework to progressively refine the solution space by adjusting noise levels, facilitating a smooth search for optimal solutions. Extensive experiments on real-world and large-scale TSP instances demonstrate that DEITSP performs favorably against existing neural approaches in terms of solution quality, inference latency, and generalization ability. Our code is available at $\href{https://github.com/DEITSP/DEITSP}{https://github.com/DEITSP/DEITSP}$.
format Preprint
id arxiv_https___arxiv_org_abs_2501_13767
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle An Efficient Diffusion-based Non-Autoregressive Solver for Traveling Salesman Problem
Wang, Mingzhao
Zhou, You
Cao, Zhiguang
Xiao, Yubin
Wu, Xuan
Pang, Wei
Jiang, Yuan
Yang, Hui
Zhao, Peng
Li, Yuanshu
Machine Learning
Recent advances in neural models have shown considerable promise in solving Traveling Salesman Problems (TSPs) without relying on much hand-crafted engineering. However, while non-autoregressive (NAR) approaches benefit from faster inference through parallelism, they typically deliver solutions of inferior quality compared to autoregressive ones. To enhance the solution quality while maintaining fast inference, we propose DEITSP, a diffusion model with efficient iterations tailored for TSP that operates in a NAR manner. Firstly, we introduce a one-step diffusion model that integrates the controlled discrete noise addition process with self-consistency enhancement, enabling optimal solution prediction through simultaneous denoising of multiple solutions. Secondly, we design a dual-modality graph transformer to bolster the extraction and fusion of features from node and edge modalities, while further accelerating the inference with fewer layers. Thirdly, we develop an efficient iterative strategy that alternates between adding and removing noise to improve exploration compared to previous diffusion methods. Additionally, we devise a scheduling framework to progressively refine the solution space by adjusting noise levels, facilitating a smooth search for optimal solutions. Extensive experiments on real-world and large-scale TSP instances demonstrate that DEITSP performs favorably against existing neural approaches in terms of solution quality, inference latency, and generalization ability. Our code is available at $\href{https://github.com/DEITSP/DEITSP}{https://github.com/DEITSP/DEITSP}$.
title An Efficient Diffusion-based Non-Autoregressive Solver for Traveling Salesman Problem
topic Machine Learning
url https://arxiv.org/abs/2501.13767