用参数插值流统一分子生成中的连续与离散变量,提升药物设计精度。
MolPIF: A Parameter Interpolation Flow Model for Molecule Generation
- 在参数空间插值,统一处理原子坐标与原子类型
- 在CrossDocked2020上超越基线,提升结合亲和力与几何保真度
- 支持灵活先验选择,适合药物先导优化场景
基于结构的药物设计(SBDD)借助深度生成模型取得进展,但连续原子坐标与离散原子类型之间的融合仍具挑战。现有方法如扩散模型和流匹配模型难以统一异构模态,常依赖分离策略或对离散变量使用不合适的欧氏度量。为此,我们提出MolPIF,一种参数插值流模型,通过在参数空间中插值分布,理论上实现连续坐标的Wasserstein-2最优传输,并为离散原子类型建立Fisher-Rao测地线。同时引入几何增强学习策略以更好捕捉原子上下文。在CrossDocked2020数据集上的大量实验表明,MolPIF在结合亲和力、化学有效性、几何保真度及化学空间覆盖方面均优于基线。此外,该模型在先导化合物优化中表现出强适应性,并支持多种先验分布(如拉普拉斯分布),为SBDD建立稳健范式。代码开源:https://github.com/BLEACH366/MolPIF。
原文摘要 · Abstract (English)
Motivation: Structure-based drug design (SBDD) has advanced with deep generative models, but bridging the gap between continuous atomic coordinates and discrete atom types remains a challenge. Current approaches, such as diffusion and flow matching models, often fail to unify these heterogeneous modalities, relying on separate strategies or ill-fitting Euclidean metrics for discrete variables. This lack of a consistent framework limits generative models' ability to capture the geometric and chemical structure of protein-ligand complexes. Results: We present MolPIF, a parameter interpolation flow mechanism designed to unify the generation of continuous and discrete molecular variables. Unlike traditional flow models that operate in sample space, MolPIF interpolates between distributions in the parameter space, theoretically recovering Wasserstein-2 optimal transport for continuous coordinates and establishing Fisher-Rao geodesics for discrete atom types. We further incorporate a geometry-enhanced learning strategy to improve the capture of atomic contexts. Extensive evaluations on the CrossDocked2020 dataset demonstrate that MolPIF outperforms baselines in binding affinity, chemical validity, geometric fidelity and chemical space coverage. Additionally, MolPIF exhibits versatility in lead optimization and offers flexible prior distribution selection (such as Laplace), establishing a robust paradigm for SBDD. Availability: Source code is freely available at https://github.com/BLEACH366/MolPIF. Supplementary information: Supplementary data are available at Bioinformatics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。