arXiv:2506.00936cs.LGcs.AI2025-06中稿 · KDD被引 2

通过对比学习提升药物代谢稳定性预测的准确性和可信度

Uncertainty-Aware Metabolic Stability Prediction with Dual-View Contrastive Learning

  • 引入双视角对比学习,融合原子与键级拓扑信息
  • 在多个数据集上超越现有方法,预测性能显著提升
  • 内置不确定性量化,适合需要可信预测的药物研发

精准预测分子代谢稳定性(MS)对药物研发至关重要,但受分子间复杂相互作用影响仍具挑战。尽管图神经网络(GNN)取得进展,现有方法存在两大缺陷:(1)原子中心的消息传递机制忽略键级拓扑特征,导致建模不完整;(2)缺乏可靠的不确定性量化。为此,我们提出TrustworthyMS,一种面向不确定性感知的代谢稳定性预测新框架。首先,分子图拓扑重映射机制通过边诱导特征传播同步原子-键相互作用,捕捉局部电子效应与全局构象约束。其次,对比拓扑-键对齐强制分子拓扑视图与键模式特征一致,增强表征鲁棒性。第三,采用贝塔-二项分布不确定性量化方法,在认知不确定性下实现预测与置信度校准。大量实验表明,TrustworthyMS在预测性能上优于当前最先进方法。

原文摘要 · Abstract (English)

Accurate prediction of molecular metabolic stability (MS) is critical for drug research and development but remains challenging due to the complex interplay of molecular interactions. Despite recent advances in graph neural networks (GNNs) for MS prediction, current approaches face two critical limitations: (1) incomplete molecular modeling due to atom-centric message-passing mechanisms that disregard bond-level topological features, and (2) prediction frameworks that lack reliable uncertainty quantification. To address these challenges, we propose TrustworthyMS, a novel contrastive learning framework designed for uncertainty-aware metabolic stability prediction. First, a molecular graph topology remapping mechanism synchronizes atom-bond interactions through edge-induced feature propagation, capturing both localized electronic effects and global conformational constraints. Second, contrastive topology-bond alignment enforces consistency between molecular topology views and bond patterns via feature alignment, enhancing representation robustness. Third, uncertainty modeling through Beta-Binomial uncertainty quantification enables simultaneous prediction and confidence calibration under epistemic uncertainty. Through extensive experiments, our results demonstrate that TrustworthyMS outperforms current state-of-the-art methods in terms of predictive performance.

代谢稳定性图神经网络不确定性量化对比学习

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