Preference Optimization for Combinatorial Optimization Problems

Mingjun Pan, Guanquan Lin, You-Wei Luo, Bin Zhu, Zhien Dai +2 more
5/13/2025
cs.LG

No official code found. Best implementation: MingjunPan/PO4COPs (28 stars, MIT, last commit 2025-06-06).

Abstract

Reinforcement Learning (RL) has emerged as a powerful tool for neural combinatorial optimization, enabling models to learn heuristics that solve complex problems without requiring expert knowledge. Despite significant progress, existing RL approaches face challenges such as diminishing reward signals and inefficient exploration in vast combinatorial action spaces, leading to inefficiency. In this paper, we propose Preference Optimization, a novel method that transforms quantitative reward signals into qualitative preference signals via statistical comparison modeling, emphasizing the superiority among sampled solutions. Methodologically, by reparameterizing the reward function in terms of policy and utilizing preference models, we formulate an entropy-regularized RL objective that aligns the policy directly with preferences while avoiding intractable computations. Furthermore, we integrate local search techniques into the fine-tuning rather than post-processing to generate high-quality preference pairs, helping the policy escape local optima. Empirical results on various benchmarks, such as the Traveling Salesman Problem (TSP), the Capacitated Vehicle Routing Problem (CVRP) and the Flexible Flow Shop Problem (FFSP), demonstrate that our method significantly outperforms existing RL algorithms, achieving superior convergence efficiency and solution quality.

View on arXivView PDF

Code Implementations(1)

MingjunPan/PO4COPsCommunity79%
MIT

[ICML 25] "Preference Optimization for Combinatorial Optimization Problems"

286PythonJun 6, 20251 years agoMIT
combinatorial-optimizationpreference-alignmentreinforcement-learning

Similar Papers

Cite this paper

@article{pan2025preference,
  title  = {Preference Optimization for Combinatorial Optimization Problems},
  author = {Mingjun Pan and Guanquan Lin and You-Wei Luo and Bin Zhu and Zhien Dai and Lijun Sun and Chun Yuan},
  year   = {2025},
  eprint = {2505.08735},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.08735v1},
  journal = {ICML 2025 2025}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.08735.svg)](https://codeofpaper.com/paper/2505.08735)

Discussion