arXiv:2502.03982cs.LG2025-02被引 4

真实药理数据随时间分布漂移,影响模型不确定性估计可靠性。

Temporal Distribution Shift in Real-World Pharmaceutical Data: Implications for Uncertainty Quantification in QSAR Models

  • 用真实药企数据检验不同不确定性估计算法在时间分布漂移下的表现。
  • 发现标签空间和描述符空间均存在显著时序变化,与实验类型相关。
  • 主流不确定性方法在明显分布漂移下性能下降,需改进鲁棒性。

定量构效关系(QSAR)模型的预测不确定性估计可加速药物发现,通过识别有前景的实验并高效分配资源。尽管已有多种计算工具用于估计机器学习模型的不确定性,但非独立同分布(i.i.d.)条件下的偏差已被证明会降低这些方法的性能。本文利用真实世界药理学数据集,系统评估不确定性估计方法在真实时间分布漂移场景下的表现。研究考察了基于集成和贝叶斯的方法,并在真实环境中分析标签空间与描述符空间的分布变化及其对不确定性估计能力的影响。结果表明,标签和描述符空间均存在显著的时间漂移,且漂移程度与检测方法类型密切相关。此外,明显的分布漂移会严重削弱常用不确定性估计方法的性能。本研究揭示了在真实数据引入的时间分布漂移下,保持不确定性估计可靠性的挑战。

原文摘要 · Abstract (English)

The estimation of uncertainties associated with predictions from quantitative structure-activity relationship (QSAR) models can accelerate the drug discovery process by identifying promising experiments and allowing an efficient allocation of resources. Several computational tools exist that estimate the predictive uncertainty in machine learning models. However, deviations from the i.i.d. setting have been shown to impair the performance of these uncertainty quantification methods. We use a real-world pharmaceutical dataset to address the pressing need for a comprehensive, large-scale evaluation of uncertainty estimation methods in the context of realistic distribution shifts over time. We investigate the performance of several uncertainty estimation methods, including ensemble-based and Bayesian approaches. Furthermore, we use this real-world setting to systematically assess the distribution shifts in label and descriptor space and their impact on the capability of the uncertainty estimation methods. Our study reveals significant shifts over time in both label and descriptor space and a clear connection between the magnitude of the shift and the nature of the assay. Moreover, we show that pronounced distribution shifts impair the performance of popular uncertainty estimation methods used in QSAR models. This work highlights the challenges of identifying uncertainty quantification methods that remain reliable under distribution shifts introduced by real-world data.

QSAR不确定性分布漂移药物发现

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