TrimNet: Learning molecular representation from triplet messages for biomedicine

Pengyong Li, Yuquan Li, Chang Yu Hsieh, Shengyu Zhang, Xianggen Liu, Huanxiang Liu, Sen Song, Xiaojun Yao

Research output: Contribution to journalArticlepeer-review

53 Citations (Scopus)

Abstract

Motivation: Computational methods accelerate drug discovery and play an important role in biomedicine, such as molecular property prediction and compound-protein interaction (CPI) identification. A key challenge is to learn useful molecular representation. In the early years, molecular properties are mainly calculated by quantum mechanics or predicted by traditional machine learning methods, which requires expert knowledge and is often labor-intensive. Nowadays, graph neural networks have received significant attention because of the powerful ability to learn representation from graph data. Nevertheless, current graph-based methods have some limitations that need to be addressed, such as large-scale parameters and insufficient bond information extraction. Results: In this study, we proposed a graph-based approach and employed a novel triplet message mechanism to learn molecular representation efficiently, named triplet message networks (TrimNet). We show that TrimNet can accurately complete multiple molecular representation learning tasks with significant parameter reduction, including the quantum properties, bioactivity, physiology and CPI prediction. In the experiments, TrimNet outperforms the previous state-of-the-art method by a significant margin on various datasets. Besides the few parameters and high prediction accuracy, TrimNet could focus on the atoms essential to the target properties, providing a clear interpretation of the prediction tasks. These advantages have established TrimNet as a powerful and useful computational tool in solving the challenging problem of molecular representation learning. Availability: The quantum and drug datasets are available on the website of MoleculeNet: http://moleculenet.ai. The source code is available in GitHub: https://github.com/yvquanli/trimnet. Contact: [email protected], [email protected]

Original languageEnglish
Article numberbbaa266
JournalBriefings in Bioinformatics
Volume22
Issue number4
DOIs
Publication statusPublished - 1 Jul 2021
Externally publishedYes

Keywords

  • compound-protein interaction
  • computational method
  • deep learning
  • graph neural networks
  • molecular property
  • molecular representation

Fingerprint

Dive into the research topics of 'TrimNet: Learning molecular representation from triplet messages for biomedicine'. Together they form a unique fingerprint.

Cite this