mlsanityData to 14 Oct 2026

Unified Start, Personalized End: Progressive Pruning for Efficient 3D Medical Image Segmentation

arXiv 2509.09267 ↗First posted 11 Sep 2025cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

3D medical image segmentation often faces heavy resource and time consumption, limiting its scalability and rapid deployment in clinical environments. Existing efficient segmentation models are typically static and manually designed prior to training, which restricts their adaptability across diverse tasks and makes it difficult to balance performance with resource efficiency. In this paper, we propose PSP-Seg, a progressive pruning framework that enables dynamic and efficient 3D segmentation. PSP-Seg begins with a redundant model and iteratively prunes redundant modules through a combination of block-wise pruning and a functional decoupling loss. We evaluate PSP-Seg on five public datasets, benchmarking it against seven state-of-the-art models and six efficient segmentation models. Results demonstrate that the lightweight variant, PSP-Seg-S, achieves performance on par with nnU-Net while reducing GPU memory usage by 42-45%, training time by 29-48%, and parameter number by 83-87% across all datasets. These findings underscore PSP-Seg's potential as a cost-effective yet high-performing alternative for widespread clinical application.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 43 references that are in the map. First posted on arXiv 11 Sep 2025.

Builds on

28 of 43 references.

References · 28

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  2. 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,198
  3. 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,136
  4. 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,121
  5. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
  6. 2021The Medical Segmentation Decathloncited by 628
  7. 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 613
  8. 2021The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classificationcited by 518
  9. 2019The Liver Tumor Segmentation Benchmark (LiTS)cited by 509
  10. 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
  11. 2018Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Accelerationcited by 367
  12. 2024nnU-Net Revisited: A Call for Rigorous Validation in 3D Medical Image Segmentationcited by 240
16 more
  1. 2021CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentationcited by 227
  2. 2023Multimodal C4: An Open, Billion-scale Corpus of Images Interleaved with Textcited by 208
  3. 2022Medical Image Segmentation Review: The success of U-Netcited by 191
  4. 2019The state of the art in kidney and kidney tumor segmentation in contrast-enhanced CT imaging: Results of the KiTS19 Challengecited by 190
  5. 2021nnFormer: Interleaved Transformer for Volumetric Segmentationcited by 180
  6. 2024Not All Experts are Equal: Efficient Expert Pruning and Skipping for Mixture-of-Experts Large Language Modelscited by 146
  7. 2022UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentationcited by 112
  8. 2023STU-Net: Scalable and Transferable Medical Image Segmentation Models Empowered by Large-Scale Supervised Pre-trainingcited by 100
  9. 2020DoDNet: Learning to segment multi-organ and tumors from multiple partially labeled datasetscited by 94
  10. 2022Task-Specific Expert Pruning for Sparse Mixture-of-Expertscited by 72
  11. 2022A Data-scalable Transformer for Medical Image Segmentation: Architecture, Model Efficiency, and Benchmarkcited by 45
  12. 2021Deep Ensembling with No Overhead for either Training or Testing: The All-Round Blessings of Dynamic Sparsitycited by 44
  13. 20193D Dilated Multi-Fiber Network for Real-time Brain Tumor Segmentation in MRIcited by 28
  14. 2022Learning from partially labeled data for multi-organ and tumor segmentationcited by 18
  15. 2023PDP: Parameter-free Differentiable Pruning is All You Needcited by 12
  16. 2024LHU-Net: a Lean Hybrid U-Net for Cost-efficient, High-performance Volumetric Segmentationcited by 4

Built on by

Not cited yet.