arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.09382cs.LGstat.ML

GraphAF: a Flow-based Autoregressive Model for Molecular Graph Generation

Department of Computer Science, Peking University(北京大学计算机科学系);Shanghai Jiao Tong University(上海交通大学);Mila - Québec AI Institute(Mila - 魁北克人工智能研究所);Université de Montréal(蒙特利尔大学);HEC Montréal(蒙特利尔高等商学院);CIFAR AI Research Chair(CIFAR人工智能研究主席(注:该机构名称为职位头衔,非机构))

Chence Shi, Minkai Xu, Zhaocheng Zhu, Weinan Zhang, Ming Zhang, Jian Tang

英文摘要:

Molecular graph generation is a fundamental problem for drug discovery and has been attracting growing attention. The problem is challenging since it requires not only generating chemically valid molecular structures but also optimizing their chemical properties in the meantime. Inspired by the recent progress in deep generative models, in this paper we propose a flow-based autoregressive model for graph generation called GraphAF. GraphAF combines the advantages of both autoregressive and flow-based approaches and enjoys: (1) high model flexibility for data density estimation; (2) efficient parallel computation for training; (3) an iterative sampling process, which allows leveraging chemical domain knowledge for valency checking. Experimental results show that GraphAF is able to generate 68% chemically valid molecules even without chemical knowledge rules and 100% valid molecules with chemical rules. The training process of GraphAF is two times faster than the existing state-of-the-art approach GCPN. After fine-tuning the model for goal-directed property optimization with reinforcement learning, GraphAF achieves state-of-the-art performance on both chemical property optimization and constrained property optimization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑