论文标题

GEM-2:下一代分子属性预测网络,通过对全身相互作用进行建模

GEM-2: Next Generation Molecular Property Prediction Network by Modeling Full-range Many-body Interactions

论文作者

Liu, Lihang, He, Donglong, Fang, Xiaomin, Zhang, Shanzhuo, Wang, Fan, He, Jingzhou, Wu, Hua

论文摘要

分子财产预测是药物和材料行业的基本任务。从物理上讲,分子的特性取决于其自身的电子结构,这是一种量子多体系统,可以通过Schr“ Odinger方程式。同时,深度学习方法也证明了它们在分子财产预测任务中的能力。受经典计算化学方法的启发,我们设计了一种新颖的方法,即GEM-2,该方法全面考虑了分子中的全及多体相互作用。多个轨道被用来建模具有不同顺序的多体之间的全范围相互作用,并且设计了一种新型的轴向注意机制,旨在近似计算成本较低的全范围相互作用建模。广泛的实验证明了GEM-2在量子化学和药物发现任务中的多种基线方法的压倒性优势。消融研究还验证了全范围多体相互作用的有效性。

Molecular property prediction is a fundamental task in the drug and material industries. Physically, the properties of a molecule are determined by its own electronic structure, which is a quantum many-body system and can be exactly described by the Schr"odinger equation. Full-range many-body interactions between electrons have been proven effective in obtaining an accurate solution of the Schr"odinger equation by classical computational chemistry methods, although modeling such interactions consumes an expensive computational cost. Meanwhile, deep learning methods have also demonstrated their competence in molecular property prediction tasks. Inspired by the classical computational chemistry methods, we design a novel method, namely GEM-2, which comprehensively considers full-range many-body interactions in molecules. Multiple tracks are utilized to model the full-range interactions between the many-bodies with different orders, and a novel axial attention mechanism is designed to approximate the full-range interaction modeling with much lower computational cost. Extensive experiments demonstrate the overwhelming superiority of GEM-2 over multiple baseline methods in quantum chemistry and drug discovery tasks. The ablation studies also verify the effectiveness of the full-range many-body interactions.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源