Conformal Prediction for Molecular Properties under Label Shift
Not on the map. Why?
Abstract (arXiv)
Drug discovery and development underpins healthcare but remains costly and failure-prone. A critical bottleneck lies in predicting molecular properties such as solubility, potency, and toxicity, which directly determine whether a candidate can advance from preclinical to clinical trials. Artificial Intelligence (AI) has accelerated this process, yet its reliability is often undermined by distribution shift, as experimental conditions frequently diverge from training data. In addition, conventional point predictions provide only single-value estimates, offering limited guidance for high-stakes experimental design. We address these challenges with a conformal prediction framework tailored to label shift. By weighting conformal scores using marginal label probability ratios, our method produces statistically rigorous prediction intervals without retraining. This enables robust uncertainty quantification even when property distributions drift, directly tackling one of the most pervasive obstacles to applying AI in real-world drug development. By moving beyond accuracy alone to provide actionable confidence measures, our approach enhances the trustworthiness of AI-driven predictions. This further aligns predictive modeling with regulatory demands for transparency and uncertainty reporting and ultimately supports more reliable decision-making in billion-dollar development pipelines.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 16 of its 34 references that are in the map. First posted on arXiv 18 Aug 2026.
Builds on
16 of 34 references.
References · 16
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,893
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 792
- 2019Conformalized Quantile Regressioncited by 629
- 2019Conformal Prediction Under Covariate Shiftcited by 519
- 2018Detecting and Correcting for Label Shift with Black Box Predictorscited by 366
- 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 337
- 2021Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Developmentcited by 264
- 2021Large-Scale Chemical Language Representations Capture Molecular Structure and Propertiescited by 206
- 2022Translation between Molecules and Natural Languagecited by 177
- 2019Regularized Learning for Domain Adaptation under Label Shiftscited by 120
- 2022C-Mixup: Improving Generalization in Regressioncited by 57
- 2023PAC Prediction Sets Under Label Shiftcited by 21
4 more
- 2023Conformal Drug Property Prediction with Density Estimation under Covariate Shiftcited by 13
- 2023Drug Discovery under Covariate Shift with Domain-Informed Prior Distributions over Functionscited by 10
- 2023Instructor-inspired Machine Learning for Robust Molecular Property Predictioncited by 8
- 2025Robust Molecular Property Prediction via Densifying Scarce Labeled Datacited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.