English

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

Machine Learning 2025-06-03 v1 Artificial Intelligence Quantitative Methods

Abstract

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.

Keywords

Cite

@article{arxiv.2506.00936,
  title  = {Uncertainty-Aware Metabolic Stability Prediction with Dual-View Contrastive Learning},
  author = {Peijin Guo and Minghui Li and Hewen Pan and Bowen Chen and Yang Wu and Zikang Guo and Leo Yu Zhang and Shengshan Hu and Shengqing Hu},
  journal= {arXiv preprint arXiv:2506.00936},
  year   = {2025}
}

Comments

This manuscript has been accepted for publication at ECML-PKDD 2025. The final version will be published in the conference proceedings

R2 v1 2026-07-01T02:53:00.796Z