IPM: Information Lossless Pre-training Strategy for Molecular Property Prediction
Given the pivotal role of molecular property prediction in drug development and material science, graph self-supervised learning has been implemented in molecular representation learning to compensate for the shortage of labeled molecules. However, current proposed methods often focus on designing data augmentation schemes and leveraging domain knowledge to improve performance, which inevitably leads to molecular semantics loss and limited generalization capability. To the end, we propose IPM, an Information lossless Pretraining strategy for Molecular property prediction that leverages the information of both the original graph and line graph of molecules. Specifically, by contrasting the given graph with the corresponding line graph, the graph encoder can fully learn the generic molecular semantic representation without profound domain knowledge. We also design a new message-passing scheme that retains information consistency during message passing between two kinds of graphs. Additionally, we present two graph contrastive losses for performance fixing and over-smoothing prevention during the learning process. Experimental results on multiple regression tasks for molecular property prediction demonstrate the effectiveness of IPM against state-of-the-art (SOTA) methods.
Authors
- Haitao Yuan (ORCID: https://orcid.org/0000-0001-6721-065X)
- He Zhu (ORCID: https://orcid.org/0000-0001-9606-150X)
- Ruomei Liu (ORCID: https://orcid.org/0009-0002-6533-6229)
- Junran Wu (ORCID: https://orcid.org/0000-0001-6742-4332)
- Yue Hou (ORCID: https://orcid.org/0009-0001-4806-624X)
- Xingyu Peng (ORCID: https://orcid.org/0000-0002-4864-4003)
- Ke Xu (ORCID: https://orcid.org/0000-0001-5447-8008)
- Shangzhe Li (ORCID: https://orcid.org/0000-0001-8570-2252)
Institutions
- Central University of Finance and Economics (CN)
- Nanyang Technological University (SG)
- Beihang University (CN)
Publication Details
- Published
- 2024-12-03
- DOI
- https://doi.org/10.1109/bibm62325.2024.10822412
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.00