Monte Carlo Policy Gradient Method for Binary Optimization

Cheng Chen, Ruitao Chen, Tianyou Li, Ruichen Ao, Zaiwen Wen
7/3/2023
math.OCcs.AIcs.LG

Official code: optsuite/MCPG (25 stars, GPL-3.0, last commit 2025-07-31).

Abstract

Binary optimization has a wide range of applications in combinatorial optimization problems such as MaxCut, MIMO detection, and MaxSAT. However, these problems are typically NP-hard due to the binary constraints. We develop a novel probabilistic model to sample the binary solution according to a parameterized policy distribution. Specifically, minimizing the KL divergence between the parameterized policy distribution and the Gibbs distributions of the function value leads to a stochastic optimization problem whose policy gradient can be derived explicitly similar to reinforcement learning. For coherent exploration in discrete spaces, parallel Markov Chain Monte Carlo (MCMC) methods are employed to sample from the policy distribution with diversity and approximate the gradient efficiently. We further develop a filter scheme to replace the original objective function by the one with the local search technique to broaden the horizon of the function landscape. Convergence to stationary points in expectation of the policy gradient method is established based on the concentration inequality for MCMC. Numerical results show that this framework is very promising to provide near-optimal solutions for quite a few binary optimization problems.

View on arXivView PDF

Code Implementations(1)

optsuite/MCPGOfficial100%
GPL-3.0
257C++, CudaJun 12, 20231 years agoGPL-3.0

Similar Papers

Cite this paper

@article{chen2023monte,
  title  = {Monte Carlo Policy Gradient Method for Binary Optimization},
  author = {Cheng Chen and Ruitao Chen and Tianyou Li and Ruichen Ao and Zaiwen Wen},
  year   = {2023},
  eprint = {2307.00783},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2307.00783v1}
}

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/2307.00783.svg)](https://codeofpaper.com/paper/2307.00783)

Discussion