Back

scUnify: A Unified Framework for Zero-shot Inference of Single-Cell Foundation Models

KIM, D.; Jeong, K.; KIM, K.

2026-03-03 bioinformatics
10.64898/2026.03.01.708392 bioRxiv
Show abstract

Foundation models (FMs) pre-trained on large-scale single-cell RNA sequencing (scRNA-seq) data provide powerful cell embeddings, but their practical usability and systematic comparison are limited by model-specific environments, preprocessing pipelines, and execution procedures. To address these challenges, we introduce scUnify, a unified zero-shot inference framework for single-cell foundation models. scUnify accepts a standard AnnData object and automatically manages environment isolation, preprocessing, and tokenization through a registry-based modular design. It employs a hierarchical distributed inference strategy that combines Ray-based task scheduling with multi-GPU data-parallel execution via HuggingFace Accelerate, enabling scalable inference on datasets containing up to one million cells. In addition, built-in integration of scIB and scGraph metrics enables standardized cross-model embedding evaluation within a single workflow. Benchmarking results demonstrate substantial reductions in inference time compared with the original model implementations, while preserving embedding quality and achieving near-linear multi-GPU scaling. scUnify is implemented in Python and is publicly available at https://github.com/DHKim327/scUnify.

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.