A Foundation Model for Cell Segmentation

Uriah Israel,Markus Marks,Rohit Dilip,Qilin Li,Changhua Yu,Emily Laubscher,Shenyi Li,Morgan Schwartz,Elora Pradhan,Ada Ates,Martin Abt,Caitlin Brown,Edward Pao,Alexander Pearson-Goulart,Pietro Perona,Georgia Gkioxari,Ross Barnowski,Yisong Yue,David Van Valen
DOI: https://doi.org/10.1101/2023.11.17.567630
2024-03-07
Abstract:Cells are a fundamental unit of biological organization, and identifying them in imaging data – cell segmentation – is a critical task for various cellular imaging experiments. While deep learning methods have led to substantial progress on this problem, most models in use are specialist models that work well for specific domains. Methods that have learned the general notion of “what is a cell” and can identify them across different domains of cellular imaging data have proven elusive. In this work, we present CellSAM, a foundation model for cell segmentation that generalizes across diverse cellular imaging data. CellSAM builds on top of the Segment Anything Model (SAM) by developing a prompt engineering approach for mask generation. We train an object detector, CellFinder, to automatically detect cells and prompt SAM to generate segmentations. We show that this approach allows a single model to achieve human-level performance for segmenting images of mammalian cells (in tissues and cell culture), yeast, and bacteria collected across various imaging modalities. We show that CellSAM has strong zero-shot performance and can be improved with a few examples via few-shot learning. We also show that CellSAM can unify bioimaging analysis workflows such as spatial transcriptomics and cell tracking. A deployed version of CellSAM is available at .
Bioinformatics
What problem does this paper attempt to address?