Cellohood: multi-granular discovery of cellular neighborhoods with a permutation-invariant set transformer auto-encoder
Mozejko, M.; Schulz, D.; Gogolewski, K.; Eling, N.; Krawczyk, J.; Bogatyrova, O.; Golab, J.; Nowis, D.; Liechti, R.; Morfouace, M.; Hong, H.; Bodenmiller, B.; Staub, E.; Szczurek, E.
Show abstract
Discovering cellular neighborhoods and their roles in disease requires computational methods that consider the full breadth of data and offer multi-level granularity and interpretability. Here, we introduce Cellohood, a permutation-invariant set transformer auto-encoder equipped with a clinical association pipeline. Cellohood encodes full readouts for bags of spatially co-localized cells and supports multi-level analyses, providing interpretability by mapping latent dimensions to spatial tissue features. Our model surpasses current methods in accuracy of cellular neighborhood detection for spatial transcriptomics of the human cortex and CODEX spleen data measuring lupus progression. Applied to cancer data across three granularity levels, our method recovers neighborhoods along the expected immune-cold to immune-hot spectrum and further refines them into biologically and clinically meaningful subclasses, revealing spatial patterns linked to prognosis, histology, stage, and tumor mutational burden, and uncovering subgroups that transcend standard classifications. Overall, Cellohood enables in-depth analysis of complex tissues, revealing clinically informative spatial neighborhoods.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Probabilistic embedding, clustering, and alignment for integrating spatial transcriptomics data with PRECAST 97%
- uniPort: a unified computational framework for single-cell data integration with optimal transport 97%
- Learning interpretable cellular and gene signature embeddings from single-cell transcriptomic data 97%
Similar papers in this journal
- SHEST: Single-cell-level artificial intelligence from haematoxylin and eosin morphology for cell type prediction and spatial transcriptomics reconstruction 97%
- SpaTM: Topic Models for Inferring Spatially Informed Transcriptional Programs 96%
- An in-depth comparison of linear and non-linear joint embedding methods for bulk and single-cell multi-omics 96%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.