Back

Calibrated Variant Effect Prediction at the Residue Level Using Conditional Score Distributions

Passi, G.; Amittai, S.; Schneidman-Duhovny, D.

2025-11-26 genomics
10.1101/2025.11.24.690189 bioRxiv
Show abstract

Effective clinical use of variant effect prediction (VEP) requires models that are both accurate and well-calibrated. Calibration refers to a models ability to produce meaningful and reliable probability estimates. Here, we propose a practical path toward robust VEP calibration by calibrating at the residue-level rather than using global or per-protein schemes. We identify variant subgroups that benefit from targeted calibration and show that, while VEPs appear well calibrated on average, they remain markedly miscalibrated within these subgroups. Leveraging these insights, we develop RaCoon (Residue-aware Calibration via Conditional distributions), implemented on ESM1b, which provides multicalibrated and interpretable predictions across diverse variant subgroups and significantly improves performance across multiple benchmarks. Targeted residue-level calibration not only improves overall calibration but, for most models, also yields gains in global AUROC. Specifically, RaCoon increases AUCROC from 0.912 to 0.924. Our calibration strategy, guided by model-specific feature distributions, is readily transferable to other VEPs.

Matching journals

The top 5 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.