Table of Contents
Fetching ...

TuneNSearch: a hybrid transfer learning and local search approach for solving vehicle routing problems

Arthur Corrêa, Cristóvão Silva, Liming Xu, Alexandra Brintrup, Samuel Moniz

TL;DR

TuneNSearch addresses cross-variant generalization in neural VRP methods by pre-training on the complex MDVRP and fine-tuning across VRP variants, leveraging an edge-aware E-GAT encoder integrated with a POMO-based Transformer. A lightweight local search with diverse operators is appended post-inference to refine solutions, delivering state-of-the-art neural-based performance across public benchmarks while maintaining polynomial runtime. Empirical results show strong generalization across instance sizes and distributions, with significant gains on multi-depot variants and competitive performance under time constraints. The approach offers practical, scalable decision-support for diverse routing scenarios, bridging learning and optimization in a unified framework.

Abstract

This paper introduces TuneNSearch, a hybrid transfer learning and local search approach for addressing diverse variants of the vehicle routing problem (VRP). Our method uses reinforcement learning to generate high-quality solutions, which are subsequently refined by an efficient local search procedure. To ensure broad adaptability across VRP variants, TuneNSearch begins with a pre-training phase on the multi-depot VRP (MDVRP), followed by a fine-tuning phase to adapt it to other problem formulations. The learning phase utilizes a Transformer-based architecture enhanced with edge-aware attention, which integrates edge distances directly into the attention mechanism to better capture spatial relationships inherent to routing problems. We show that the pre-trained model generalizes effectively to single-depot variants, achieving performance comparable to models trained specifically on single-depot instances. Simultaneously, it maintains strong performance on multi-depot variants, an ability that models pre-trained solely on single-depot problems lack. For example, on 100-node instances of multi-depot variants, TuneNSearch outperforms a model pre-trained on the CVRP by 44%. In contrast, on 100-node instances of single-depot variants, TuneNSearch performs similar to the CVRP model. To validate the effectiveness of our method, we conduct extensive computational experiments on public benchmark and randomly generated instances. Across multiple CVRPLIB datasets, TuneNSearch consistently achieves performance deviations of less than 3% from the best-known solutions in the literature, compared to 6-25% for other neural-based models, depending on problem complexity. Overall, our approach demonstrates strong generalization to different problem sizes, instance distributions, and VRP formulations, while maintaining polynomial runtime complexity despite the integration of the local search algorithm.

TuneNSearch: a hybrid transfer learning and local search approach for solving vehicle routing problems

TL;DR

TuneNSearch addresses cross-variant generalization in neural VRP methods by pre-training on the complex MDVRP and fine-tuning across VRP variants, leveraging an edge-aware E-GAT encoder integrated with a POMO-based Transformer. A lightweight local search with diverse operators is appended post-inference to refine solutions, delivering state-of-the-art neural-based performance across public benchmarks while maintaining polynomial runtime. Empirical results show strong generalization across instance sizes and distributions, with significant gains on multi-depot variants and competitive performance under time constraints. The approach offers practical, scalable decision-support for diverse routing scenarios, bridging learning and optimization in a unified framework.

Abstract

This paper introduces TuneNSearch, a hybrid transfer learning and local search approach for addressing diverse variants of the vehicle routing problem (VRP). Our method uses reinforcement learning to generate high-quality solutions, which are subsequently refined by an efficient local search procedure. To ensure broad adaptability across VRP variants, TuneNSearch begins with a pre-training phase on the multi-depot VRP (MDVRP), followed by a fine-tuning phase to adapt it to other problem formulations. The learning phase utilizes a Transformer-based architecture enhanced with edge-aware attention, which integrates edge distances directly into the attention mechanism to better capture spatial relationships inherent to routing problems. We show that the pre-trained model generalizes effectively to single-depot variants, achieving performance comparable to models trained specifically on single-depot instances. Simultaneously, it maintains strong performance on multi-depot variants, an ability that models pre-trained solely on single-depot problems lack. For example, on 100-node instances of multi-depot variants, TuneNSearch outperforms a model pre-trained on the CVRP by 44%. In contrast, on 100-node instances of single-depot variants, TuneNSearch performs similar to the CVRP model. To validate the effectiveness of our method, we conduct extensive computational experiments on public benchmark and randomly generated instances. Across multiple CVRPLIB datasets, TuneNSearch consistently achieves performance deviations of less than 3% from the best-known solutions in the literature, compared to 6-25% for other neural-based models, depending on problem complexity. Overall, our approach demonstrates strong generalization to different problem sizes, instance distributions, and VRP formulations, while maintaining polynomial runtime complexity despite the integration of the local search algorithm.

Paper Structure

This paper contains 29 sections, 15 equations, 10 figures, 11 tables, 3 algorithms.

Figures (10)

  • Figure 1: Encoder-decoder structure of TuneNSearch.
  • Figure 1: Sensitivity analysis of the number of encoder layers $L$.
  • Figure 2: TuneNSearch pre-training and fine-tuning overview.
  • Figure 2: Sensitivity analysis of the hidden dimension $h_x$, hidden edge dimension $h_e$ and number of heads $H$.
  • Figure 3: Overview of the inference process of TuneNSearch.
  • ...and 5 more figures