A Robust Volumetric Transformer for Accurate 3D Tumor Segmentation
Not on the map. Why?
Abstract (arXiv)
We propose a Transformer architecture for volumetric segmentation, a challenging task that requires keeping a complex balance in encoding local and global spatial cues, and preserving information along all axes of the volume. Encoder of the proposed design benefits from self-attention mechanism to simultaneously encode local and global cues, while the decoder employs a parallel self and cross attention formulation to capture fine details for boundary refinement. Empirically, we show that the proposed design choices result in a computationally efficient model, with competitive and promising results on the Medical Segmentation Decathlon (MSD) brain tumor segmentation (BraTS) Task. We further show that the representations learned by our model are robust against data corruptions. hrefhttps://github.com/himashi92/VT-UNetOur code implementation is publicly available.
Key facts. Cited by 41 papers in the map (data through 14 Oct 2026). Builds on 24 of its 30 references that are in the map. First posted on arXiv 26 Nov 2021.
Builds on
24 of 30 references.
References · 24
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2020End-to-End Object Detection with Transformerscited by 7,414
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 2021ViViT: A Video Vision Transformercited by 1,620
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,558
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
12 more
- 2020Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformerscited by 1,369
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2021Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetcited by 1,101
- 2021Video Swin Transformercited by 1,091
- 2021Transformers in Vision: A Surveycited by 1,013
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
- 2021The Medical Segmentation Decathloncited by 628
- 2021Intriguing Properties of Vision Transformerscited by 446
- 2021TransBTS: Multimodal Brain Tumor Segmentation Using Transformercited by 265
- 2021CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentationcited by 227
- 2019Local Relation Networks for Image Recognitioncited by 210
- 2020nnU-Net for Brain Tumor Segmentationcited by 103
Built on by
Citing papers per year: 2022: 6; 2023: 12; 2024: 12; 2025: 9; 2026: 2.
41 papers.
Sort:
By area:
- 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentationcited by 162
- 2023SwinMM: Masked Multi-view with Swin Transformers for 3D Medical Image Segmentationcited by 35
- 2022Transforming medical imaging with Transformers? A comparative review of key properties, current progresses, and future perspectivescited by 33
- 2022CKD-TransBTS: Clinical Knowledge-Driven Hybrid Transformer with Modality-Correlated Cross-Attention for Brain Tumor Segmentationcited by 24
- 2025Primus: Enforcing Attention Usage for 3D Medical Image Segmentationcited by 24
- 2023A Recent Survey of Vision Transformers for Medical Image Segmentationcited by 23
- 20233D TransUNet: Advancing Medical Image Segmentation through Vision Transformerscited by 20
- 2022A survey on attention mechanisms for medical applications: are we moving towards better algorithms?cited by 15
- 2024EM-Net: Efficient Channel and Frequency Learning with Mamba for 3D Medical Image Segmentationcited by 12
- 2022Recent Progress in Transformer-based Medical Image Analysiscited by 9
- 2024D-Net: Dynamic Large Kernel with Dynamic Feature Fusion for Volumetric Medical Image Segmentationcited by 9
- 2025Rethinking Brain Tumor Segmentation from the Frequency Domain Perspectivecited by 9
29 more
- 2023CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentationcited by 7
- 2022Multi-branch Cascaded Swin Transformers with Attention to k-space Sampling Pattern for Accelerated MRI Reconstructioncited by 4
- 2023Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensemblescited by 4
- 2022PIP: Positional-encoding Image Priorcited by 3
- 2023Boosting Convolution with Efficient MLP-Permutation for Volumetric Medical Image Segmentationcited by 3
- 2023RadarFormer: Lightweight and Accurate Real-Time Radar Object Detection Modelcited by 3
- 2023Med-Tuning: A New Parameter-Efficient Tuning Framework for Medical Volumetric Segmentationcited by 3
- 2025Concentrate on Weakness: Mining Hard Prototypes for Few-Shot Medical Image Segmentationcited by 3
- 2023Transformer-based end-to-end classification of variable-length volumetric datacited by 2
- 2023Deep evidential fusion with uncertainty quantification and contextual discounting for multimodal medical image segmentationcited by 2
- 2024Architecture Analysis and Benchmarking of 3D U-shaped Deep Learning Models for Thoracic Anatomical Segmentationcited by 2
- 2024DocParseNet: Advanced Semantic Segmentation and OCR Embeddings for Efficient Scanned Document Annotationcited by 2
- 2024Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detectioncited by 2
- 2024Spectral U-Net: Enhancing Medical Image Segmentation via Spectral Decompositioncited by 2
- 2025GBT-SAM: A Parameter-Efficient Depth-Aware Model for Generalizable Brain tumour Segmentation on mp-MRIcited by 2
- 2024UKAN-EP: Enhancing U-KAN with Efficient Attention and Pyramid Aggregation for 3D Multi-Modal MRI Brain Tumor Segmentationcited by 1
- 2025QMaxViT-Unet+: A Query-Based MaxViT-Unet with Edge Enhancement for Scribble-Supervised Segmentation of Medical Imagescited by 1
- 2025Trans2-CBCT: A Dual-Transformer Framework for Sparse-View CBCT Reconstructioncited by 1
- 2023CATS v2: Hybrid encoders for robust medical segmentationcited by 0
- 2024Modality-Aware and Shift Mixer for Multi-modal Brain Tumor Segmentationcited by 0
- 20243D-EffiViTCaps: 3D Efficient Vision Transformer with Capsule for Medical Image Segmentationcited by 0
- 2024A dual-task mutual learning framework for predicting post-thrombectomy cerebral hemorrhagecited by 0
- 2024Bilateral Hippocampi Segmentation in Low Field MRIs Using Mutual Feature Learning via Dual-Viewscited by 0
- 2024End-to-end Triple-domain PET Enhancement: A Hybrid Denoising-and-reconstruction Framework for Reconstructing Standard-dose PET Images from Low-dose PET Sinogramscited by 0
- 2025Selective Complementary Feature Fusion and Modal Feature Compression Interaction for Brain Tumor Segmentationcited by 0
- 2025MSD-KMamba: Bidirectional Spatial-Aware Multi-Modal 3D Brain Segmentation via Multi-scale Self-Distilled Fusion Strategycited by 0
- 2025Pancakes: Consistent Multi-Protocol Image Segmentation Across Biomedical Domainscited by 0
- 2026Probing and Inducing Clinical Modality Grounding in Brain Tumor Segmentationcited by 0
- 2026MCSeg: Pre-training and Fine-tuning Volumetric Pyramid Transformer for Multi-modal Cardiac Image Segmentationcited by 0
- 2026MCSeg: Pre-training and Fine-tuning Volumetric Pyramid Transformer for Multi-modal Cardiac Image Segmentation31 Aug 2026
- 2026Probing and Inducing Clinical Modality Grounding in Brain Tumor Segmentation20 Mar 2026
- 2025Pancakes: Consistent Multi-Protocol Image Segmentation Across Biomedical Domains15 Dec 2025
- 2025MSD-KMamba: Bidirectional Spatial-Aware Multi-Modal 3D Brain Segmentation via Multi-scale Self-Distilled Fusion Strategy28 Sep 2025
- 2025Trans2-CBCT: A Dual-Transformer Framework for Sparse-View CBCT Reconstruction20 Jun 2025
- 2025Rethinking Brain Tumor Segmentation from the Frequency Domain Perspective11 Jun 2025
- 2025Concentrate on Weakness: Mining Hard Prototypes for Few-Shot Medical Image Segmentation28 May 2025
- 2025Selective Complementary Feature Fusion and Modal Feature Compression Interaction for Brain Tumor Segmentation20 Mar 2025
- 2025GBT-SAM: A Parameter-Efficient Depth-Aware Model for Generalizable Brain tumour Segmentation on mp-MRI6 Mar 2025
- 2025Primus: Enforcing Attention Usage for 3D Medical Image Segmentation3 Mar 2025
- 2025QMaxViT-Unet+: A Query-Based MaxViT-Unet with Edge Enhancement for Scribble-Supervised Segmentation of Medical Images14 Feb 2025
- 2024End-to-end Triple-domain PET Enhancement: A Hybrid Denoising-and-reconstruction Framework for Reconstructing Standard-dose PET Images from Low-dose PET Sinograms4 Dec 2024
29 more
- 2024Bilateral Hippocampi Segmentation in Low Field MRIs Using Mutual Feature Learning via Dual-Views23 Oct 2024
- 2024EM-Net: Efficient Channel and Frequency Learning with Mamba for 3D Medical Image Segmentation26 Sep 2024
- 2024Spectral U-Net: Enhancing Medical Image Segmentation via Spectral Decomposition13 Sep 2024
- 2024A dual-task mutual learning framework for predicting post-thrombectomy cerebral hemorrhage1 Aug 2024
- 2024UKAN-EP: Enhancing U-KAN with Efficient Attention and Pyramid Aggregation for 3D Multi-Modal MRI Brain Tumor Segmentation1 Aug 2024
- 2024Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detection1 Jul 2024
- 2024DocParseNet: Advanced Semantic Segmentation and OCR Embeddings for Efficient Scanned Document Annotation25 Jun 2024
- 20243D-EffiViTCaps: 3D Efficient Vision Transformer with Capsule for Medical Image Segmentation25 Mar 2024
- 2024D-Net: Dynamic Large Kernel with Dynamic Feature Fusion for Volumetric Medical Image Segmentation15 Mar 2024
- 2024Modality-Aware and Shift Mixer for Multi-modal Brain Tumor Segmentation4 Mar 2024
- 2024Architecture Analysis and Benchmarking of 3D U-shaped Deep Learning Models for Thoracic Anatomical Segmentation5 Feb 2024
- 2023A Recent Survey of Vision Transformers for Medical Image Segmentation1 Dec 2023
- 2023CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentation8 Nov 2023
- 2023Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensembles24 Oct 2023
- 20233D TransUNet: Advancing Medical Image Segmentation through Vision Transformers11 Oct 2023
- 2023Deep evidential fusion with uncertainty quantification and contextual discounting for multimodal medical image segmentation12 Sep 2023
- 2023CATS v2: Hybrid encoders for robust medical segmentation11 Aug 2023
- 2023SwinMM: Masked Multi-view with Swin Transformers for 3D Medical Image Segmentation24 Jul 2023
- 2023Transformer-based end-to-end classification of variable-length volumetric data13 Jul 2023
- 2023Med-Tuning: A New Parameter-Efficient Tuning Framework for Medical Volumetric Segmentation21 Apr 2023
- 2023RadarFormer: Lightweight and Accurate Real-Time Radar Object Detection Model17 Apr 2023
- 2023Boosting Convolution with Efficient MLP-Permutation for Volumetric Medical Image Segmentation23 Mar 2023
- 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentation17 Mar 2023
- 2022PIP: Positional-encoding Image Prior25 Nov 2022
- 2022Recent Progress in Transformer-based Medical Image Analysis13 Aug 2022
- 2022Multi-branch Cascaded Swin Transformers with Attention to k-space Sampling Pattern for Accelerated MRI Reconstruction18 Jul 2022
- 2022CKD-TransBTS: Clinical Knowledge-Driven Hybrid Transformer with Modality-Correlated Cross-Attention for Brain Tumor Segmentation15 Jul 2022
- 2022Transforming medical imaging with Transformers? A comparative review of key properties, current progresses, and future perspectives2 Jun 2022
- 2022A survey on attention mechanisms for medical applications: are we moving towards better algorithms?26 Apr 2022
- 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentationcited by 162
- 2023SwinMM: Masked Multi-view with Swin Transformers for 3D Medical Image Segmentationcited by 35
- 2022Transforming medical imaging with Transformers? A comparative review of key properties, current progresses, and future perspectivescited by 33
- 2022CKD-TransBTS: Clinical Knowledge-Driven Hybrid Transformer with Modality-Correlated Cross-Attention for Brain Tumor Segmentationcited by 24
- 2025Primus: Enforcing Attention Usage for 3D Medical Image Segmentationcited by 24
- 2023A Recent Survey of Vision Transformers for Medical Image Segmentationcited by 23
- 20233D TransUNet: Advancing Medical Image Segmentation through Vision Transformerscited by 20
- 2022A survey on attention mechanisms for medical applications: are we moving towards better algorithms?cited by 15
- 2022Recent Progress in Transformer-based Medical Image Analysiscited by 9
- 2024D-Net: Dynamic Large Kernel with Dynamic Feature Fusion for Volumetric Medical Image Segmentationcited by 9
- 2025Rethinking Brain Tumor Segmentation from the Frequency Domain Perspectivecited by 9
- 2023CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentationcited by 7
12 most cited of 38 All 38 (JSON)
- 2026MCSeg: Pre-training and Fine-tuning Volumetric Pyramid Transformer for Multi-modal Cardiac Image Segmentation31 Aug 2026
- 2026Probing and Inducing Clinical Modality Grounding in Brain Tumor Segmentation20 Mar 2026
- 2025Pancakes: Consistent Multi-Protocol Image Segmentation Across Biomedical Domains15 Dec 2025
- 2025Trans2-CBCT: A Dual-Transformer Framework for Sparse-View CBCT Reconstruction20 Jun 2025
- 2025Rethinking Brain Tumor Segmentation from the Frequency Domain Perspective11 Jun 2025
- 2025Concentrate on Weakness: Mining Hard Prototypes for Few-Shot Medical Image Segmentation28 May 2025
- 2025Selective Complementary Feature Fusion and Modal Feature Compression Interaction for Brain Tumor Segmentation20 Mar 2025
- 2025GBT-SAM: A Parameter-Efficient Depth-Aware Model for Generalizable Brain tumour Segmentation on mp-MRI6 Mar 2025
- 2025Primus: Enforcing Attention Usage for 3D Medical Image Segmentation3 Mar 2025
- 2025QMaxViT-Unet+: A Query-Based MaxViT-Unet with Edge Enhancement for Scribble-Supervised Segmentation of Medical Images14 Feb 2025
- 2024End-to-end Triple-domain PET Enhancement: A Hybrid Denoising-and-reconstruction Framework for Reconstructing Standard-dose PET Images from Low-dose PET Sinograms4 Dec 2024
- 2024Bilateral Hippocampi Segmentation in Low Field MRIs Using Mutual Feature Learning via Dual-Views23 Oct 2024
12 newest of 38 All 38 (JSON)