An active learning approach for clustering single-cell RNA-seq data.

Journal: Laboratory investigation; a journal of technical methods and pathology
Published Date:

Abstract

Single-cell RNA sequencing (scRNA-seq) data has been widely used to profile cellular heterogeneities with a high-resolution picture. Clustering analysis is a crucial step of scRNA-seq data analysis because it provides a chance to identify and uncover undiscovered cell types. Most methods for clustering scRNA-seq data use an unsupervised learning strategy. Since the clustering step is separated from the cell annotation and labeling step, it is not uncommon for a totally exotic clustering with poor biological interpretability to be generated-a result generally undesired by biologists. To solve this problem, we proposed an active learning (AL) framework for clustering scRNA-seq data. The AL model employed a learning algorithm that can actively query biologists for labels, and this manual labeling is expected to be applied to only a subset of cells. To develop an optimal active learning approach, we explored several key parameters of the AL model in the experiments with four real scRNA-seq datasets. We demonstrate that the proposed AL model outperformed state-of-the-art unsupervised clustering methods with less than 1000 labeled cells. Therefore, we conclude that AL model is a promising tool for clustering scRNA-seq data that allows us to achieve a superior performance effectively and efficiently.

Authors

  • Xiang Lin
    Department of Computer Science, New Jersey Institute of Technology, Newark, NJ, USA.
  • Haoran Liu
    State Key Laboratory for Manufacturing Systems Engineering, Xi'an Jiaotong University, No.28, Xianning West Road, Xi'an, Shaanxi 710049, P.R. China.
  • Zhi Wei
    Department of Computer Science, New Jersey Institute of Technology, Newark, New Jersey, USA, zhiwei@njit.edu.
  • Senjuti Basu Roy
    Department of Computer Science, New Jersey Institute of Technology, Newark, NJ, USA.
  • Nan Gao
    Department of Biological Sciences, Rutgers University, Newark, NJ, USA.