Back

MOL-AE: Auto-Encoder Based Molecular Representation Learning With 3D Cloze Test Objective

Yang, J.; Zheng, K.; Long, S.; Nie, Z.; Zhang, M.; Dai, X.; Ma, W.-Y.; Zhou, H.

2024-04-15 bioinformatics
10.1101/2024.04.13.589331 bioRxiv
Show abstract

3D molecular representation learning has gained tremendous interest and achieved promising performance in various downstream tasks. A series of recent approaches follow a prevalent framework: an encoder-only model coupled with a coordinate denoising objective. However, through a series of analytical experiments, we prove that the encoderonly model with coordinate denoising objective exhibits inconsistency between pre-training and downstream objectives, as well as issues with disrupted atomic identifiers. To address these two issues, we propose MO_SCPLOWOLC_SCPLOW-AE for molecular representation learning, an auto-encoder model using positional encoding as atomic identifiers. We also propose a new training objective named 3D Cloze Test to make the model learn better atom spatial relationships from real molecular substructures. Empirical results demonstrate that MO_SCPLOWOLC_SCPLOW-AE achieves a large margin performance gain compared to the current state-of-the-art 3D molecular modeling approach. The source codes of MO_SCPLOWOLC_SCPLOW-AE are publicly available at https://github.com/yjwtheonly/MolAE.

Matching journals

The top 3 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.