Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhou, Tailin, Chen, Zhilin, Lyu, Wenlong, Chen, Zhitang, Tsang, Danny H. K., Zhang, Jun
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912815962390528
author Zhou, Tailin
Chen, Zhilin
Lyu, Wenlong
Chen, Zhitang
Tsang, Danny H. K.
Zhang, Jun
author_facet Zhou, Tailin
Chen, Zhilin
Lyu, Wenlong
Chen, Zhitang
Tsang, Danny H. K.
Zhang, Jun
contents Optimizing complex systems, from discovering therapeutic drugs to designing high-performance materials, remains a fundamental challenge across science and engineering, as the underlying rules are often unknown and costly to evaluate. Offline optimization aims to optimize designs for target scores using pre-collected datasets without system interaction. However, conventional approaches may fail beyond training data, predicting inaccurate scores and generating inferior designs. This paper introduces ManGO, a diffusion-based framework that learns the design-score manifold, capturing the design-score interdependencies holistically. Unlike existing methods that treat design and score spaces in isolation, ManGO unifies forward prediction and backward generation, attaining generalization beyond training data. Key to this is its derivative-free guidance for conditional generation, coupled with adaptive inference-time scaling that dynamically optimizes denoising paths. Extensive evaluations demonstrate that ManGO outperforms 24 single- and 10 multi-objective optimization methods across diverse domains, including synthetic tasks, robot control, material design, DNA sequence, and real-world engineering optimization.
format Preprint
id arxiv_https___arxiv_org_abs_2506_05680
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization
Zhou, Tailin
Chen, Zhilin
Lyu, Wenlong
Chen, Zhitang
Tsang, Danny H. K.
Zhang, Jun
Machine Learning
Artificial Intelligence
Optimizing complex systems, from discovering therapeutic drugs to designing high-performance materials, remains a fundamental challenge across science and engineering, as the underlying rules are often unknown and costly to evaluate. Offline optimization aims to optimize designs for target scores using pre-collected datasets without system interaction. However, conventional approaches may fail beyond training data, predicting inaccurate scores and generating inferior designs. This paper introduces ManGO, a diffusion-based framework that learns the design-score manifold, capturing the design-score interdependencies holistically. Unlike existing methods that treat design and score spaces in isolation, ManGO unifies forward prediction and backward generation, attaining generalization beyond training data. Key to this is its derivative-free guidance for conditional generation, coupled with adaptive inference-time scaling that dynamically optimizes denoising paths. Extensive evaluations demonstrate that ManGO outperforms 24 single- and 10 multi-objective optimization methods across diverse domains, including synthetic tasks, robot control, material design, DNA sequence, and real-world engineering optimization.
title Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2506.05680