ICML 2026poster0 citations

Unsupervised Diffusion for Combinatorial Optimization via Adjoint Matching

Shengyu Feng, Tarun Suresh, Yiming Yang

Abstract

Neural solvers have recently emerged as powerful tools for combinatorial optimization (CO). Among them, diffusion models have shown strong promise due to their ability to capture highly multimodal solution distributions in CO through iterative generative processes. However, training diffusion models typically requires large collections of near-optimal solutions, which limits their scalability and generalization. We address this fundamental challenge by extending adjoint matching, a powerful unsupervised diffusion training framework based on chain-rule–style gradient propagation in continuous spaces, to discrete combinatorial domains. Our approach resolves the broken-gradient issue inherent to discrete data and unifies local and global training objectives within a single principled framework. Empirically, our method consistently outperforms existing unsupervised baselines and achieves performance comparable to supervised diffusion models.

DiffusionOptimizationTheoryMultimodal
BibTeX
@inproceedings{
feng2026unsupervised,
title={Unsupervised Diffusion Solver for Combinatorial Optimization via Combinatorial Adjoint Matching},
author={Shengyu Feng and Tarun Suresh and Yiming Yang},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=y6LGSCnNOz}
}