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

Yuyao Zhai, Liang Chen, Minghua Deng

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence

Main Track. Pages 6143-6151.

https://doi.org/10.24963/ijcai.2024/679

PDF BibTeX

Abstract

Automatic cell type annotation aims to transfer the label knowledge from label-abundant reference data to label-scarce target data, which makes encouraging progress in single-cell RNA-seq data analysis. While previous works have focused on classifying close-set cells and detecting open-set cells during testing, it is still essential to be able to classify unknown cell types as human beings. Additionally, few efforts have been devoted to addressing the challenge of common long-tail dilemma in cell type annotation data. Therefore, in this paper, we propose an innovative distribution-independent universal cell type identification framework called scDET from the perspective of autonomously equilibrated dual-consultative contrastive learning. Our model can generate fine-grained predictions for both close-set and open-set cell types in a long-tailed open-world environment. scDET consists of a contrastive-learning branch and a pseudo-labeling branch, which work collaboratively to provide interactive supervision. Specifically, the contrastive-learning branch provides reliable distribution estimation to regularize the predictions of the pseudo-labeling branch, which in turn guides itself through self-balanced knowledge transfer and a designed novel soft contrastive loss. Extensive experimental results on various evaluation datasets demonstrate the superior performance of scDET over other state-of-the-art single-cell clustering and annotation methods.

Keywords

  • Multidisciplinary Topics and Applications: MTA: Bioinformatics
  • Multidisciplinary Topics and Applications: MTA: Other
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards