SwarmMAP: Swarm Learning for Decentralized Cell Type Annotation in Single Cell Sequencing Data

bioRxiv (Cold Spring Harbor Laboratory)

Abstract

Abstract Rapid technological advancements have made it possible to generate single-cell data at a large scale. Several laboratories around the world can now generate single-cell transcriptomic data from different tissues. Unsupervised clustering, followed by annotation of the cell type of the identified clusters, is a crucial step in single-cell analyses. However, there is no consensus on the marker genes to use for annotation, and celltype annotation is currently mostly done by manual inspection of marker genes, which is irreproducible, and poorly scalable. Additionally, patient-privacy is also a critical issue with human datasets. There is a critical need to standardize and automate celltype annotation across datasets in a privacy-preserving manner. Here, we developed SwarmMAP that uses Swarm Learning to train machine learning models for cell-type classification based on single-cell sequencing data in a decentralized way. SwarmMAP does not require any exchange of raw data between data centers. SwarmMAP has a F1-score of 0.93, 0.98, and 0.88 for cell type classification in human heart, lung, and breast datasets, respectively. Swarm Learning-based models yield an average performance of 0.907 which is on par with the performance achieved by models trained on centralized data ( p -val= 0.937 , Mann-Whitney U Test). We also find that increasing the number of datasets increases cell-type prediction accuracy and enables handling higher cell-type diversity. Together, these findings demonstrate that Swarm Learning is a viable approach to automate cell-type annotation. SwarmMAP is available at https://github.com/hayatlab/SwarmMAP .

Authors 7

1. Oliver Lester Saldanha

   Else Kröner Fresenius Center for Digital Health · Technische Universität Dresden

   Affiliation as printed

   Else Kroener Fresenius Center for Digital Health, Technical University Dresden, Fetscherstraße 74, Dresden, 01307, Saxony, Germany

   TU Dresden

2. Vivien Goepp Aachen Department of Medicine 2

   RWTH Aachen University

   Affiliation as printed

   Department of Medicine 2, RWTH Aachen University, Medical Faculty, Pauwelsstrasse 30, Aachen, 52074, North Rhine-Westphalia, Germany

   RWTH Aachen University

3. Hyojin Kim Aachen

   University Hospital Carl Gustav Carus · Technische Universität Dresden · RWTH Aachen University

   Affiliation as printed

   Department of Medicine I, Faculty of Medicine and University Hospital Carl Gustav Carus, Technical University Dresden Fetscherstraße 74, Dresden, 01307, Saxony, Germany

   RWTH Aachen university

4. Jie Ping Zhu

   Else Kröner Fresenius Center for Digital Health · Technische Universität Dresden

   Affiliation as printed

   Else Kroener Fresenius Center for Digital Health, Technical University Dresden, Fetscherstraße 74, Dresden, 01307, Saxony, Germany

5. Rafael Kramann Aachen Department of Medicine 2

   RWTH Aachen University

   Affiliation as printed

   Department of Medicine 2, RWTH Aachen University, Medical Faculty, Pauwelsstrasse 30, Aachen, 52074, North Rhine-Westphalia, Germany

   RWTH Aachen University

6. Sikander Hayat Aachen Department of Medicine 2 Medical Faculty

   RWTH Aachen University

   Affiliation as printed

   Department of Medicine 2, RWTH Aachen University, Medical Faculty, Pauwelsstrasse 30, Aachen, 52074, North Rhine-Westphalia, Germany

   RWTH Aachen University, Medical Faculty

7. Jakob Nikolas Kather corresponding

   University Hospital Heidelberg · National Center for Tumor Diseases · University Hospital Carl Gustav Carus · Else Kröner Fresenius Center for Digital Health · Technische Universität Dresden

   Affiliation as printed

   Department of Medicine I, Faculty of Medicine and University Hospital Carl Gustav Carus, Technical University Dresden Fetscherstraße 74, Dresden, 01307, Saxony, Germany

   Else Kroener Fresenius Center for Digital Health, Technical University Dresden, Fetscherstraße 74, Dresden, 01307, Saxony, Germany

   Medical Oncology, National Center for Tumor Diseases (NCT), University Hospital Heidelberg, Im Neuenheimer Feld 460, Heidelberg, 69120, Baden-Wuerttemberg, Germany

   TU Dresden