mlsanityData to 7 Oct 2026

SegMate: Asymmetric Attention-Based Lightweight Architecture for Efficient Multi-Organ Segmentation

arXiv 2602.23903 ↗First posted 27 Feb 2026cs.CV cs.LGcited by 0

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

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  2. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  3. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
  4. 2017Squeeze-and-Excitation Networkscited by 4,763
  5. 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
  6. 2018CBAM: Convolutional Block Attention Modulecited by 2,463
  7. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
  8. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  9. 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
  10. 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,471
  11. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
  12. 2017A Survey on Deep Learning in Medical Image Analysiscited by 1,182
14 more
  1. 2021EfficientNetV2: Smaller Models and Faster Trainingcited by 1,127
  2. 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 954
  3. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 823
  4. 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 597
  5. 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
  6. 2022AMOS: A Large-Scale Abdominal Multi-Organ Benchmark for Versatile Medical Image Segmentationcited by 347
  7. 20183D MRI brain tumor segmentation using autoencoder regularizationcited by 324
  8. 2024SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentationcited by 266
  9. 2019MultiResUNet : Rethinking the U-Net Architecture for Multimodal Biomedical Image Segmentationcited by 195
  10. 2021nnFormer: Interleaved Transformer for Volumetric Segmentationcited by 179
  11. 2023FastViT: A Fast Hybrid Vision Transformer using Structural Reparameterizationcited by 161
  12. 2024MambaOut: Do We Really Need Mamba for Vision?cited by 142
  13. 20223D UX-Net: A Large Kernel Volumetric ConvNet Modernizing Hierarchical Transformer for Medical Image Segmentationcited by 94
  14. 2022Multimodal Multi-Head Convolutional Attention with Various Kernel Sizes for Medical Image Super-Resolutioncited by 17

Built on by

Not cited yet.