arXiv:2605.24841cs.LG2026-05

通过解码器耦合漂移实现高效分子属性条件生成

DriftingMol: Decoder-Coupled Drift for One-Pass Property-Conditional Molecular Generation

论文配图:DriftingMol: Decoder-Coupled Drift for One-Pass Property-Conditional Molecular Generation
图 1 · 摘自论文原文
  • 用冻结的VAE解码器特征作为漂移信号源
  • 在ZINC250K上达0.510的QED相关性,唯一性94.7%
  • 仅需一次生成器评估,适合快速分子设计

属性条件分子生成需在低采样成本下生成有效且多样的分子,并响应连续目标值。本文提出DriftingMol,一种两阶段框架,将漂移模型适配至SELFIES分子潜在空间。使用冻结的SELFIES beta-VAE提供潜在空间,其解码器的隐藏表示作为漂移特征映射。在解码器耦合漂移中,解码器权重固定,但漂移梯度通过解码器特征映射反向传播至DiT生成器,诱导与分子解码对齐的回拉度量。在ZINC250K上,默认设置实现QED Spearman相关性0.493,唯一性94.7%;最强的解码器耦合条件达0.510。在匹配协议的四属性条件设置下,解码器耦合漂移达到最高均值Spearman相关性0.598。在15种控制变体中,保留通过解码器特征的梯度路径的模型表现优于潜在空间、随机特征和外部特征漂移变体;而解码器断开或停止梯度的控制则导致近零的QED相关性和极低唯一性。结果表明,解码器耦合漂移是低成本属性偏置分子生成的有效机制,仅需一次生成器评估和一次冻结解码器遍历。

原文摘要 · Abstract (English)

Property-conditional molecular generation should produce valid, diverse molecules while responding to continuous target values at low sampling cost. We introduce DriftingMol, a two-stage framework that adapts drifting models to a SELFIES latent molecular space. A frozen SELFIES beta-VAE provides the latent space, and the hidden representation of its decoder serves as the drift feature map. In decoder-coupled drift, decoder weights remain fixed, but drift gradients are backpropagated through the decoder feature map to a DiT generator, inducing a pullback metric aligned with molecular decoding. On ZINC250K, the default setting achieves QED Spearman correlation 0.493 with 94.7% uniqueness, while the strongest decoder-coupled condition reaches 0.510. Under protocol-matched four-property conditioning, decoder-coupled drift reaches mean Spearman correlation up to 0.598. Across 15 controlled variants, models that preserve the gradient path through decoder features achieve higher correlations than the tested latent-space, random-feature, and external-feature drift variants, while detached or stop-gradient decoder controls yield near-zero QED correlation and very low uniqueness. These results indicate that decoder-coupled drift is a useful low-cost mechanism for property-biased molecular generation, requiring one generator evaluation and one frozen decoder pass.

分子生成扩散模型属性控制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。