CellViT: Vision Transformers for Precise Cell Segmentation and Classification
Not on the map. Why?
Abstract (arXiv)
Nuclei detection and segmentation in hematoxylin and eosin-stained (H&E) tissue images are important clinical tasks and crucial for a wide range of applications. However, it is a challenging task due to nuclei variances in staining and size, overlapping boundaries, and nuclei clustering. While convolutional neural networks have been extensively used for this task, we explore the potential of Transformer-based networks in this domain. Therefore, we introduce a new method for automated instance segmentation of cell nuclei in digitized tissue samples using a deep learning architecture based on Vision Transformer called CellViT. CellViT is trained and evaluated on the PanNuke dataset, which is one of the most challenging nuclei instance segmentation datasets, consisting of nearly 200,000 annotated Nuclei into 5 clinically important classes in 19 tissue types. We demonstrate the superiority of large-scale in-domain and out-of-domain pre-trained Vision Transformers by leveraging the recently published Segment Anything Model and a ViT-encoder pre-trained on 104 million histological image patches - achieving state-of-the-art nuclei detection and instance segmentation performance on the PanNuke dataset with a mean panoptic quality of 0.50 and an F1-detection score of 0.83. The code is publicly available at https://github.com/TIO-IKIM/CellViT
Key facts. Cited by 108 papers in the map (data through 14 Oct 2026). Builds on 38 of its 76 references that are in the map. First posted on arXiv 27 Jun 2023.
Builds on
38 of 76 references.
References · 38
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2023Segment Anythingcited by 8,234
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2017Mask R-CNNcited by 7,277
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
- 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,940
26 more
- 2015Fast R-CNNcited by 2,933
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
- 2020Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformerscited by 1,369
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
- 2018Panoptic Segmentationcited by 802
- 2020Data Efficient and Weakly Supervised Computational Pathology on Whole Slide Imagescited by 745
- 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 675
- 2020Image Segmentation Using Deep Learning: A Surveycited by 660
- 2018Albumentations: fast and flexible image augmentationscited by 560
- 2018HoVer-Net: Simultaneous Segmentation and Classification of Nuclei in Multi-Tissue Histology Imagescited by 388
- 2022Scaling Vision Transformers to Gigapixel Images via Hierarchical Self-Supervised Learningcited by 359
- 2018Cell Detection with Star-convex Polygonscited by 183
- 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 172
- 2020U-Net and its variants for medical image segmentation: theory and applicationscited by 169
- 2020PanNuke Dataset Extension, Insights and Baselinescited by 114
- 2016DCAN: Deep Contour-Aware Networks for Accurate Gland Segmentationcited by 84
- 2021Medical Image Segmentation Using Squeeze-and-Expansion Transformerscited by 57
- 2018Micro-Net: A unified model for segmentation of various objects in microscopy imagescited by 56
- 2021CoNIC: Colon Nuclei Identification and Counting Challenge 2022cited by 45
- 2022One Model is All You Need: Multi-Task Learning Enables Simultaneous Histology Image Segmentation and Classificationcited by 38
- 2021CPP-Net: Context-aware Polygon Proposal Network for Nucleus Segmentationcited by 32
- 2022Nuclei instance segmentation and classification in histopathology images with StarDistcited by 31
- 2019Nuclear Instance Segmentation using a Proposal-Free Spatially Aware Deep Learning Frameworkcited by 10
- 2022Valuing Vicinity: Memory attention framework for context-based semantic segmentation in histopathologycited by 3
Built on by
Citing papers per year: 2023: 4; 2024: 22; 2025: 46; 2026: 36.
108 papers.
Sort:
By area:
- 2024BiomedParse: a biomedical foundation model for image parsing of everything everywhere all at oncecited by 111
- 2024HEST-1k: A Dataset for Spatial Transcriptomics and Histology Image Analysiscited by 107
- 2023A General-Purpose Self-Supervised Model for Computational Pathologycited by 42
- 2024How to build the best medical image segmentation algorithm using foundation models: a comprehensive empirical study with Segment Anything Modelcited by 32
- 2024Foundation Models for Biomedical Image Segmentation: A Surveycited by 28
- 2024Segment Any Cell: A SAM-based Auto-prompting Fine-tuning Framework for Nuclei Segmentationcited by 24
- 2023A Recent Survey of Vision Transformers for Medical Image Segmentationcited by 23
- 2024STimage-1K4M: A histopathology image-gene expression dataset for spatial transcriptomicscited by 23
- 2023Unleashing the Power of Prompt-driven Nucleus Instance Segmentationcited by 19
- 2024ViTGaze: Gaze Following with Interaction Features in Vision Transformerscited by 15
- 2025PixCell: A generative foundation model for digital histopathology imagescited by 11
- 2025Segment Anything for Histopathologycited by 9
38 more
- 2025PathDiff: Histopathology Image Synthesis with Unpaired Text and Mask Conditionscited by 9
- 2024CoTCoNet: An Optimized Coupled Transformer-Convolutional Network with an Adaptive Graph Reconstruction for Leukemia Detectioncited by 7
- 2024CellSeg1: Robust Cell Segmentation with One Training Imagecited by 6
- 2025MedicoSAM: Robust Improvement of SAM for Medical Imagingcited by 6
- 2024NuSegDG: Integration of Heterogeneous Space and Gaussian Kernel for Domain-Generalized Nuclei Segmentationcited by 5
- 2024Evaluating Cell AI Foundation Models in Kidney Pathology with Human-in-the-Loop Enrichmentcited by 5
- 2025Medical Image Classification with KAN-Integrated Transformers and Dilated Neighborhood Attentioncited by 5
- 2025Towards Effective and Efficient Context-aware Nucleus Detection in Histopathology Whole Slide Imagescited by 5
- 2025MambaVesselNet++: A Hybrid CNN-Mamba Architecture for Medical Image Segmentationcited by 5
- 2024OMG-Net: A Deep Learning Framework Deploying Segment Anything to Detect Pan-Cancer Mitotic Figures from Haematoxylin and Eosin-Stained Slidescited by 4
- 2024RankByGene: Gene-Guided Histopathology Representation Learning Through Cross-Modal Ranking Consistencycited by 4
- 2025Revisiting Automatic Data Curation for Vision Foundation Models in Digital Pathologycited by 4
- 2025A Multi-Stage Auto-Context Deep Learning Framework for Tissue and Nuclei Segmentation and Classification in H&E-Stained Histological Images of Advanced Melanomacited by 4
- 2025Ensemble learning of pathology foundation models for precision oncologycited by 4
- 2025ACS-SegNet: An Attention-Based CNN-SegFormer Segmentation Network for Tissue Segmentation in Histopathologycited by 4
- 2023Morphological Profiling for Drug Discovery in the Era of Deep Learningcited by 3
- 2024LKCell: Efficient Cell Nuclei Instance Segmentation with Large Convolution Kernelscited by 3
- 2024Segmentation of Muscularis Propria in Colon Histopathology Images Using Vision Transformers for Hirschsprung's Diseasecited by 3
- 2025COIN: Confidence Score-Guided Distillation for Annotation-Free Cell Segmentationcited by 3
- 2025The Four Color Theorem for Cell Instance Segmentationcited by 3
- 2025Co-Seg++: Mutual Prompt-Guided Collaborative Learning for Versatile Medical Segmentationcited by 3
- 2025Co-Seg: Mutual Prompt-Guided Collaborative Learning for Tissue and Nuclei Segmentationcited by 3
- 2025MATCH: Multi-faceted Adaptive Topo-Consistency for Semi-Supervised Histopathology Segmentationcited by 3
- 2025Cell Instance Segmentation: The Devil Is in the Boundariescited by 3
- 2024NuLite -- Lightweight and Fast Model for Nuclei Instance Segmentation and Classificationcited by 2
- 2024Automated Classification of Cell Shapes: A Comparative Evaluation of Shape Descriptorscited by 2
- 2025Histo-Miner: Deep learning based tissue features extraction pipeline from H&E whole slide images of cutaneous squamous cell carcinomacited by 2
- 2025MIPHEI-ViT: Multiplex Immunofluorescence Prediction from H&E Images using ViT Foundation Modelscited by 2
- 2025Fine-grained Multi-class Nuclei Segmentation with Molecular-empowered All-in-SAM Modelcited by 2
- 2025EM3M: An Electron Micrograph Dataset for Microstructural Segmentation and Generationcited by 2
- 2025Learning the Language of Histopathology Images reveals Prognostic Subgroups in Invasive Lung Adenocarcinoma Patientscited by 2
- 2025MorphGen: Morphology-Guided Representation Learning for Robust Single-Domain Generalization in Histopathological Cancer Classificationcited by 2
- 2025PEaRL: Pathway-Enhanced Representation Learning for Gene and Pathway Expression Prediction from Histologycited by 2
- 2025MUSE: Multi-Scale Dense Self-Distillation for Nucleus Detection and Classificationcited by 2
- 2025Supervise Less, See More: Training-free Nuclear Instance Segmentation with Prototype-Guided Promptingcited by 2
- 2026Disco: Densely-overlapping Cell Instance Segmentation via Adjacency-aware Collaborative Coloringcited by 2
- 2026NuNext: Reframing Nucleus Detection as Next-Point Detectioncited by 2
- 2026Beyond Attention Heatmaps: How to Get Better Explanations for Multiple Instance Learning Models in Histopathologycited by 2
50 most cited of 108 All 108 (JSON)
- 2026From UNI2-h to ConvNeXt-T: Lightweight Nuclei Instance Segmentation via Knowledge Distillation28 Sep 2026
- 2026TEAMS: Text-prompted spatiotEmporal dual-heAd Mamba Snake18 Aug 2026
- 2026Lymphocyte Mimicry Correction via Region-Level Tissue Reasoning and Unbalanced Optimal Transport17 Aug 2026
- 2026From Multi-Resolution Cells to Gigapixel Whole Slide Images Foundation Model for Computational Pathology4 Aug 2026
- 2026Beyond Classification: Pathology Foundation Models as Detection Encoders for Mitotic Figures30 Jul 2026
- 2026Towards Autonomous and Auditable Medical Imaging Model Development12 Jul 2026
- 2026MalariAI: A Label-Resilient Decoupled Framework for Annotation-Agnostic Cell Segmentation and Explainable Stage Classification in Dense Malaria Blood Smears1 Jul 2026
- 2026CellDETR: A Detection-Guided Framework for Scalable Cell Representation Learning from Histopathology Images28 Jun 2026
- 2026Rethinking the Adaptation of Vision Foundation Models for Efficient Cell Segmentation20 Jun 2026
- 2026μMatch: Foundation Models for Semi-supervised Learning and Domain Adaptation in EM19 Jun 2026
- 2026SegTME-UNI2: A Foundation Model-Based Framework for Generalisable Multiclass Cell Segmentation and LLM-Driven Tumour Microenvironment Characterisation in Histopathology16 Jun 2026
- 2026Atlas H&E-TME: Scalable AI-Based Tissue Profiling at Expert Pathologist-Level Accuracy10 Jun 2026
38 more
- 2026Maximum Matching Accuracy: An Instance Segmentation Evaluation Metric Utilizing Globally Optimal Matching8 Jun 2026
- 2026DualGate-Net: A Prior-Gated Dual-Encoder Framework for Histopathology Cell Detection5 Jun 2026
- 2026CodeCytos: AI-assisted spatial molecular imaging analysis via code-augmented agent action space30 May 2026
- 2026One Click per Cell Type Suffices: Training-free Group Interaction for Cell Instance Segmentation28 May 2026
- 2026DUET: Dual-Paradigm Adaptive Expert Triage with Single-cell Inductive Prior for Spatial Transcriptomics Prediction13 May 2026
- 2026Leveraging Spatial Transcriptomics as Alternative to Manual Annotations for Deep Learning-Based Nuclei Analysis26 Apr 2026
- 2026Cell Instance Segmentation via Multi-Task Image-to-Image Schr"odinger Bridge14 Apr 2026
- 2026OpenTME: An Open Dataset of AI-powered H&E Tumor Microenvironment Profiles from TCGA13 Apr 2026
- 2026Adapting SAM to Nuclei Instance Segmentation and Classification via Cooperative Fine-Grained Refinement30 Mar 2026
- 2026Revisiting foundation models for cell instance segmentation18 Mar 2026
- 2026HistoAtlas: A Pan-Cancer Morphology Atlas Linking Histomics to Molecular Programs and Clinical Outcomes17 Mar 2026
- 2026Beyond Attention Heatmaps: How to Get Better Explanations for Multiple Instance Learning Models in Histopathology9 Mar 2026
- 2026NuNext: Reframing Nucleus Detection as Next-Point Detection7 Mar 2026
- 2026DeNuC: Decoupling Nuclei Detection and Classification in Histopathology4 Mar 2026
- 2026Benchmarking Computational Pathology Foundation Models For Semantic Segmentation21 Feb 2026
- 2026Context-aware Skin Cancer Epithelial Cell Classification with Scalable Graph Transformers17 Feb 2026
- 2026Generalization of Self-Supervised Vision Transformers for Protein Localization Across Microscopy Domains5 Feb 2026
- 2026Disco: Densely-overlapping Cell Instance Segmentation via Adjacency-aware Collaborative Coloring5 Feb 2026
- 2026Explainable Pathomics Feature Visualization via Correlation-aware Conditional Feature Editing5 Feb 2026
- 2026FG-TreeSeg: Flow-Guided Tree Crown Segmentation without Instance Annotations31 Jan 2026
- 2026NucFuseRank: Dataset Fusion and Performance Ranking for Nuclei Instance Segmentation27 Jan 2026
- 2026PanopMamba: Vision State Space Modeling for Nuclei Panoptic Segmentation23 Jan 2026
- 2026Inferring Clinically Relevant Molecular Subtypes of Pancreatic Cancer from Routine Histopathology Using Deep Learning6 Jan 2026
- 2026LSP-DETR: Efficient and Scalable Nuclei Segmentation in Whole-Slide Images6 Jan 2026
- 2025CellMamba: Adaptive Mamba for Accurate and Efficient Cell Detection25 Dec 2025
- 2025IMILIA: interpretable multiple instance learning for inflammation prediction in IBD from H&E whole slide images15 Dec 2025
- 2025Supervise Less, See More: Training-free Nuclear Instance Segmentation with Prototype-Guided Prompting25 Nov 2025
- 2025Adaptive Multi-Scale Integration Unlocks Robust Cell Annotation in Histopathology Images17 Nov 2025
- 2025Rank-Aware Agglomeration of Foundation Models for Immunohistochemistry Image Cell Counting16 Nov 2025
- 2025MUSE: Multi-Scale Dense Self-Distillation for Nucleus Detection and Classification7 Nov 2025
- 2025ITC-RWKV: Interactive Tissue-Cell Modeling with Recurrent Key-Value Aggregation for Histopathological Subtyping24 Oct 2025
- 2025ACS-SegNet: An Attention-Based CNN-SegFormer Segmentation Network for Tissue Segmentation in Histopathology23 Oct 2025
- 2025Morphology-Aware Prognostic model for Five-Year Survival Prediction in Colorectal Cancer from H&E Whole Slide Images16 Oct 2025
- 2025Cell Instance Segmentation: The Devil Is in the Boundaries10 Oct 2025
- 2025PEaRL: Pathway-Enhanced Representation Learning for Gene and Pathway Expression Prediction from Histology3 Oct 2025
- 2025MATCH: Multi-faceted Adaptive Topo-Consistency for Semi-Supervised Histopathology Segmentation2 Oct 2025
- 2025Evaluating New AI Cell Foundation Models on Challenging Kidney Pathology Cases Unaddressed by Previous Foundation Models1 Oct 2025
- 2025OCELOT 2023: Cell Detection from Cell-Tissue Interaction Challenge11 Sep 2025
50 newest of 108 All 108 (JSON)
- 2025Medical Image Classification with KAN-Integrated Transformers and Dilated Neighborhood Attentioncited by 5
- 2025MambaVesselNet++: A Hybrid CNN-Mamba Architecture for Medical Image Segmentationcited by 5
- 2025CellMamba: Adaptive Mamba for Accurate and Efficient Cell Detectioncited by 1
- 2026PanopMamba: Vision State Space Modeling for Nuclei Panoptic Segmentationcited by 0
- 2026TEAMS: Text-prompted spatiotEmporal dual-heAd Mamba Snakecited by 0
- 2026TEAMS: Text-prompted spatiotEmporal dual-heAd Mamba Snake18 Aug 2026
- 2026PanopMamba: Vision State Space Modeling for Nuclei Panoptic Segmentation23 Jan 2026
- 2025CellMamba: Adaptive Mamba for Accurate and Efficient Cell Detection25 Dec 2025
- 2025MambaVesselNet++: A Hybrid CNN-Mamba Architecture for Medical Image Segmentation26 Jul 2025
- 2025Medical Image Classification with KAN-Integrated Transformers and Dilated Neighborhood Attention19 Feb 2025
- 2024BiomedParse: a biomedical foundation model for image parsing of everything everywhere all at oncecited by 111
- 2024HEST-1k: A Dataset for Spatial Transcriptomics and Histology Image Analysiscited by 107
- 2023A General-Purpose Self-Supervised Model for Computational Pathologycited by 42
- 2024How to build the best medical image segmentation algorithm using foundation models: a comprehensive empirical study with Segment Anything Modelcited by 32
- 2024Foundation Models for Biomedical Image Segmentation: A Surveycited by 28
- 2024Segment Any Cell: A SAM-based Auto-prompting Fine-tuning Framework for Nuclei Segmentationcited by 24
- 2023A Recent Survey of Vision Transformers for Medical Image Segmentationcited by 23
- 2024STimage-1K4M: A histopathology image-gene expression dataset for spatial transcriptomicscited by 23
- 2023Unleashing the Power of Prompt-driven Nucleus Instance Segmentationcited by 19
- 2024ViTGaze: Gaze Following with Interaction Features in Vision Transformerscited by 15
- 2025PixCell: A generative foundation model for digital histopathology imagescited by 11
- 2025Segment Anything for Histopathologycited by 9
12 most cited of 98 All 98 (JSON)
- 2026From Multi-Resolution Cells to Gigapixel Whole Slide Images Foundation Model for Computational Pathology4 Aug 2026
- 2026Beyond Classification: Pathology Foundation Models as Detection Encoders for Mitotic Figures30 Jul 2026
- 2026Towards Autonomous and Auditable Medical Imaging Model Development12 Jul 2026
- 2026MalariAI: A Label-Resilient Decoupled Framework for Annotation-Agnostic Cell Segmentation and Explainable Stage Classification in Dense Malaria Blood Smears1 Jul 2026
- 2026CellDETR: A Detection-Guided Framework for Scalable Cell Representation Learning from Histopathology Images28 Jun 2026
- 2026Rethinking the Adaptation of Vision Foundation Models for Efficient Cell Segmentation20 Jun 2026
- 2026μMatch: Foundation Models for Semi-supervised Learning and Domain Adaptation in EM19 Jun 2026
- 2026SegTME-UNI2: A Foundation Model-Based Framework for Generalisable Multiclass Cell Segmentation and LLM-Driven Tumour Microenvironment Characterisation in Histopathology16 Jun 2026
- 2026Atlas H&E-TME: Scalable AI-Based Tissue Profiling at Expert Pathologist-Level Accuracy10 Jun 2026
- 2026Maximum Matching Accuracy: An Instance Segmentation Evaluation Metric Utilizing Globally Optimal Matching8 Jun 2026
- 2026DualGate-Net: A Prior-Gated Dual-Encoder Framework for Histopathology Cell Detection5 Jun 2026
- 2026CodeCytos: AI-assisted spatial molecular imaging analysis via code-augmented agent action space30 May 2026
12 newest of 98 All 98 (JSON)