Discovering Latent Scientific Concepts through Discrete Representation Learning
Stoimcev, M.; Verma, M.; Filliol, A.; Skamagki, M.; Flowers, S.; Cohen, J.; Azadian, Z.; Newlin, N.; Abdel-Mottaleb, M. S.; Dzeroski, S.; Romesser, P.; Lowe, S.; Dimitrova, N.
Show abstract
Scientific imaging is undergoing a fundamental transformation. Advances in multiplex immunofluorescence, computational pathology, and electron microscopy now enable observation of biological and physical systems at unprecedented spatial resolution. Rather than simply classifying known objects, these data offer the opportunity to discover previously unknown organizational principles. However, scientific image datasets are typically highly curated, information-rich, and limited in size, making reliable representation learning particularly challenging. We show that standard continuous self-supervised adaptation can become unstable in this regime, allowing teacher representations to drift and degrading downstream transfer. We introduce Kodiak, a discrete representation learning architecture that discovers latent scientific concepts through Sinkhorn-balanced codebook assignments. Instead of learning unconstrained continuous embeddings, Kodiak constructs a reusable vocabulary of discrete concepts that provides stable supervision for both masked-patch prediction and cross-view representation learning within a single-stage framework. Across five scientific imaging domains, including multiplex immunofluorescence for pancreatic cancer pathology and scanning electron microscopy, Kodiak consistently improves representation quality under both continued foundation-model adaptation and training from scratch. Beyond improved classification and transfer,the learned concepts are interpretable, reusable, and produce spatially coherent concept maps that support applications such as cell phenotyping, segmentation-free patch phenotyping, spatial niche discovery, tissue organization, and materials characterization. We view Kodiak as the representation-learning engine of our foundation model for spatial biology. More broadly, our results suggest that discrete representation learning provides a principled foundation for discovering reusable latent scientific concepts that can accelerate scientific interpretation across imaging domains.
Matching journals
The top 6 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Unraveling the complexity of rat object vision requires a full convolutional network - and beyond 93%
- Hierarchical confounder discovery in the experiment-machine learning cycle 92%
- CellContrast: Reconstructing Spatial Relationships in Single-Cell RNA Sequencing Data via Deep Contrastive Learning 92%
Similar papers in this journal
- DGAT: A Dual-Graph Attention Network for Inferring Spatial Protein Landscapes from Transcriptomics 96%
- CAPTAIN: A multimodal foundation model pretrained on co-assayed single-cell RNA and protein 95%
- Multi-modal Diffusion Model with Dual-Cross-Attention for Multi-Omics Data Generation and Translation 95%
"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.