Manifold-constrained nucleus-level denoising diffusion model for structure-based drug design

Abstract

AI models have shown great potential in structure-based drug design, generating ligands with high binding affinities. However, existing models have often overlooked a crucial physical prior: Atoms must maintain a minimum pairwise distance to avoid atomic collision, a phenomenon governed by the balance of attractive and repulsive forces. To mitigate such atomic collisions, we propose NucleusDiff. It enforces spatial distance constraints between atomic nuclei and auxiliary mesh points placed on a spherical surface around each atom, approximating van der Waals boundaries to reduce atomic collisions. We quantitatively evaluate NucleusDiff using the CrossDocked2020 dataset and a COVID-19 therapeutic target, demonstrating that NucleusDiff reduces collision rate by up to 100.00% and enhances binding affinity by up to 22.16%, surpassing state-of-the-art models for structure-based drug design. We also provide qualitative analysis through manifold sampling, visually confirming the effectiveness of NucleusDiff in reducing atomic collisions and improving binding affinities.