轻量级网络Med-2D SegNet实现高精度医学图像分割,参数仅207万。
Med-2D SegNet: A Light Weight Deep Neural Network for Medical 2D Image Segmentation
- 设计紧凑的Med Block,融合维度扩展与参数压缩,提升特征提取效率。
- 在20个数据集上平均Dice达89.77%,多任务表现领先。
- 适合临床部署,尤其适用于算力有限的医疗场景。
精准高效的医学图像分割对临床诊断与手术规划至关重要,但受解剖结构差异和低复杂度模型需求制约,仍具挑战。本文提出Med-2D SegNet,一种新型高效分割架构,在多个基准数据集(包括KVASIR-SEG、PH2、EndoVis和GLAS)上实现卓越性能,20个不同数据集平均Dice相似系数(DSC)达89.77%。其核心为紧凑的Med Block,通过维度扩展与参数缩减设计,实现精确特征提取,同时将模型参数控制在仅207万。该网络在跨数据集泛化方面表现优异,尤其在结肠镜息肉分割任务中,仅在KVASIR-SEG训练后即在未见数据集上表现良好,展现了零样本学习潜力。在二分类与多分类任务中均达到顶尖水平,重新定义了准确率与效率的平衡,为临床环境及资源受限场景中的高性能诊断工具开发提供新范式。
原文摘要 · Abstract (English)
Accurate and efficient medical image segmentation is crucial for advancing clinical diagnostics and surgical planning, yet remains a complex challenge due to the variability in anatomical structures and the demand for low-complexity models. In this paper, we introduced Med-2D SegNet, a novel and highly efficient segmentation architecture that delivers outstanding accuracy while maintaining a minimal computational footprint. Med-2D SegNet achieves state-of-the-art performance across multiple benchmark datasets, including KVASIR-SEG, PH2, EndoVis, and GLAS, with an average Dice similarity coefficient (DSC) of 89.77% across 20 diverse datasets. Central to its success is the compact Med Block, a specialized encoder design that incorporates dimension expansion and parameter reduction, enabling precise feature extraction while keeping model parameters to a low count of just 2.07 million. Med-2D SegNet excels in cross-dataset generalization, particularly in polyp segmentation, where it was trained on KVASIR-SEG and showed strong performance on unseen datasets, demonstrating its robustness in zero-shot learning scenarios, even though we acknowledge that further improvements are possible. With top-tier performance in both binary and multi-class segmentation, Med-2D SegNet redefines the balance between accuracy and efficiency, setting a new benchmark for medical image analysis. This work paves the way for developing accessible, high-performance diagnostic tools suitable for clinical environments and resource-constrained settings, making it a step forward in the democratization of advanced medical technology.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。