Learning Generalizable Models for Vehicle Routing Problems via Knowledge Distillation
Jieyi Bi, Yining Ma, Jiahai Wang, Zhiguang Cao, Jinbiao Chen, Yuan Sun, Yeow Meng Chee
Abstract
Recent neural methods for vehicle routing problems always train and test the deep models on the same instance distribution (i.e., uniform). To tackle the consequent cross-distribution generalization concerns, we bring the knowledge distillation to this field and propose an Adaptive Multi-Distribution Knowledge Distillation (AMDKD) scheme for learning more generalizable deep models. Particularly, our AMDKD leverages various knowledge from multiple teachers trained on exemplar distributions to yield a light-weight yet generalist student model. Meanwhile, we equip AMDKD with an adaptive strategy that allows the student to concentrate on difficult distributions, so as to absorb hard-to-master knowledge more effectively. Extensive experimental results show that, compared with the baseline neural methods, our AMDKD is able to achieve competitive results on both unseen in-distribution and out-of-distribution instances, which are either randomly synthesized or adopted from benchmark datasets (i.e., TSPLIB and CVRPLIB). Notably, our AMDKD is generic, and consumes less computational resources for inference. † Equally contributed.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b364baff-1f5c-48fb-be66-7c197cf4c844Cited by top-tier papers41
- DIFUSCO: Graph-based Diffusion Solvers for Combinatorial OptimizationZhiqing Sun, Yiming YangNeurIPS 2023 · 356 citations
- DeepACO: Neural-enhanced Ant Systems for Combinatorial OptimizationHaoran Ye, Jiarui Wang, Zhiguang Cao, Helan Liang et al.NeurIPS 2023 · 158 citations
- Learning to Search Feasible and Infeasible Regions of Routing Problems with Flexible Neural k-OptYining Ma, Zhiguang Cao, Yeow Meng CheeNeurIPS 2023 · 129 citations
- GLOP: Learning Global Partition and Local Construction for Solving Large-Scale Routing Problems in Real-TimeHaoran Ye, Jiarui Wang, Helan Liang, Zhiguang Cao et al.AAAI 2024 · 100 citations
- Towards Omni-generalizable Neural Methods for Vehicle Routing ProblemsJianan Zhou, Yaoxin Wu, Wen Song, Zhiguang Cao et al.ICML 2023 · 90 citations
Builds on15
- POMO: Policy Optimization with Multiple Optima for Reinforcement LearningYeong-Dae Kwon, Jinho Choo, Byoungjip Kim, Iljoo Yoon et al.NeurIPS 2020 · 731 citations
- Does Knowledge Distillation Really Work?Samuel Stanton, Pavel Izmailov, Polina Kirichenko, Alexander A. Alemi et al.NeurIPS 2021 · 318 citations
- Self-Distillation Amplifies Regularization in Hilbert SpaceHossein Mobahi, Mehrdad Farajtabar, Peter L. BartlettNeurIPS 2020 · 298 citations
- FastBERT: a Self-distilling BERT with Adaptive Inference TimeWeijie Liu, Peng Zhou, Zhiruo Wang, Zhe Zhao et al.ACL 2020 · 257 citations
- Generalize a Small Pre-trained Model to Arbitrarily Large TSP InstancesZhang-Hua Fu, Kai-Bin Qiu, Hongyuan ZhaAAAI 2021 · 247 citations
Related papers
- MTL-KD: Multi-Task Learning Via Knowledge Distillation for Generalizable Neural Vehicle Routing SolverYuepeng Zheng, Fu Luo, Zhenkun Wang, Yaoxin Wu et al.NeurIPS 2025 · 13 citations
- Combination-of-Experts with Knowledge Sharing for Cross-Task Vehicle Routing ProblemsZikang Yu, Jinbiao Chen, Jiahai WangICLR 2026
- Distilling Autoregressive Models to Obtain High-Performance Non-autoregressive Solvers for Vehicle Routing Problems with Faster Inference SpeedYubin Xiao, Di Wang, Boyang Li, Mingzhao Wang et al.AAAI 2024 · 34 citations
- Ensemble-based Deep Reinforcement Learning for Vehicle Routing Problems under Distribution ShiftYuan Jiang, Zhiguang Cao, Yaoxin Wu, Wen Song et al.NeurIPS 2023 · 43 citations
- Learning to Solve Routing Problems via Distributionally Robust OptimizationYuan Jiang, Yaoxin Wu, Zhiguang Cao, Jie ZhangAAAI 2022 · 60 citations
