Skip to yearly menu bar Skip to main content


Poster

Equivariant Masked Position Prediction for Efficient Molecular Representation

Junyi An · Chao Qu · Yun-Fei Shi · XinHao Liu · Qianwei Tang · Fenglei Cao · Yuan Qi

Hall 3 + Hall 2B #329
[ ]
Fri 25 Apr midnight PDT — 2:30 a.m. PDT

Abstract:

Graph neural networks (GNNs) have shown considerable promise in computational chemistry. However, the limited availability of molecular data raises concerns regarding GNNs' ability to effectively capture the fundamental principles of physics and chemistry, which constrains their generalization capabilities. To address this challenge, we introduce a novel self-supervised approach termed Equivariant Masked Position Prediction (EMPP), grounded in intramolecular potential and force theory. Unlike conventional attribute masking techniques, EMPP formulates a nuanced position prediction task that is more well-defined and enhances the learning of quantum mechanical features. EMPP also bypasses the approximation of the Gaussian mixture distribution commonly used in denoising methods, allowing for more accurate acquisition of physical properties. Experimental results indicate that EMPP significantly enhances performance of advanced molecular architectures, surpassing state-of-the-art self-supervised approaches. Our code is released in https://github.com/ajy112/EMPP.

Live content is unavailable. Log in and register to view live content