MOLGAECL: Molecular Graph Contrastive Learning via Graph Auto-Encoder Pretraining and Fine-Tuning Based on Drug-Drug Interaction Prediction.

Journal: Journal of chemical information and modeling
PMID:

Abstract

Drug-drug interactions influence drug efficacy and patient prognosis, providing substantial research value. Some existing methods struggle with the challenges posed by sparse networks or lack the capability to integrate data from multiple sources. In this study, we propose MOLGAECL, a novel approach based on graph autoencoder pretraining and molecular graph contrastive learning. Initially, a large number of unlabeled molecular graphs are pretrained using a graph autoencoder, where graph contrastive learning is applied for more accurate representation of the drugs. Subsequently, a full-parameter fine-tuning is performed on different data sets to adapt the model for drug interaction-related prediction tasks. To assess the effectiveness of MOLGAECL, comparison experiments with state-of-the-art methods, fine-tuning comparison experiments, and parameter sensitivity analysis are conducted. Extensive experimental results demonstrate the superior performance of MOLGAECL. Specifically, MOLGAECL achieves an average increase of 6.13% in accuracy, 6.14% in AUROC, and 8.16% in AUPRC across all data sets.

Authors

  • Yu Li
    Department of Public Health, Shihezi University School of Medicine, 832000, China.
  • Lin-Xuan Hou
    School of Computer Science, Northwestern Polytechnical University, Xi'an710129, China.
  • Hai-Cheng Yi
    Xinjiang Technical Institute of Physics and Chemistry, Chinese Academy of Sciences, Urumqi 830011, China. yihaicheng17@mails.ucas.ac.cn.
  • Zhu-Hong You
    Xinjiang Technical Institute of Physics and Chemistry, Chinese Academy of Sciences, Urumqi, 830011, China. zhuhongyou@ms.xjb.ac.cn.
  • Shi-Hong Chen
    School of Computer Science and Artificial Intelligence, Changzhou University, Changzhou 213164, China.
  • Jia Zheng
    School of Advanced Manufacturing Engineering, Chongqing University of Posts and Telecommunications, Chongqing 400065, China.
  • Yang Yuan
  • Cheng-Gang Mi
    Foreign Language and Literature Institute, Xi'an International Studies University, Xi'an710129, China.