SegMate: Asymmetric Attention-Based Lightweight Architecture for Efficient Multi-Organ Segmentation
Not on the map. Why?
Abstract (arXiv)
State-of-the-art models for medical image segmentation achieve excellent accuracy but require substantial computational resources, limiting deployment in resource-constrained clinical settings. We present SegMate, an efficient 2.5D framework that achieves state-of-the-art accuracy, while considerably reducing computational requirements. Our efficient design is the result of meticulously integrating asymmetric architectures, attention mechanisms, multi-scale feature fusion, slice-based positional conditioning, and multi-task optimization. We demonstrate the efficiency-accuracy trade-off of our framework across three modern backbones (EfficientNetV2-M, MambaOut-Tiny, FastViT-T12). We perform experiments on three datasets: TotalSegmentator, SegTHOR and AMOS22. Compared with the vanilla models, SegMate reduces computation (GFLOPs) by up to 2.5x and memory footprint (VRAM) by up to 2.1x, while generally registering performance gains of around 1%. On TotalSegmentator, we achieve a Dice score of 93.51% with only 295MB peak GPU memory. Zero-shot cross-dataset evaluations on SegTHOR and AMOS22 demonstrate strong generalization, with Dice scores of up to 86.85% and 89.35%, respectively. We release our open-source code at https://github.com/andreibunea99/SegMate.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 41 references that are in the map. First posted on arXiv 27 Feb 2026.
Builds on
26 of 41 references.
References · 26
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2017Squeeze-and-Excitation Networkscited by 4,763
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
- 2018CBAM: Convolutional Block Attention Modulecited by 2,463
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,471
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
- 2017A Survey on Deep Learning in Medical Image Analysiscited by 1,182
14 more
- 2021EfficientNetV2: Smaller Models and Faster Trainingcited by 1,127
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 954
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 823
- 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 597
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
- 2022AMOS: A Large-Scale Abdominal Multi-Organ Benchmark for Versatile Medical Image Segmentationcited by 347
- 20183D MRI brain tumor segmentation using autoencoder regularizationcited by 324
- 2024SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentationcited by 266
- 2019MultiResUNet : Rethinking the U-Net Architecture for Multimodal Biomedical Image Segmentationcited by 195
- 2021nnFormer: Interleaved Transformer for Volumetric Segmentationcited by 179
- 2023FastViT: A Fast Hybrid Vision Transformer using Structural Reparameterizationcited by 161
- 2024MambaOut: Do We Really Need Mamba for Vision?cited by 142
- 20223D UX-Net: A Large Kernel Volumetric ConvNet Modernizing Hierarchical Transformer for Medical Image Segmentationcited by 94
- 2022Multimodal Multi-Head Convolutional Attention with Various Kernel Sizes for Medical Image Super-Resolutioncited by 17
Built on by
Not cited yet.