A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimization
Sebastian Sanokowski, Sepp Hochreiter, Sebastian Lehner
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. *
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.
Cited by top-tier papers39
- Physics-Informed Diffusion ModelsJan-Hendrik Bastek, WaiChing Sun, Dennis M. KochmannICLR 2025 · 165 citations
- MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal ControlYuchen Zhu, Wei Guo, Jaemoo Choi, Guan-Horng Liu et al.NeurIPS 2025 · 24 citations
- Proximal Diffusion Neural SamplerWei Guo, Jaemoo Choi, Yuchen Zhu, Molei Tao et al.ICLR 2026 · 19 citations
- Discrete Neural Flow Samplers with Locally Equivariant TransformerZijing Ou, Ruixiang Zhang, Yingzhen LiNeurIPS 2025 · 14 citations
- FrontierCO: Real-World and Large-Scale Evaluation of Machine Learning Solvers for Combinatorial OptimizationShengyu Feng, Weiwei Sun, Shanda Li, Ameet Talwalkar et al.ICLR 2026 · 13 citations
Builds on20
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Structured Denoising Diffusion Models in Discrete State-SpacesJacob Austin, Daniel D. Johnson, Jonathan Ho, Daniel Tarlow et al.NeurIPS 2021 · 2,256 citations
- On the Variance of the Adaptive Learning Rate and BeyondLiyuan Liu, Haoming Jiang, Pengcheng He, Weizhu Chen et al.ICLR 2020 · 2,210 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
Related papers
- Improved sampling via learned diffusionsLorenz Richter, Julius BernerICLR 2024 · 103 citations
- Rethinking Losses for Diffusion Bridge SamplersSebastian Sanokowski, Lukas Gruber, Christoph Bartmann, Sepp Hochreiter et al.NeurIPS 2025 · 7 citations
- Forward-Learned Discrete Diffusion: Learning how to noise to denoise fasterGrigory Bartosh, Teodora Pandeva, Sushrut Karmalkar, Javier ZazoICLR 2026 · 4 citations
- Scalable Discrete Diffusion Samplers: Combinatorial Optimization and Statistical PhysicsSebastian Sanokowski, Wilhelm Franz Berghammer, Haoyu Peter Wang, Martin Ennemoser et al.ICLR 2025
- S4S: Solving for a Fast Diffusion Model SolverEric Frankel, Sitan Chen, Jerry Li, Pang Wei Koh et al.ICML 2025
