scProtoTransformer: Scalable Reference Mapping Across Molecules, Cells and Donors

Abstract

The rapid accumulation of single-cell data has made it possible to comprehensively characterize biological systems at molecular, cellular, and donor levels. However, scalable reference mapping across different resolutions remains a major challenge in current research. Here, we propose scProtoTransformer, a prototype-based Transformer architecture designed to achieve scalable reference mapping across molecular, cell, and donor levels. scProtoTransformer introduces a knowledge-guided prototype tokenizer that projects gene expression into biologically interpretable pathway prototypes, effectively reducing numerical batch effects while preserving biological semantic patterns. Furthermore, by leveraging knowledge distilled from the foundation model and a dynamic supervised fine-tuning strategy, scProtoTransformer achieves robust biological representations with reduced pre-training requirements. Benchmark experiments across molecular, cell, and donor-level reference mapping demonstrate that scProtoTransformer delivers competitive or even superior performance compared with state-of-the-art approaches, while providing interpretability through biologically prototypes. Together, these results establish scProtoTransformer as a unified framework for scalable reference mapping, laying the foundation for systematic understanding from genes to individuals.