Scmaskgan: masked multi-scale CNN and attention-enhanced GAN for scRNA-seq dropout imputation.

Journal: BMC bioinformatics
Published Date:

Abstract

Single-cell RNA sequencing (scRNA-seq) enables high-resolution analysis of cellular heterogeneity, but dropout events, where gene expression is undetected in individual cells, present a significant challenge. We propose scMASKGAN, which transforms matrix imputation into a pixel restoration task to improve the recovery of missing gene expression data. Specifically, we integrate masking, convolutional neural networks (CNNs), attention mechanisms, and residual networks (ResNets) to effectively address dropout events in scRNA-seq data. The masking mechanism ensures the preservation of complete cellular information, while convolution and attention mechanisms are employed to capture both global and local features. Residual networks augment feature representation and effectively mitigate the risk of model overfitting. Additionally, cell-type labels are incorporated as constraints to guide the methods in learning more accurate cellular features. Finally, multiple experiments were conducted to evaluate the methods' performance using seven different data types and scRNA-seq data from ten neuroblastoma samples. The results demonstrate that the data imputed by scMASKGAN not only perform excellently across various evaluation metrics but also significantly enhance the effectiveness of downstream analyses, enabling a more comprehensive exploration of underlying biological information.

Authors

  • You Wu
    Tsinghua University School of Medicine, Beijing, China.
  • Li Xu
    College of Acupuncture and Massage, Shandong University of Traditional Chinese Medicine, Jinan, China.
  • Xiaohong Cong
    College of Computer Science and Technology, Harbin Engineering University, Harbin, China.
  • Hanxiao Li
    School of Medicine, South China University of Technology, Guangzhou, Guangdong 510006, China; Department of Radiology, Guangdong Provincial People's Hospital, Guangdong Academy of Medical Sciences, Guangzhou 510080, China.
  • Yanli Li
    Intuitive Surgical, Inc., Sunnyvale, California, USA.