@inproceedings{
kim2023singlecell,
title={Single-cell Masked Autoencoder: An Accurate and Interpretable Automated Immunophenotyper},
author={Jaesik Kim and Matei Ionita and Matthew Lee and Michelle McKeague and Ajinkya Pattekar and Mark Painter and Joost Wagenaar and Van Truong and Dylan Norton and Divij Mathew and Yonghyun Nam and Sokratis Apostolidis and Patryk Orzechowski and Sang-Hyuk Jung and Jakob Woerner and Yidi Huang and Nuala J. Meyer and Allison R. Greenplate and Dokyoon Kim and John Wherry},
booktitle={NeurIPS 2023 AI for Science Workshop},
year={2023},
url={https://openreview.net/forum?id=2mq6uezuGj}
}
High-throughput single-cell cytometry data are crucial for understanding the immune system’s role in diseases and treatment response. However, the prevailing methods used for analyzing cytometry data, specifically manual gating and clustering methods, have certain limitations with scalability, robustness, and accuracy. In this study, we propose a single-cell masked autoencoder (scMAE), which offers an automated solution for immunophenotyping tasks such as cell type prediction. Our model aims to preserve the cell type definitions designed by the user, making interpretation and cross-study comparisons more accessible. The scMAE model follows a pre-train and fine-tune paradigm. During pre-training, scMAE utilizes Masked Single-cell Modelling (MScM) to learn relationships between protein markers in immune cells without the need for prior labeling information. Subsequently, the scMAE is fine-tuned on multiple specialized tasks, using a smaller designated portion of labeled data. Through evaluation experiments, we demonstrated that the pre-trained scMAE overcomes limitations of manual gating and clustering methods, providing accurate and interpretable cellular immunophenotyping. The introduction of scMAE represents a significant advancement in immunology research, enabling prediction and interpretation of cellular-level in immune disease.
Single-cell Masked Autoencoder: An Accurate and Interpretable Automated Immunophenotyper
Jaesik Kim Matei Ionita Matthew Eric Lee Michelle McKeague Ajinkya Pattekar Mark Painter Joost Wagenaar Van Quynh-Thi Truong Dylan Norton Divij Mathew Yonghyun Nam Sokratis Apostolidis Patryk Orzechowski Sang-Hyuk Jung Jakob Woerner Yidi Huang Nuala J. Meyer Allison R. Greenplate Dokyoon Kim John Wherry