Keeping SCORE enables interpretable uncertainty-aware classification from diffusion models for genomics
Benjamin Kuznets-SpeckiD, Jaekwon JungiD, Pornchanan Pholraksa, Adrianne Zhong, Leon SchwartziD, Ekta PrashnaniiD, Suriyanarayanan VaikuntanathaniD, Yogesh GoyaliD
Abstract
Classifying cellular states from high-dimensional molecular and genomic measurements requires methods that provide not only accurate predictions but also calibrated uncertainty and interpretability. Current nonlinear classifiers offer accuracy but often lack uncertainty quantification and mechanistic insights into the features that matter most. We introduce Keeping SCORE, a framework that transforms conditional diffusion models into probabilistic engines for classification and regression by computing exact likelihoods along stochastic noising trajectories. We first benchmark Keeping SCORE on image recognition tasks (handwritten digits, natural photos). We then apply Keeping SCORE to single-cell transcriptomics across a 22-million-cell atlas, classifying 164 cell types with accuracy matching or exceeding state-of-the-art methods, while uniquely providing posterior probability estimates and prediction confidence. For genetic perturbation mapping across 100 CRISPRi conditions in a multi-study Perturb-seq dataset, our approach again matches or surpasses discriminative baselines, with feature-level attributions identifying which genomic features drive each decision. Applied to large-scale protein sequence data, our framework accurately regresses mutational stability effects, attributing them quantitatively to positions along the input sequence. Keeping SCORE requires no retraining or architectural changes to existing diffusion models, providing portable, interpretable, and uncertainty-aware predictions for biological discovery.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 0
- Organism
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Discrete Diffusion for Single-Cell Gene Expression Modeling
Sanjukta Bhattacharya, Christian Gensbigler, … Jonathan Gill Lees·bioRxiv·Feb 20, 2026
- PertDiffBench: Benchmarking Diffusion Models for Single-Cell Perturbation Response Prediction
zj Song, Yujia Xiang, … Linhai Xie·bioRxiv·Jun 13, 2026
- A General Single-Cell Analysis Framework via Conditional Diffusion Generative Models
Wenzhuo Tang, Renming Liu, … Jiliang Tang·bioRxiv·Oct 13, 2023
- Predicting the unseen: a diffusion-based debiasing framework for transcriptional response prediction at single-cell resolution
Ergan Shang, Yuting Wei, Kathryn Roeder·Proceedings of the National Academy of SciencesPeer-reviewed·Dec 26, 2025
- Auditing Confidence-Calibrated Benchmarking of Single-Cell Perturbation Models
Manuel Gallone, Shyam Ravidath, Srihith Madireddy·bioRxiv·Oct 19, 2026
- Improving genomic language model reliability under distribution shift
Gavin Hearne, Mohammad S. Refahi, … GAIL L. ROSEN·bioRxiv·Mar 20, 2026