轻量级医学图像分割模型SimpleUNet,以极简设计实现高精度与高效能。
Simple is what you need for efficient and accurate medical image segmentation
- 通过跳接路径的局部特征选择减少冗余,提升分割效果。
- 0.67MB版本在乳腺病变数据集上达85.76% DSC,优于U-Net和TransUNet。
- 适合部署于资源受限的医疗边缘设备,兼顾速度与准确率。
尽管现代分割模型常侧重性能而忽视实用性,本文倡导以简洁与高效为核心的设计理念,提出SimpleUNet——一种可扩展的超轻量级医学图像分割模型。其三大创新包括:(1)跳接路径中的部分特征选择机制,降低冗余同时提升性能;(2)固定宽度架构,防止网络阶段间参数指数级增长;(3)自适应特征融合模块,在极低计算开销下增强表征能力。在16 KB参数配置下创下纪录,0.67 MB版本在多中心乳腺病灶数据集上达到85.76% mean DSC / 75.60% IoU,显著优于LBUNet、U-Net及TransUNet。在皮肤病变数据集(ISIC 2017/2018: mDice 84.86%/88.77%)与内窥镜息肉分割(KVASIR-SEG: 86.46%/76.48% mDice/mIoU)上亦持续领先。结果表明,极致压缩无需牺牲性能,为高效精准医疗图像分割提供新思路。代码详见https://github.com/Frankyu5666666/SimpleUNet。
原文摘要 · Abstract (English)
While modern segmentation models often prioritize performance over practicality, we advocate a design philosophy prioritizing simplicity and efficiency, and attempted high performance segmentation model design. This paper presents SimpleUNet, a scalable ultra-lightweight medical image segmentation model with three key innovations: (1) A partial feature selection mechanism in skip connections for redundancy reduction while enhancing segmentation performance; (2) A fixed-width architecture that prevents exponential parameter growth across network stages; (3) An adaptive feature fusion module achieving enhanced representation with minimal computational overhead. With a record-breaking 16 KB parameter configuration, SimpleUNet outperforms LBUNet and other lightweight benchmarks across multiple public datasets. The 0.67 MB variant achieves superior efficiency (8.60 GFLOPs) and accuracy, attaining a mean DSC/IoU of 85.76%/75.60% on multi-center breast lesion datasets, surpassing both U-Net and TransUNet. Evaluations on skin lesion datasets (ISIC 2017/2018: mDice 84.86%/88.77%) and endoscopic polyp segmentation (KVASIR-SEG: 86.46%/76.48% mDice/mIoU) confirm consistent dominance over state-of-the-art models. This work demonstrates that extreme model compression need not compromise performance, providing new insights for efficient and accurate medical image segmentation. Codes can be found at https://github.com/Frankyu5666666/SimpleUNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。