SegViT: Semantic Segmentation with Plain Vision Transformers
Not on the map. Why?
Abstract (arXiv)
We explore the capability of plain Vision Transformers (ViTs) for semantic segmentation and propose the SegVit. Previous ViT-based segmentation networks usually learn a pixel-level representation from the output of the ViT. Differently, we make use of the fundamental component -- attention mechanism, to generate masks for semantic segmentation. Specifically, we propose the Attention-to-Mask (ATM) module, in which the similarity maps between a set of learnable class tokens and the spatial feature maps are transferred to the segmentation masks. Experiments show that our proposed SegVit using the ATM module outperforms its counterparts using the plain ViT backbone on the ADE20K dataset and achieves new state-of-the-art performance on COCO-Stuff-10K and PASCAL-Context datasets. Furthermore, to reduce the computational cost of the ViT backbone, we propose query-based down-sampling (QD) and query-based up-sampling (QU) to build a Shrunk structure. With the proposed Shrunk structure, the model can save up to 40% computations while maintaining competitive performance.
Key facts. Cited by 106 papers in the map (data through 14 Oct 2026). Builds on 40 of its 49 references that are in the map. First posted on arXiv 12 Oct 2022.
Builds on
40 of 49 references.
References · 40
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2020End-to-End Object Detection with Transformerscited by 7,414
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,859
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,223
- 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,940
- 2016Pyramid Scene Parsing Networkcited by 2,479
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,351
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
28 more
- 2021Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutionscited by 1,962
- 2021Vision Transformers for Dense Predictioncited by 1,803
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 2020Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformerscited by 1,369
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,271
- 2021Per-Pixel Classification is Not All You Need for Semantic Segmentationcited by 1,152
- 2018Dual Attention Network for Scene Segmentationcited by 1,003
- 2016COCO-Stuff: Thing and Stuff Classes in Contextcited by 864
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2019Segmentation Transformer: Object-Contextual Representations for Semantic Segmentationcited by 576
- 2021How to train your ViT? Data, Augmentation, and Regularization in Vision Transformerscited by 569
- 2021Twins: Revisiting the Design of Spatial Attention in Vision Transformerscited by 565
- 2016RefineNet: Multi-Path Refinement Networks for High-Resolution Semantic Segmentationcited by 465
- 2018Context Encoding for Semantic Segmentationcited by 290
- 2021K-Net: Towards Unified Image Segmentationcited by 212
- 2019High-Resolution Representations for Labeling Pixels and Regionscited by 211
- 2019Expectation-Maximization Attention Networks for Semantic Segmentationcited by 147
- 2020Context Prior for Scene Segmentationcited by 121
- 2021P2T: Pyramid Pooling Transformer for Scene Understandingcited by 84
- 2019GFF: Gated Fully Fusion for Semantic Segmentationcited by 57
- 2021Mining Contextual Information Beyond Image for Semantic Segmentationcited by 48
- 2020Spatial Pyramid Based Graph Reasoning for Semantic Segmentationcited by 47
- 2021ISNet: Integrate Image-Level and Semantic-Level Context for Semantic Segmentationcited by 40
- 2022StructToken : Rethinking Semantic Segmentation with Structural Priorcited by 30
- 2020Tensor Low-Rank Reconstruction for Semantic Segmentationcited by 29
- 2020EfficientFCN: Holistically-guided Decoding for Semantic Segmentationcited by 15
- 2020GINet: Graph Interaction Network for Scene Parsingcited by 13
- 2021Dynamic Neural Representational Decoders for High-Resolution Semantic Segmentationcited by 11
Built on by
Citing papers per year: 2022: 3; 2023: 30; 2024: 28; 2025: 32; 2026: 13.
106 papers.
Sort:
By area:
- 2022ZegCLIP: Towards Adapting CLIP for Zero-shot Semantic Segmentationcited by 178
- 2023CLIPSelf: Vision Transformer Distills Itself for Open-Vocabulary Dense Predictioncited by 118
- 2023Delivering Arbitrary-Modal Semantic Segmentationcited by 117
- 2023Region-Aware Pretraining for Open-Vocabulary Object Detection with Vision Transformerscited by 91
- 2023Stronger, Fewer, & Superior: Harnessing Vision Foundation Models for Domain Generalized Semantic Segmentationcited by 83
- 2025Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Groundingcited by 83
- 2022ViTPose++: Vision Transformer for Generic Body Pose Estimationcited by 65
- 2023Dynamic Token Pruning in Plain Vision Transformers for Semantic Segmentationcited by 62
- 2023A Survey on Open-Vocabulary Detection and Segmentation: Past, Present, and Futurecited by 61
- 2024LaSagnA: Language-based Segmentation Assistant for Complex Queriescited by 55
- 2023Exploring Plain ViT Reconstruction for Multi-class Unsupervised Anomaly Detectioncited by 47
- 2023Conflict-Based Cross-View Consistency for Semi-Supervised Semantic Segmentationcited by 35
38 more
- 2023Contrastive Feature Masking Open-Vocabulary Vision Transformercited by 31
- 2024ALGM: Adaptive Local-then-Global Token Merging for Efficient Semantic Segmentation with Plain Vision Transformerscited by 28
- 2023SegViTv2: Exploring Efficient and Continual Semantic Segmentation with Plain Vision Transformerscited by 25
- 2024Context-Guided Spatial Feature Reconstruction for Efficient Semantic Segmentationcited by 25
- 2025USP: Unified Self-Supervised Pretraining for Image Generation and Understandingcited by 24
- 2024Parameter-Efficient Fine-Tuning for Pre-Trained Vision Models: A Survey and Benchmarkcited by 23
- 2023InvPT++: Inverted Pyramid Multi-Task Transformer for Visual Scene Understandingcited by 22
- 2023SCTNet: Single-Branch CNN with Transformer Semantic Information for Real-Time Segmentationcited by 22
- 2024Centering the Value of Every Modality: Towards Efficient and Resilient Modality-agnostic Semantic Segmentationcited by 22
- 2024Multi-Layer Transformers Gradient Can be Approximated in Almost Linear Timecited by 22
- 2024Context-Aware Interaction Network for RGB-T Semantic Segmentationcited by 21
- 2024Textual Query-Driven Mask Transformer for Domain Generalized Segmentationcited by 21
- 2026Generation Enhances Understanding in Unified Multimodal Models via Multi-Representation Generationcited by 20
- 2023Scattering Vision Transformer: Spectral Mixing Matterscited by 17
- 2024Cascade-CLIP: Cascaded Vision-Language Embeddings Alignment for Zero-Shot Semantic Segmentationcited by 17
- 2025Hypergraph Vision Transformers: Images are More than Nodes, More than Edgescited by 16
- 2023BPKD: Boundary Privileged Knowledge Distillation For Semantic Segmentationcited by 15
- 2023TagCLIP: Improving Discrimination Ability of Open-Vocabulary Semantic Segmentationcited by 14
- 2023Win-Win: Training High-Resolution Vision Transformers from Two Windowscited by 12
- 2024Open-Vocabulary Segmentation with Unpaired Mask-Text Supervisioncited by 12
- 2024MAGIC++: Efficient and Resilient Modality-Agnostic Semantic Segmentation via Hierarchical Modality Selectioncited by 12
- 2024CatLIP: CLIP-level Visual Recognition Accuracy with 2.7x Faster Pre-training on Web-scale Image-Text Datacited by 10
- 2024LookHere: Vision Transformers with Directed Attention Generalize and Extrapolatecited by 10
- 2025Frequency-Dynamic Attention Modulation for Dense Predictioncited by 10
- 2023Stitched ViTs are Flexible Vision Backbonescited by 8
- 2023APTv2: Benchmarking Animal Pose Estimation and Tracking with a Large-scale Dataset and Beyondcited by 8
- 2024Memorize What Matters: Emergent Scene Decomposition from Multitraversecited by 8
- 2024RGB2Point: 3D Point Cloud Generation from Single RGB Imagescited by 7
- 2025Dynamic Dictionary Learning for Remote Sensing Image Segmentationcited by 7
- 2024Language-Driven Visual Consensus for Zero-Shot Semantic Segmentationcited by 6
- 2024MLIP: Efficient Multi-Perspective Language-Image Pretraining with Exhaustive Data Utilizationcited by 6
- 2024CathAction: A Benchmark for Endovascular Intervention Understandingcited by 6
- 2023Multi-Scale And Token Mergence: Make Your ViT More Efficientcited by 5
- 2025FocusTrack: A Self-Adaptive Local Sampling Algorithm for Efficient Anti-UAV Trackingcited by 5
- 2025Revisiting Efficient Semantic Segmentation: Learning Offsets for Better Spatial and Class Feature Alignmentcited by 5
- 2022Representation Separation for Semantic Segmentation with Vision Transformerscited by 4
- 2023ShiftNAS: Improving One-shot NAS via Probability Shiftcited by 4
- 2024SSA-Seg: Semantic and Spatial Adaptive Pixel-level Classifier for Semantic Segmentationcited by 4
50 most cited of 106 All 106 (JSON)
- 2026FLASHSWIN: Unlocking Large Windows and Dense Tokens in Swin Vision Transformers with Memory Efficient Attention3 Oct 2026
- 2026Decoder-Agnostic Token Merging for Vision Transformers: A Systematic Study of G2TM16 Sep 2026
- 2026SELECT: SELEctive Context Transfer for Class-Incremental Semantic Segmentation31 Aug 2026
- 2026DiD It in 87 Minutes: A Label-Free Softmax-to-Linear Adaptation of Vision Transformers for Object Detection23 Aug 2026
- 2026Learning a Sampling-Free Variational DNN Plugin from Tiny Training Sets to Refine OOD Segmentation With Uncertainty Estimation14 Jun 2026
- 2026Towards Real-Time Autonomous Navigation: Transformer-Based Catheter Tip Tracking in Fluoroscopy14 May 2026
- 2026MPM: Mutual Pair Merging for Efficient Vision Transformers7 Apr 2026
- 2026LiPS: Lightweight Panoptic Segmentation for Resource-Constrained Robotics1 Apr 2026
- 2026DCP-CLIP:A Coarse-to-Fine Framework for Open-Vocabulary Semantic Segmentation with Dual Interaction14 Mar 2026
- 2026Combining Microscopy Data and Metadata for Reconstruction of Cellular Traction Forces Using a Hybrid Vision Transformer-U-Net11 Mar 2026
- 2026Risk-Aware World Model Predictive Control for Generalizable End-to-End Autonomous Driving26 Feb 2026
- 2026Generation Enhances Understanding in Unified Multimodal Models via Multi-Representation Generation29 Jan 2026
38 more
- 2026Urban Socio-Semantic Segmentation with Vision-Language Reasoning15 Jan 2026
- 2025BiCoR-Seg: Bidirectional Co-Refinement Framework for High-Resolution Remote Sensing Image Segmentation23 Dec 2025
- 2025LithoSeg: A Coarse-to-Fine Framework for High-Precision Lithography Segmentation15 Nov 2025
- 2025Differentiable Hierarchical Visual Tokenization4 Nov 2025
- 2025BARL: Bilateral Alignment in Representation and Label Spaces for Semi-Supervised Volumetric Medical Image Segmentation19 Oct 2025
- 2025ClustViT: Clustering-based Token Merging for Semantic Segmentation2 Oct 2025
- 2025Where Do Tokens Go? Understanding Pruning Behaviors in STEP at High Resolutions17 Sep 2025
- 2025I-Segmenter: Integer-Only Vision Transformer for Efficient Semantic Segmentation12 Sep 2025
- 2025Patch-Level Kernel Alignment for Dense Self-Supervised Learning6 Sep 2025
- 2025Integrating SAM Supervision for 3D Weakly Supervised Point Cloud Segmentation27 Aug 2025
- 2025Exploiting Information Redundancy in Attention Maps for Extreme Quantization of Vision Transformers22 Aug 2025
- 2025Revisiting Efficient Semantic Segmentation: Learning Offsets for Better Spatial and Class Feature Alignment12 Aug 2025
- 2025VesselRW: Weakly Supervised Subcutaneous Vessel Segmentation via Learned Random Walk Propagation9 Aug 2025
- 2025DualResolution Residual Architecture with Artifact Suppression for Melanocytic Lesion Segmentation9 Aug 2025
- 2025Edge Detection for Organ Boundaries via Top Down Refinement and SubPixel Upsampling9 Aug 2025
- 2025Revisiting Continual Semantic Segmentation with Pre-trained Vision Models6 Aug 2025
- 2025Deeply Dual Supervised learning for melanoma recognition4 Aug 2025
- 2025Rein++: Efficient Generalization and Adaptation for Semantic Segmentation with Vision Foundation Models3 Aug 2025
- 2025Glass Surface Segmentation with an RGB-D Camera via Weighted Feature Fusion for Service Robots3 Aug 2025
- 2025FastSmoothSAM: A Fast Smooth Method For Segment Anything Model20 Jul 2025
- 2025Frequency-Dynamic Attention Modulation for Dense Prediction16 Jul 2025
- 2025Spatial Frequency Modulation for Semantic Segmentation16 Jul 2025
- 2025Improving Black-Box Generative Attacks via Generator Semantic Consistency23 Jun 2025
- 2025Semantic segmentation with reward23 May 2025
- 2025EGFormer: Towards Efficient and Generalizable Multimodal Semantic Segmentation20 May 2025
- 2025FocusTrack: A Self-Adaptive Local Sampling Algorithm for Efficient Anti-UAV Tracking18 Apr 2025
- 2025Hypergraph Vision Transformers: Images are More than Nodes, More than Edges11 Apr 2025
- 2025Dynamic Dictionary Learning for Remote Sensing Image Segmentation9 Mar 2025
- 2025Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Grounding8 Mar 2025
- 2025USP: Unified Self-Supervised Pretraining for Image Generation and Understanding8 Mar 2025
- 2025VPNeXt -- Rethinking Dense Decoding for Plain Vision Transformer23 Feb 2025
- 2025Thicker and Quicker: A Jumbo Token for Fast Plain Vision Transformers20 Feb 2025
- 2025Vision Transformer Neural Architecture Search for Out-of-Distribution Generalization: Benchmark and Insights7 Jan 2025
- 2024MAGIC++: Efficient and Resilient Modality-Agnostic Semantic Segmentation via Hierarchical Modality Selection22 Dec 2024
- 2024LVS-Net: A Lightweight Vessels Segmentation Network for Retinal Image Analysis8 Dec 2024
- 2024Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspective5 Nov 2024
- 2024EP-SAM: Weakly Supervised Histopathology Segmentation via Enhanced Prompt with Segment Anything17 Oct 2024
- 2024big.LITTLE Vision Transformer for Efficient Visual Recognition14 Oct 2024
50 newest of 106 All 106 (JSON)
- 2023Dynamic Token Pruning in Plain Vision Transformers for Semantic Segmentationcited by 62
- 2024ALGM: Adaptive Local-then-Global Token Merging for Efficient Semantic Segmentation with Plain Vision Transformerscited by 28
- 2023Multi-Scale And Token Mergence: Make Your ViT More Efficientcited by 5
- 2026MPM: Mutual Pair Merging for Efficient Vision Transformerscited by 2
- 2025Where Do Tokens Go? Understanding Pruning Behaviors in STEP at High Resolutionscited by 0
- 2026Decoder-Agnostic Token Merging for Vision Transformers: A Systematic Study of G2TMcited by 0
- 2026Decoder-Agnostic Token Merging for Vision Transformers: A Systematic Study of G2TM16 Sep 2026
- 2026MPM: Mutual Pair Merging for Efficient Vision Transformers7 Apr 2026
- 2025Where Do Tokens Go? Understanding Pruning Behaviors in STEP at High Resolutions17 Sep 2025
- 2024ALGM: Adaptive Local-then-Global Token Merging for Efficient Semantic Segmentation with Plain Vision Transformers14 Jun 2024
- 2023Dynamic Token Pruning in Plain Vision Transformers for Semantic Segmentation2 Aug 2023
- 2023Multi-Scale And Token Mergence: Make Your ViT More Efficient8 Jun 2023
- 2025Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Groundingcited by 83
- 2024Textual Query-Driven Mask Transformer for Domain Generalized Segmentationcited by 21
- 2024Cascade-CLIP: Cascaded Vision-Language Embeddings Alignment for Zero-Shot Semantic Segmentationcited by 17
- 2024Language-Driven Visual Consensus for Zero-Shot Semantic Segmentationcited by 6
- 2025Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Grounding8 Mar 2025
- 2024Textual Query-Driven Mask Transformer for Domain Generalized Segmentation12 Jul 2024
- 2024Cascade-CLIP: Cascaded Vision-Language Embeddings Alignment for Zero-Shot Semantic Segmentation2 Jun 2024
- 2024Language-Driven Visual Consensus for Zero-Shot Semantic Segmentation13 Mar 2024
- 2022ZegCLIP: Towards Adapting CLIP for Zero-shot Semantic Segmentationcited by 178
- 2023CLIPSelf: Vision Transformer Distills Itself for Open-Vocabulary Dense Predictioncited by 118
- 2023Delivering Arbitrary-Modal Semantic Segmentationcited by 117
- 2023Region-Aware Pretraining for Open-Vocabulary Object Detection with Vision Transformerscited by 91
- 2023Stronger, Fewer, & Superior: Harnessing Vision Foundation Models for Domain Generalized Semantic Segmentationcited by 83
- 2022ViTPose++: Vision Transformer for Generic Body Pose Estimationcited by 65
- 2023A Survey on Open-Vocabulary Detection and Segmentation: Past, Present, and Futurecited by 61
- 2024LaSagnA: Language-based Segmentation Assistant for Complex Queriescited by 55
- 2023Exploring Plain ViT Reconstruction for Multi-class Unsupervised Anomaly Detectioncited by 47
- 2023Conflict-Based Cross-View Consistency for Semi-Supervised Semantic Segmentationcited by 35
- 2023Contrastive Feature Masking Open-Vocabulary Vision Transformercited by 31
- 2023SegViTv2: Exploring Efficient and Continual Semantic Segmentation with Plain Vision Transformerscited by 25
12 most cited of 91 All 91 (JSON)
- 2026FLASHSWIN: Unlocking Large Windows and Dense Tokens in Swin Vision Transformers with Memory Efficient Attention3 Oct 2026
- 2026SELECT: SELEctive Context Transfer for Class-Incremental Semantic Segmentation31 Aug 2026
- 2026DiD It in 87 Minutes: A Label-Free Softmax-to-Linear Adaptation of Vision Transformers for Object Detection23 Aug 2026
- 2026Learning a Sampling-Free Variational DNN Plugin from Tiny Training Sets to Refine OOD Segmentation With Uncertainty Estimation14 Jun 2026
- 2026Towards Real-Time Autonomous Navigation: Transformer-Based Catheter Tip Tracking in Fluoroscopy14 May 2026
- 2026LiPS: Lightweight Panoptic Segmentation for Resource-Constrained Robotics1 Apr 2026
- 2026DCP-CLIP:A Coarse-to-Fine Framework for Open-Vocabulary Semantic Segmentation with Dual Interaction14 Mar 2026
- 2026Combining Microscopy Data and Metadata for Reconstruction of Cellular Traction Forces Using a Hybrid Vision Transformer-U-Net11 Mar 2026
- 2026Generation Enhances Understanding in Unified Multimodal Models via Multi-Representation Generation29 Jan 2026
- 2025BiCoR-Seg: Bidirectional Co-Refinement Framework for High-Resolution Remote Sensing Image Segmentation23 Dec 2025
- 2025LithoSeg: A Coarse-to-Fine Framework for High-Precision Lithography Segmentation15 Nov 2025
- 2025Differentiable Hierarchical Visual Tokenization4 Nov 2025
12 newest of 91 All 91 (JSON)