MMIL: A novel algorithm for disease associated cell type discovery

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Craig, Erin, Keyes, Timothy, Sarno, Jolanda, Zaslavsky, Maxim, Nolan, Garry, Davis, Kara, Hastie, Trevor, Tibshirani, Robert
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916283813986304
author Craig, Erin
Keyes, Timothy
Sarno, Jolanda
Zaslavsky, Maxim
Nolan, Garry
Davis, Kara
Hastie, Trevor
Tibshirani, Robert
author_facet Craig, Erin
Keyes, Timothy
Sarno, Jolanda
Zaslavsky, Maxim
Nolan, Garry
Davis, Kara
Hastie, Trevor
Tibshirani, Robert
contents Single-cell datasets often lack individual cell labels, making it challenging to identify cells associated with disease. To address this, we introduce Mixture Modeling for Multiple Instance Learning (MMIL), an expectation maximization method that enables the training and calibration of cell-level classifiers using patient-level labels. Our approach can be used to train e.g. lasso logistic regression models, gradient boosted trees, and neural networks. When applied to clinically-annotated, primary patient samples in Acute Myeloid Leukemia (AML) and Acute Lymphoblastic Leukemia (ALL), our method accurately identifies cancer cells, generalizes across tissues and treatment timepoints, and selects biologically relevant features. In addition, MMIL is capable of incorporating cell labels into model training when they are known, providing a powerful framework for leveraging both labeled and unlabeled data simultaneously. Mixture Modeling for MIL offers a novel approach for cell classification, with significant potential to advance disease understanding and management, especially in scenarios with unknown gold-standard labels and high dimensionality.
format Preprint
id arxiv_https___arxiv_org_abs_2406_08322
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MMIL: A novel algorithm for disease associated cell type discovery
Craig, Erin
Keyes, Timothy
Sarno, Jolanda
Zaslavsky, Maxim
Nolan, Garry
Davis, Kara
Hastie, Trevor
Tibshirani, Robert
Quantitative Methods
Machine Learning
Methodology
Single-cell datasets often lack individual cell labels, making it challenging to identify cells associated with disease. To address this, we introduce Mixture Modeling for Multiple Instance Learning (MMIL), an expectation maximization method that enables the training and calibration of cell-level classifiers using patient-level labels. Our approach can be used to train e.g. lasso logistic regression models, gradient boosted trees, and neural networks. When applied to clinically-annotated, primary patient samples in Acute Myeloid Leukemia (AML) and Acute Lymphoblastic Leukemia (ALL), our method accurately identifies cancer cells, generalizes across tissues and treatment timepoints, and selects biologically relevant features. In addition, MMIL is capable of incorporating cell labels into model training when they are known, providing a powerful framework for leveraging both labeled and unlabeled data simultaneously. Mixture Modeling for MIL offers a novel approach for cell classification, with significant potential to advance disease understanding and management, especially in scenarios with unknown gold-standard labels and high dimensionality.
title MMIL: A novel algorithm for disease associated cell type discovery
topic Quantitative Methods
Machine Learning
Methodology
url https://arxiv.org/abs/2406.08322