Saved in:
| Main Authors: | , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2411.05676 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866912111590899712 |
|---|---|
| author | Hou, Xiaoyang Zhu, Tian Ren, Milong Bu, Dongbo Gao, Xin Zhang, Chunming Sun, Shiwei |
| author_facet | Hou, Xiaoyang Zhu, Tian Ren, Milong Bu, Dongbo Gao, Xin Zhang, Chunming Sun, Shiwei |
| contents | Generating molecular graphs is crucial in drug design and discovery but remains challenging due to the complex interdependencies between nodes and edges. While diffusion models have demonstrated their potentiality in molecular graph design, they often suffer from unstable training and inefficient sampling. To enhance generation performance and training stability, we propose GGFlow, a discrete flow matching generative model incorporating optimal transport for molecular graphs and it incorporates an edge-augmented graph transformer to enable the direct communications among chemical bounds. Additionally, GGFlow introduces a novel goal-guided generation framework to control the generative trajectory of our model, aiming to design novel molecular structures with the desired properties. GGFlow demonstrates superior performance on both unconditional and conditional molecule generation tasks, outperforming existing baselines and underscoring its effectiveness and potential for wider application. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2411_05676 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Improving Molecular Graph Generation with Flow Matching and Optimal Transport Hou, Xiaoyang Zhu, Tian Ren, Milong Bu, Dongbo Gao, Xin Zhang, Chunming Sun, Shiwei Machine Learning Artificial Intelligence Generating molecular graphs is crucial in drug design and discovery but remains challenging due to the complex interdependencies between nodes and edges. While diffusion models have demonstrated their potentiality in molecular graph design, they often suffer from unstable training and inefficient sampling. To enhance generation performance and training stability, we propose GGFlow, a discrete flow matching generative model incorporating optimal transport for molecular graphs and it incorporates an edge-augmented graph transformer to enable the direct communications among chemical bounds. Additionally, GGFlow introduces a novel goal-guided generation framework to control the generative trajectory of our model, aiming to design novel molecular structures with the desired properties. GGFlow demonstrates superior performance on both unconditional and conditional molecule generation tasks, outperforming existing baselines and underscoring its effectiveness and potential for wider application. |
| title | Improving Molecular Graph Generation with Flow Matching and Optimal Transport |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2411.05676 |