arXiv:2512.18454cs.LGq-bio.QM2025-12被引 2

用扩散模型检测分子复合物的分布外数据,无需标签即可判断预测可靠性。

Out-of-Distribution Detection in Molecular Complexes via Diffusion Models for Irregular Graphs

  • 构建统一连续扩散模型,同时处理3D坐标与离散特征,实现无监督密度建模。
  • 在蛋白-配体复合物上验证,对未见蛋白家族的识别准确率显著提升。
  • 结合轨迹统计特征,提升检测灵敏度,适合几何深度学习场景应用。

预测型机器学习模型通常在分布内数据上表现良好,但在分布外(OOD)输入上性能下降。可靠部署需具备稳健的OOD检测能力,然而对于融合连续几何与离散身份、且无序构造的不规则3D图结构而言尤为困难。本文提出一种基于扩散模型的分子3D图数据概率性OOD检测框架,可在完全无监督条件下学习训练分布的密度。核心创新在于引入统一连续扩散过程,将类别身份嵌入连续空间并以交叉熵训练,通过后验均值插值解析获取扩散得分。由此生成单一自洽的概率流常微分方程(PF-ODE),输出每样本对数似然,提供分布偏移的严格典型性评分。在蛋白-配体复合物上验证,通过从训练中剔除完整蛋白家族构建严格OOD数据集,发现PF-ODE似然能有效识别被保留家族为分布外,并与独立结合亲和力模型(GEMS)的预测误差强相关,实现新复合物的预测可靠性事前估计。此外,多尺度PF-ODE轨迹统计量——包括路径扭曲度、流动刚度与向量场不稳定性——提供互补的OOD信息。联合建模这些轨迹特征可得高灵敏度检测器,优于仅依赖似然的基线,为几何深度学习提供无标签的OOD量化流程。

原文摘要 · Abstract (English)

Predictive machine learning models generally excel on in-distribution data, but their performance degrades on out-of-distribution (OOD) inputs. Reliable deployment therefore requires robust OOD detection, yet this is particularly challenging for irregular 3D graphs that combine continuous geometry with categorical identities and are unordered by construction. Here, we present a probabilistic OOD detection framework for complex 3D graph data built on a diffusion model that learns a density of the training distribution in a fully unsupervised manner. A key ingredient we introduce is a unified continuous diffusion over both 3D coordinates and discrete features: categorical identities are embedded in a continuous space and trained with cross-entropy, while the corresponding diffusion score is obtained analytically via posterior-mean interpolation from predicted class probabilities. This yields a single self-consistent probability-flow ODE (PF-ODE) that produces per-sample log-likelihoods, providing a principled typicality score for distribution shift. We validate the approach on protein-ligand complexes and construct strict OOD datasets by withholding entire protein families from training. PF-ODE likelihoods identify held-out families as OOD and correlate strongly with prediction errors of an independent binding-affinity model (GEMS), enabling a priori reliability estimates on new complexes. Beyond scalar likelihoods, we show that multi-scale PF-ODE trajectory statistics - including path tortuosity, flow stiffness, and vector-field instability - provide complementary OOD information. Modeling the joint distribution of these trajectory features yields a practical, high-sensitivity detector that improves separation over likelihood-only baselines, offering a label-free OOD quantification workflow for geometric deep learning.

分子生成扩散模型异常检测几何深度学习

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