Abstract:Single-cell RNA sequencing (scRNA-seq) has been widely used in cancer research to understand the complex gene expression diversity and cancer heterogeneity. However, manual annotation of cell types in the scRNA-seq pipeline is time-consuming and depends on the expertise of analyzers, which can significantly influence the results of downstream analyses. To address this problem, we proposed a novel machine learning framework utilizing the LightGBM model for automated and efficient cell-type annotation of scRNA-seq. Two independent scRNA-seq datasets of non-small cell lung cancer (NSCLC) downloaded from the Gene Expression Omnibus (GEO) were used to train and test our model. A standard procedure is applied to both scRNA-seq datasets for quality control and preprocessing, in which poor-quality cells with low gene expressions or high scores for cellular stress/death were excluded. In addition, Harmony is applied to mitigate batch effects in scRNA-seq that could cause variability due to non-biological factors in experiments. Nine different cell types, including endothelial, epithelial, fibroblast, macrophages, mast, plasma, pulmonary alveolar, B, and T cells, were manually labeled in the two datasets by the providers, which were also examined using gene markers corresponding to different cell types from PanglaoDB and DAVID. These manually labeled cell types were used as the ground truth for training and testing our model. In the training stage, the training dataset (containing 85,000 cells from 44 NSCLC samples) of scRNA-seq was used to train the LightGBM model with its high-variable genes. Then, the model would be evaluated using an independent test dataset (containing 8,000 cells from 18 NSCLC samples) by comparing the automatically predicted and manually labeled cell types. The training result showed that our model could successfully specify the nine different cell types, achieving an overall average accuracy, F1 score, and precision of 0.86 each respectively. In the independent dataset test, the model demonstrated good generalization, showing high predictive performance across all cell types, with an average accuracy, F1 score, and precision of 0.8, 0.78, and 0.8, respectively. Specific to the predictions in the test dataset, we found that some epithelial cells were mistakenly identified as other cell types. This might be because of the complex gene expression patterns exhibited by tumor epithelial cells, making accurate predictions challenging. The proposed machine learning framework facilitates cell labeling and unravels the intricate heterogeneity within lung cancer datasets. The combination of LightGBM and standardized preprocessing establishes a benchmark for high-throughput, accurate single-cell analysis, paving the way for discoveries that are more targeted and have significant clinical impact. Citation Format: Tsung Hsien Chuang, Liang-Chuan Lai, Tzu-Pin Lu, Mong-Hsun Tsai, Hsiang-Han Chen, Eric Y. Chuang. Enhancing single-cell RNA sequencing analysis in cancer research: A machine learning framework based on LightGBM for automated cell type annotation [abstract]. In: Proceedings of the American Association for Cancer Research Annual Meeting 2024; Part 1 (Regular s); 2024 Apr 5-10; San Diego, CA. Philadelphia (PA): AACR; Cancer Res 2024;84(6_Suppl) nr 878.

Knowledge-based classification of fine-grained immune cell types in single-cell RNA-Seq data

Multimodal hierarchical classification of CITE-seq data delineates immune cell states across lineages and tissues

Identification of cell types, states and programs by learning gene set representations

An elastic-net logistic regression approach to generate classifiers and gene signatures for types of immune cells and T helper cell subsets

Identifying the Immunological Gene Signatures of Immune Cell Subtypes

SCINA: Semi-Supervised Analysis of Single Cells in silico

scImmOmics: a manually curated resource of single-cell multi-omics immune data

Evaluation of single-cell classifiers for single-cell RNA sequencing data sets

A machine learning framework for scRNA-seq UMI threshold optimization and accurate classification of cell types

A comparison of scRNA-seq annotation methods based on experimentally labeled immune cell subtype dataset

Abstract 878: Enhancing single-cell RNA sequencing analysis in cancer research: A machine learning framework based on LightGBM for automated cell type annotation

Dissecting the human immune system with single cell RNA sequencing technology

Characterization of Spleen and Lymph Node Cell Types Via CITE-seq and Machine Learning Methods

scClassify: sample size estimation and multiscale classification of cells using single and multiple reference

Identification of Colon Immune Cell Marker Genes Using Machine Learning Methods.

Automated identification of Cell Types in Single Cell RNA Sequencing

Classification of PBMC cell types using scRNAseq, ANN, and incremental learning

Integrating single-cell multi-omics and prior biological knowledge for a functional characterization of the immune system

CTISL: a dynamic stacking multi-class classification approach for identifying cell types from single-cell RNA-seq data

Hierarchical cell-type identifier accurately distinguishes immune-cell subtypes enabling precise profiling of tissue microenvironment with single-cell RNA-sequencing

Digitaldlsorter: Deep-Learning on scRNA-Seq to Deconvolute Gene Expression Data