返回
An efficient curriculum learning-based strategy for molecular graph learning
DOI:10.1093/bib/bbac099.png)
摘要
En 中文
Computational methods have been widely applied to resolve various core issues in drug discovery, such as molecular property prediction. In recent years, a data-driven computational method-deep learning had achieved a number of impressive successes in various domains. In drug discovery, graph neural networks (GNNs) take molecular graph data as input and learn graph-level representations in non-Euclidean space. An enormous amount of well-performed GNNs have been proposed for molecular graph learning. Meanwhile, efficient use of molecular data during training process, however, has not been paid enough attention. Curriculum learning (CL) is proposed as a training strategy by rearranging training queue based on calculated samples' difficulties, yet the effectiveness of CL method has not been determined in molecular graph learning. In this study, inspired by chemical domain knowledge and task prior information, we proposed a novel CL-based training strategy to improve the training efficiency of molecular graph learning, called CurrMG. Consisting of a difficulty measurer and a training scheduler, CurrMG is designed as a plug-and-play module, which is model-independent and easy-to-use on molecular data. Extensive experiments demonstrated that molecular graph learning models could benefit from CurrMG and gain noticeable improvement on five GNN models and eight molecular property prediction tasks (overall improvement is 4.08%). We further observed CurrMG's encouraging potential in resource-constrained molecular property prediction. These results indicate that CurrMG can be used as a reliable and efficient training strategy for molecular graph learning.
Keyword:
curriculum learning
training strategy
molecular graph learning
molecular property prediction
drug discovery
期刊
IF:
7.7
论文数:
5.8K
被引数:
2.7W
机构
引用论文
Evidential Deep Learning for Guided Molecular Property Prediction and Discovery用于引导分子特性预测和发现的证据深度学习
ACS CENTRAL SCIENCE
IF10.4
MG-BERT: leveraging unsupervised atomic representation learning for molecular property predictionMg-bert: 利用无监督原子表示学习进行分子性质预测

