A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimization

Sebastian Sanokowski, Sepp Hochreiter, Sebastian Lehner
6/3/2024
cs.LGcs.AIcs.DM

No official code found. Best implementation: ml-jku/DIffUCO (69 stars, last commit 2025-10-09).

Abstract

Learning to sample from intractable distributions over discrete sets without relying on corresponding training data is a central problem in a wide range of fields, including Combinatorial Optimization. Currently, popular deep learning-based approaches rely primarily on generative models that yield exact sample likelihoods. This work introduces a method that lifts this restriction and opens the possibility to employ highly expressive latent variable models like diffusion models. Our approach is conceptually based on a loss that upper bounds the reverse Kullback-Leibler divergence and evades the requirement of exact sample likelihoods. We experimentally validate our approach in data-free Combinatorial Optimization and demonstrate that our method achieves a new state-of-the-art on a wide range of benchmark problems.

View on arXivView PDF

Code Implementations(1)

ml-jku/DIffUCOCommunity100%
697Shell, PythonMay 3, 20241 years ago

Similar Papers

Cite this paper

@article{sanokowski2024diffusion,
  title  = {A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimization},
  author = {Sebastian Sanokowski and Sepp Hochreiter and Sebastian Lehner},
  year   = {2024},
  eprint = {2406.01661},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.01661v3},
  journal = {ICML 2024 2024}
}

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

Discussion