Structure-Based Drug Design via Diffusion Models Guided by Non-Differentiable Metrics

May 1, 2025·
YiFan Cai
YiFan Cai
· 1 min read
Abstract
This work introduces a diffusion-guided molecular generation approach that seamlessly incorporates non-differentiable metrics—such as empirical property scores or domain heuristics—into the sampling process. Unlike traditional gradient-based optimization, our method leverages a conditional diffusion prior and a gradient-free feedback loop to navigate chemical space effectively, producing candidate molecules with enhanced validity and alignment with multiple design objectives. This project highlights how generative diffusion frameworks can be adapted for realistic drug design scenarios where scoring functions are irregular, discontinuous, or otherwise non-differentiable.
Type
Publication
Undergraduate Thesis
publications

Method Overview

Zero-order Optimization via Gradient Approximation (SPSA)
Zero-order Optimization via Gradient Approximation (SPSA)
Adaptive Trajectory Pruning (ATP)
Adaptive Trajectory Pruning (ATP)
YiFan Cai
Authors
YiFan Cai (he/him)
Graduate student
I am a graduate student in Systems Engineering at the University of Pennsylvania, with a B.S. in Computer Science from ShanghaiTech University. I have research experience in diffusion models, computer vision, robotic manipulation, and molecular drug design. My current interests focus on world models, computer vision, and artificial intelligence.