Attention-Based Graph Neural Network for Label Propagation in Single-Cell Omics.

Journal: Genes
Published Date:

Abstract

Single-cell data analysis has been at forefront of development in biology and medicine since sequencing data have been made available. An important challenge in single-cell data analysis is the identification of cell types. Several methods have been proposed for cell-type identification. However, these methods do not capture the higher-order topological relationship between different samples. In this work, we propose an attention-based graph neural network that captures the higher-order topological relationship between different samples and performs transductive learning for predicting cell types. The evaluation of our method on both simulation and publicly available datasets demonstrates the superiority of our method, scAGN, in terms of prediction accuracy. In addition, our method works best for highly sparse datasets in terms of F1 score, precision score, recall score, and Matthew's correlation coefficients as well. Further, our method's runtime complexity is consistently faster compared to other methods.

Authors

  • Rahul Bhadani
    Department of Electrical & Computer Engineering, The University of Arizona, Tucson, AZ 85721, USA.
  • Zhuo Chen
    State Key Laboratory Breeding Base of Green Pesticide and Agricultural Bioengineering, Key Laboratory of Green Pesticide and Agricultural Bioengineering, Ministry of Education, Guizhou University, Huaxi District, Guiyang 550025, China. Electronic address: gychenzhuo@aliyun.com.
  • Lingling An
    School of Computer Science and Technology, Xidian University, Xi'an, China.