CurrMG

CurrMG applies curriculum learning to optimize graph neural network (GNN) training for molecular graph learning, improving molecular property prediction in drug discovery.


Key Features:

  • Curriculum Learning Strategy: Implements curriculum learning (CL) to reorder training samples by calculated difficulty using domain-specific chemical knowledge and task-related priors.
  • Model-independent Module: Functions as a plug-and-play, model-independent module for integration with various molecular graph learning frameworks and GNNs.
  • Difficulty Measurer: Quantifies each training sample's difficulty based on chemical knowledge and task-related information.
  • Training Scheduler: Arranges the sequence of training samples from easier to harder to optimize GNN learning trajectories.

Scientific Applications:

  • Molecular Property Prediction: Enhances GNN training for predicting molecular properties relevant to drug discovery.
  • Graph-level Representation Learning: Improves learning of graph-level representations from non-Euclidean molecular data using GNNs.
  • Benchmark Performance: Demonstrated an overall improvement of 4.08% across five GNN models and eight molecular property prediction tasks.
  • Resource-constrained Training: Provides efficiency gains that aid model training in resource-constrained computational environments.

Methodology:

CurrMG computes sample difficulty using domain-specific chemical knowledge and task-related priors and employs a training scheduler to present data from easy to hard, thereby rearranging data presentation to optimize GNN training.

Topics

Details

License:
Not licensed
Tool Type:
command-line tool, workflow
Programming Languages:
Python
Added:
7/20/2022
Last Updated:
7/20/2022

Operations

Publications

Gu Y, Zheng S, Xu Z, Yin Q, Li L, Li J. An efficient curriculum learning-based strategy for molecular graph learning. Briefings in Bioinformatics. 2022;23(3). doi:10.1093/bib/bbac099. PMID:35368074.

PMID: 35368074
Funding: - Chinese Academy of Medical Sciences: 2021-I2M-1-056 - National Key Research and Development Program of China: 2016YFC0901901, 2017YFC0907503 - National Natural Science Foundation of China: 81601573