English

A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimization

Machine Learning 2025-08-25 v3 Artificial Intelligence Discrete Mathematics Machine Learning

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.

Keywords

Cite

@article{arxiv.2406.01661,
  title  = {A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimization},
  author = {Sebastian Sanokowski and Sepp Hochreiter and Sebastian Lehner},
  journal= {arXiv preprint arXiv:2406.01661},
  year   = {2025}
}

Comments

Accepted at ICML 2024