SGEDiff: a subgraph-enriched diffusion model for structure-based 3D molecular generation

Structure-based molecular generation is an emerging approach in computer-aided drug discovery, enabling the design of compounds that that complement the three-dimensional structure of target proteins. However, most diffusion-based 3D molecular generative models still face several limitations, such as imbalanced protein-ligand representations or reliance on predefined binding pockets. To address these limitations, we propose SGEDiff, a novel subgraph enriched generative framework for 3D molecule generation. Our model hierarchically fuses subgraph and global graph representations to capture both local binding patterns and key structural features of protein pockets. Furthermore, an integrated pocket prediction module identifies binding regions in unseen proteins, eliminating reliance on predefined pocket coordinates. Experimental results show that SGEDiff outperforms baseline diffusion-based methods in generating high-affinity molecules across diverse targets. Moreover, practical applications in de novo drug design demonstrate improved success rates in generating compounds for novel protein targets, underscoring its potential to advance structure-based drug discovery.