TrimNet: learning molecular representation from triplet messages for biomedicine
Motivation: Computational methods accelerate drug discovery and play an important role in biomedicine, such as molecular property prediction and compound-protein interaction 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 that 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 compound-protein interaction (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.
PDFCode
Task | Dataset | Model | Metric Name | Metric Value | Global Rank | Benchmark |
---|---|---|---|---|---|---|
Drug Discovery | BACE | TrimNet | AUC | 0.878 | # 1 | |
Drug Discovery | ClinTox | TrimNet | AUC | 0.948 | # 1 | |
Drug Discovery | HIV dataset | TrimNet | AUC | 0.804 | # 3 | |
Drug Discovery | MUV | TrimNet | AUC | 0.851 | # 1 | |
Drug Discovery | Tox21 | TrimNet | AUC | 0.860 | # 5 | |
Drug Discovery | ToxCast | TrimNet | AUC | 0.777 | # 1 |