Saved in:
Bibliographic Details
Main Authors: Niu, Yifan, Gao, Ziqi, Xu, Tingyang, Liu, Yang, Bian, Yatao, Rong, Yu, Huang, Junzhou, Li, Jia
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2503.01488
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915180360761344
author Niu, Yifan
Gao, Ziqi
Xu, Tingyang
Liu, Yang
Bian, Yatao
Rong, Yu
Huang, Junzhou
Li, Jia
author_facet Niu, Yifan
Gao, Ziqi
Xu, Tingyang
Liu, Yang
Bian, Yatao
Rong, Yu
Huang, Junzhou
Li, Jia
contents Exploring chemical space to find novel molecules that simultaneously satisfy multiple properties is crucial in drug discovery. However, existing methods often struggle with trading off multiple properties due to the conflicting or correlated nature of chemical properties. To tackle this issue, we introduce InversionGNN framework, an effective yet sample-efficient dual-path graph neural network (GNN) for multi-objective drug discovery. In the direct prediction path of InversionGNN, we train the model for multi-property prediction to acquire knowledge of the optimal combination of functional groups. Then the learned chemical knowledge helps the inversion generation path to generate molecules with required properties. In order to decode the complex knowledge of multiple properties in the inversion path, we propose a gradient-based Pareto search method to balance conflicting properties and generate Pareto optimal molecules. Additionally, InversionGNN is able to search the full Pareto front approximately in discrete chemical space. Comprehensive experimental evaluations show that InversionGNN is both effective and sample-efficient in various discrete multi-objective settings including drug discovery.
format Preprint
id arxiv_https___arxiv_org_abs_2503_01488
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle InversionGNN: A Dual Path Network for Multi-Property Molecular Optimization
Niu, Yifan
Gao, Ziqi
Xu, Tingyang
Liu, Yang
Bian, Yatao
Rong, Yu
Huang, Junzhou
Li, Jia
Machine Learning
Exploring chemical space to find novel molecules that simultaneously satisfy multiple properties is crucial in drug discovery. However, existing methods often struggle with trading off multiple properties due to the conflicting or correlated nature of chemical properties. To tackle this issue, we introduce InversionGNN framework, an effective yet sample-efficient dual-path graph neural network (GNN) for multi-objective drug discovery. In the direct prediction path of InversionGNN, we train the model for multi-property prediction to acquire knowledge of the optimal combination of functional groups. Then the learned chemical knowledge helps the inversion generation path to generate molecules with required properties. In order to decode the complex knowledge of multiple properties in the inversion path, we propose a gradient-based Pareto search method to balance conflicting properties and generate Pareto optimal molecules. Additionally, InversionGNN is able to search the full Pareto front approximately in discrete chemical space. Comprehensive experimental evaluations show that InversionGNN is both effective and sample-efficient in various discrete multi-objective settings including drug discovery.
title InversionGNN: A Dual Path Network for Multi-Property Molecular Optimization
topic Machine Learning
url https://arxiv.org/abs/2503.01488