Preference-Guided Diffusion for Multi-Objective Offline Optimization
Yashas Annadani, Syrine Belakaria, Stefano Ermon, Stefan Bauer, Barbara Engelhardt
Abstract
Offline multi-objective optimization aims to identify Pareto-optimal solutions given a dataset of designs and their objective values. In this work, we propose a preference-guided diffusion model that generates Pareto-optimal designs by leveraging a classifier-based guidance mechanism. Our guidance classifier is a preference model trained to predict the probability that one design dominates another, directing the diffusion model toward optimal regions of the design space. Crucially, this preference model generalizes beyond the training distribution, enabling the discovery of Pareto-optimal solutions outside the observed dataset. We introduce a novel diversity-aware preference guidance, augmenting Pareto dominance preference with diversity criteria. This ensures that generated solutions are optimal and well-distributed across the objective space, a capability absent in prior generative methods for offline multi-objective optimization. We evaluate our approach on various continuous offline multi-objective optimization tasks and find that it consistently outperforms other inverse/generative approaches while remaining competitive with forward/ surrogate-based optimization methods. Our results highlight the effectiveness of classifier-guided diffusion models in generating diverse and high-quality solutions that approximate the Pareto front well.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 4f057727-4fe0-4604-a24f-7a7727aebe63Cited by top-tier papers5
- Steering Generative Models with Experimental Data for Protein Fitness OptimizationJason Yang, Wenda Chu, Daniel Khalil, Raul Astudillo et al.NeurIPS 2025 · 13 citations
- SPREAD: Sampling-based Pareto front Refinement via Efficient Adaptive DiffusionSedjro Salomon Hotegni, Sebastian PeitzICLR 2026 · 3 citations
- Pareto-Conditioned Diffusion Models for Offline Multi-Objective OptimizationJatan Shrestha, Santeri Heiskanen, Kari Hepola, Severi Rissanen et al.ICLR 2026 · 2 citations
- TMS: Trajectory-Mixed Supervision for On-Policy Self DistillationRana Khan, Zijie Liu, Zhen Tan, Charles Fleming et al.ICML 2026
- Amortized Active Generation of Pareto SetsDaniel M. Steinberg, Asiri Wijesinghe, Rafael Oliveira, Piotr Koniusz et al.NeurIPS 2025
Builds on25
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Diffusion Models Beat GANs on Image SynthesisPrafulla Dhariwal, Alexander Quinn NicholNeurIPS 2021 · 13,211 citations
- Direct Preference Optimization: Your Language Model is Secretly a Reward ModelRafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D. Manning et al.NeurIPS 2023 · 10,924 citations
- Gradient Surgery for Multi-Task LearningTianhe Yu, Saurabh Kumar, Abhishek Gupta, Sergey Levine et al.NeurIPS 2020 · 2,261 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
- ParetoFlow: Guided Flows in Multi-Objective OptimizationYe Yuan, Can Chen, Christopher Pal, Xue LiuICLR 2025
- Diffusion Models for Black-Box OptimizationSiddarth Krishnamoorthy, Satvik Mehul Mashkaria, Aditya GroverICML 2023 · 94 citations
- Prior-Guided Diffusion Planning for Offline Reinforcement LearningDonghyeon Ki, JunHyeok Oh, Seong-Woong Shim, Byung-Jun LeeNeurIPS 2025 · 16 citations
- Diversity-Driven Offline Multi-Objective Optimization via Nested Pareto Set LearningYiyi Zhu, Yaolin Wen, Xiang Xia, Xin An et al.ICML 2026
- MODULI: Unlocking Preference Generalization via Diffusion Models for Offline Multi-Objective Reinforcement LearningYifu Yuan, Zhenrui Zheng, Zibin Dong, Jianye HaoICML 2025
