Graph Diffusion Policy Optimization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liu, Yijing, Du, Chao, Pang, Tianyu, Li, Chongxuan, Lin, Min, Chen, Wei
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913563168210944
author Liu, Yijing
Du, Chao
Pang, Tianyu
Li, Chongxuan
Lin, Min
Chen, Wei
author_facet Liu, Yijing
Du, Chao
Pang, Tianyu
Li, Chongxuan
Lin, Min
Chen, Wei
contents Recent research has made significant progress in optimizing diffusion models for downstream objectives, which is an important pursuit in fields such as graph generation for drug design. However, directly applying these models to graph presents challenges, resulting in suboptimal performance. This paper introduces graph diffusion policy optimization (GDPO), a novel approach to optimize graph diffusion models for arbitrary (e.g., non-differentiable) objectives using reinforcement learning. GDPO is based on an eager policy gradient tailored for graph diffusion models, developed through meticulous analysis and promising improved performance. Experimental results show that GDPO achieves state-of-the-art performance in various graph generation tasks with complex and diverse objectives. Code is available at https://github.com/sail-sg/GDPO.
format Preprint
id arxiv_https___arxiv_org_abs_2402_16302
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Graph Diffusion Policy Optimization
Liu, Yijing
Du, Chao
Pang, Tianyu
Li, Chongxuan
Lin, Min
Chen, Wei
Machine Learning
Artificial Intelligence
Computational Engineering, Finance, and Science
Recent research has made significant progress in optimizing diffusion models for downstream objectives, which is an important pursuit in fields such as graph generation for drug design. However, directly applying these models to graph presents challenges, resulting in suboptimal performance. This paper introduces graph diffusion policy optimization (GDPO), a novel approach to optimize graph diffusion models for arbitrary (e.g., non-differentiable) objectives using reinforcement learning. GDPO is based on an eager policy gradient tailored for graph diffusion models, developed through meticulous analysis and promising improved performance. Experimental results show that GDPO achieves state-of-the-art performance in various graph generation tasks with complex and diverse objectives. Code is available at https://github.com/sail-sg/GDPO.
title Graph Diffusion Policy Optimization
topic Machine Learning
Artificial Intelligence
Computational Engineering, Finance, and Science
url https://arxiv.org/abs/2402.16302