arrow
Return

AGDIFF: Attention-Enhanced Diffusion for Molecular Geometry Prediction

delete2025-02-11
delete0
PRE
AI
A
André Brasil Vieira Wyzykowski
F
Fatemeh Fathi Niazi
A
Alex Dickson *
DOI:10.1021/acs.jcim.4c01896delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Accurate prediction of molecular geometries is crucial for drug discovery and materials science. Existing fast conformer prediction algorithms often rely on approximate empirical energy functions, resulting in low accuracy. More accurate methods like ab initio molecular dynamics and Markov chain Monte Carlo can be computationally expensive due to the need for evaluating quantum mechanical energy functions. To address this, we introduce AGDIFF, a novel machine learning framework that utilizes diffusion models for efficient and accurate molecular structure prediction. AGDIFF extends previous models (such as GeoDiff) by enhancing the global, local, and edge encoders with attention mechanisms, an improved SchNet architecture, batch normalization, and feature expansion techniques. AGDIFF outperforms GeoDiff on both the GEOM-QM9 and GEOM-Drugs data sets. For GEOM-QM9, with a threshold (delta) of 0.5 & Aring;, AGDIFF achieves a mean COV-R of 93.08% and a mean MAT-R of 0.1965 & Aring;. On the more complex GEOM-Drugs data set, using delta = 1.25 & Aring;, AGDIFF attains a median COV-R of 100.00% and a mean MAT-R of 0.8237 & Aring;. These findings demonstrate AGDIFF's potential to advance molecular modeling techniques, enabling more efficient and accurate prediction of molecular geometries, thus contributing to computational chemistry, drug discovery, and materials design. https://github.com/ADicksonLab/AGDIFF
Keywords:
DESCRIPTORS
MECHANICS

Journal

Journal of Chemical Information and Modeling cover
Journal of Chemical Information and Modeling
IF:
5.3
Papers:
9.1K
Citations:
4.0W

Organization

M
michigan state university
Scholars:
3.6W
Papers: 3.2W
Citations: 44