vix.ing · top · new · best · stats · spec

Bridging the Gap between Learning and Inference for Diffusion-Based Molecule Generation

2024/11/08 by Peidong Liu, Wenbo Zhang, Liu, Peidong +7
Biochemistry, Genetics and Molecular Biology · Engineering · Medicine · #Advanced Biosensing Techniques and Applications #FOS: Computer and information sciences #Machine Learning (cs.LG) #Microfluidic and Capillary Electrophoresis Applications #Monoclonal and Polyclonal Antibodies Research

paper · pdf · doi:10.48550/arxiv.2411.05472

openalex publication_date 2024/11/08 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

The paradigm shift toward structure-driven molecule generation has been propelled by advances in deep generative models, such as variational auto-encoders and diffusion models. However, these generative models for molecular design remain constrained by exposure bias, error accumulation, and suboptimal handling of activity cliffs. Here, we introduce DiffGap, a diffusion-based framework that integrates adaptive sampling and pseudo-molecule estimation to bridge the gap between training objectives and inference dynamics in 3D molecule generation. By dynamically aligning intermediate denoising steps with realistic generation trajectories, DiffGap enables the diffusion model to adapt to input biases in advance during the training phase. A temperature annealing module further controls the aligning strength of the adaptive alignment process, ensuring stable learning of the data distribution. Evaluated on the CrossDocked2020 benchmark, DiffGap outperforms existing methods in docking scores and binding affinity, demonstrating superior fidelity in generating drug-like molecules. Our work establishes a principled approach to harmonize generative training with inference mechanics, offering a robust computational toolkit for accelerating structure-based therapeutic discovery. The source code of DiffGap is available at https://github.com/neusymlab/DiffGap.

Related