Return
TransACO: Transformer-Guided Ant Colony Algorithm for Combinatorial Optimization
DOI:10.1109/tetci.2026.3683773.png)
Abstract
En 中文
Ant Colony Optimization (ACO) is a heuristic algorithm that has been widely used to solve combinatorial optimization problems. However, determining heuristic measures and initial pheromones levels in an ACO is problem-specific and often requires expert intervention, posing a significant challenge for the users. We investigate how the Transformer, a powerful learning model based on a self-attention mechanism, can be used to enable the automatic design of heuristic measures and pheromones. We present Transformer-Guided ACO (TransACO), a framework that combines neural methods with heuristic algorithms. The Transformer is designed as a learner that encodes problem instances, captures individual interactions, and generates heuristic measures and pheromone matrix. Additionally, a candidate point perturbation local search technique is proposed to enhance exploration. We conduct extensive experiments on four representative combinatorial optimization problems—traveling salesman problem, capacitated vehicle routing problem, orienteering problem, and multiple knapsack problem—across diverse instance sizes. Notably, in cross-scale generalization experiments, a model trained solely on small instances successfully transfers to 10k-node problems: on TSP-10 k, TransACO achieves a 4.12% improvement, and on CVRP-10 k, it yields a 1.16% gain over strong baselines. These findings demonstrate the robustness, adaptability, and scalability of TransACO.
Keywords:
Ant colony optimization
combinatorial optimization
transformer
deep learning
Journal
I
IF:
6.5
Papers:
1.4K
Citations:
4.5K

