Abstract:Over the past decade, single-cell transcriptomic technologies have experienced remarkable advancements, enabling the simultaneous profiling of gene expressions across thousands of individual cells. Cell type identification plays an essential role in exploring tissue heterogeneity and characterizing cell state differences. With more and more well-annotated reference data becoming available, massive automatic identification methods have sprung up to simplify the annotation process on unlabeled target data by transferring the cell type knowledge. However, in practice, the target data often include some novel cell types that are not in the reference data. Most existing works usually classify these private cells as one generic 'unassigned' group and learn the features of known and novel cell types in a coupled way. They are susceptible to the potential batch effects and fail to explore the fine-grained semantic knowledge of novel cell types, thus hurting the model's discrimination ability. Additionally, emerging spatial transcriptomic technologies, such as in situ hybridization, sequencing and multiplexed imaging, present a novel challenge to current cell type identification strategies that predominantly neglect spatial organization. Consequently, it is imperative to develop a versatile method that can proficiently annotate single-cell transcriptomics data, encompassing both spatial and non-spatial dimensions. To address these issues, we propose a new, challenging yet realistic task called universal cell type identification for single-cell and spatial transcriptomics data. In this task, we aim to give semantic labels to target cells from known cell types and cluster labels to those from novel ones. To tackle this problem, instead of designing a suboptimal two-stage approach, we propose an end-to-end algorithm called scBOL from the perspective of Bipartite prototype alignment. Firstly, we identify the mutual nearest clusters in reference and target data as their potential common cell types. On this basis, we mine the cycle-consistent semantic anchor cells to build the intrinsic structure association between two data. Secondly, we design a neighbor-aware prototypical learning paradigm to strengthen the inter-cluster separability and intra-cluster compactness within each data, thereby inspiring the discriminative feature representations. Thirdly, driven by the semantic-aware prototypical learning framework, we can align the known cell types and separate the private cell types from them among reference and target data. Such an algorithm can be seamlessly applied to various data types modeled by different foundation models that can generate the embedding features for cells. Specifically, for non-spatial single-cell transcriptomics data, we use the autoencoder neural network to learn latent low-dimensional cell representations, and for spatial single-cell transcriptomics data, we apply the graph convolution network to capture molecular and spatial similarities of cells jointly. Extensive results on our carefully designed evaluation benchmarks demonstrate the superiority of scBOL over various state-of-the-art cell type identification methods. To our knowledge, we are the pioneers in presenting this pragmatic annotation task, as well as in devising a comprehensive algorithmic framework aimed at resolving this challenge across varied types of single-cell data. Finally, scBOL is implemented in Python using the Pytorch machine-learning library, and it is freely available at https://github.com/aimeeyaoyao/scBOL.

Contrastive Learning for Robust Cell Annotation and Representation from Single-Cell Transcriptomics

Large-Scale Cell Representation Learning via Divide-and-Conquer Contrastive Learning

Predicting cell types with supervised contrastive learning on cells and their types

CellContrast: Reconstructing spatial relationships in single-cell RNA sequencing data via deep contrastive learning

Sctab: Scaling Cross-Tissue Single-Cell Annotation Models

Scellseg: a Style-Aware Cell Instance Segmentation Tool with Pre-Training and Contrastive Fine-Tuning

scARE: Attribution Regularization for Single Cell Representation Learning

CELLama: Foundation Model for Single Cell and Spatial Transcriptomics by Cell Embedding Leveraging Language Model Abilities

Network Embedding-Based Representation Learning for Single Cell RNA-seq Data

Contrastive learning enables rapid mapping to multimodal single-cell atlas of multimillion scale

scEVOLVE: cell-type incremental annotation without forgetting for single-cell RNA-seq data

GLOBE: a Contrastive Learning-Based Framework for Integrating Single-Cell Transcriptome Datasets.

scMMT: a multi-use deep learning approach for cell annotation, protein prediction and embedding in single-cell RNA-seq data

Cell Type Identification from Single-Cell Transcriptomic Data Via Gene Embedding.

scELMo: Embeddings from Language Models are Good Learners for Single-cell Data Analysis

Distribution-Independent Cell Type Identification for Single-Cell RNA-seq Data

Towards robust and generalizable representations of extracellular data using contrastive learning

scBOL: a universal cell type identification framework for single-cell and spatial transcriptomics data

A Supervised Contrastive Framework for Learning Disentangled Representations of Cell Perturbation Data

Adversarial learning enables unbiased organism-wide cross-species alignment of single-cell RNA data at scale

Capturing cell heterogeneity in representations of cell populations for image-based profiling using contrastive learning