ADARES: A Single-cell Classification Model Based on Adversarial Data Augmentation and Residual Networks.

Zhi-Peng Zhu,Pengyong Han,Lei Wang,Xi-Shang Dong,San-Rong Liu,Zheng-Wei Li
DOI: https://doi.org/10.1145/3614008.3614029
2023-01-01
Abstract:Single-cell RNA sequencing (scRNA-seq) helps researchers identify the expression status of genes from the bulk level to the single-cell level, where the recognition of cell types plays an important role in this area. However, due to the heterogeneity and complexity of single-cell data, there are some limitations to the performance of traditional classification methods, such as the neglect of higher-order expression between cells and the subpar classification results on large-scale datasets, etc. In this paper, we propose a novel deep learning model combining residual network and adversarial data augmentation for cell types classification (ADARES). Specifically, we first construct a gene expression matrix and a gene adjacency matrix through gene filtering. Next, these two matrices are fed into the residual network to extract the embedding of nodes and a softmax feature aggregation layer is adopted to fuse the embedding. In addition, we introduce a perturbation with gradient in the process of training to further enhance node features and improve the robustness of our model. Extensive experiments demonstrate that ADARES has achieved satisfactory results on several benchmark single-cell datasets, our proposed model can be used as an effective method for recognizing cell types.
What problem does this paper attempt to address?