计算机科学
不确定度量化
机器学习
适用范围
回归
人工智能
分位数
数据挖掘
可靠性(半导体)
数量结构-活动关系
数学
统计
量子力学
物理
功率(物理)
作者
Peiyao Li,Lan Hua,Zhechao Ma,Wenbo Hu,Ye Liu,Jun Zhu
标识
DOI:10.1021/acs.jcim.4c01139
摘要
Drug discovery and development is a complex and costly process, with a substantial portion of the expense dedicated to characterizing the absorption, distribution, metabolism, excretion, and toxicity (ADMET) properties of new drug candidates. While the advent of deep learning and molecular graph neural networks (GNNs) has significantly enhanced in silico ADMET prediction capabilities, reliably quantifying prediction uncertainty remains a critical challenge. The performance of GNNs is influenced by both the volume and the quality of the data. Hence, determining the reliability and extent of a prediction is as crucial as achieving accurate predictions, especially for out-of-domain (OoD) compounds. This paper introduces a novel GNN model called conformalized fusion regression (CFR). CFR combined a GNN model with a joint mean-quantile regression loss and an ensemble-based conformal prediction (CP) method. Through rigorous evaluation across various ADMET tasks, we demonstrate that our framework provides accurate predictions, reliable probability calibration, and high-quality prediction intervals, outperforming existing uncertainty quantification methods.
科研通智能强力驱动
Strongly Powered by AbleSci AI