arXiv:2509.18493cs.CV2025-09ICCV被引 39

轻量级医学图像分割模型,参数少100倍仍更准。

MK-UNet: Multi-kernel Lightweight CNN for Medical Image Segmentation

  • 用多核深度卷积块捕捉多尺度空间特征
  • 仅0.316M参数,六项基准测试均超当前最佳
  • 适合手机、便携设备等资源受限场景

本文提出MK-UNet,一种面向医学图像分割的超轻量级多核U型卷积神经网络。核心是自研的多核深度卷积块(MKDC),通过多个卷积核协同处理,有效捕捉复杂多分辨率空间关系。同时结合通道、空间及分组门控注意力机制,强化图像显著特征。该模型仅需0.316M参数和0.314G FLOPs,却在六个二值医学图像分割基准上超越现有最先进方法。相比TransUNet,参数和计算量分别减少约333倍和123倍,且DICE分数更高;相较UNeXt,在参数减少4.7倍的情况下,DICE分数提升达6.7%。此外,其性能亦优于MedT、CMUNeXt、EGE-UNet和Rolling-UNet等近期轻量级网络,且所需计算资源更低。这一显著性能提升与计算效率的双重突破,使MK-UNet成为资源受限场景下实时高保真医疗诊断的理想选择,如床旁设备。代码已开源:https://github.com/SLDGroup/MK-UNet。

原文摘要 · Abstract (English)

In this paper, we introduce MK-UNet, a paradigm shift towards ultra-lightweight, multi-kernel U-shaped CNNs tailored for medical image segmentation. Central to MK-UNet is the multi-kernel depth-wise convolution block (MKDC) we design to adeptly process images through multiple kernels, while capturing complex multi-resolution spatial relationships. MK-UNet also emphasizes the images salient features through sophisticated attention mechanisms, including channel, spatial, and grouped gated attention. Our MK-UNet network, with a modest computational footprint of only 0.316M parameters and 0.314G FLOPs, represents not only a remarkably lightweight, but also significantly improved segmentation solution that provides higher accuracy over state-of-the-art (SOTA) methods across six binary medical imaging benchmarks. Specifically, MK-UNet outperforms TransUNet in DICE score with nearly 333$\times$ and 123$\times$ fewer parameters and FLOPs, respectively. Similarly, when compared against UNeXt, MK-UNet exhibits superior segmentation performance, improving the DICE score up to 6.7% margins while operating with 4.7$\times$ fewer #Params. Our MK-UNet also outperforms other recent lightweight networks, such as MedT, CMUNeXt, EGE-UNet, and Rolling-UNet, with much lower computational resources. This leap in performance, coupled with drastic computational gains, positions MK-UNet as an unparalleled solution for real-time, high-fidelity medical diagnostics in resource-limited settings, such as point-of-care devices. Our implementation is available at https://github.com/SLDGroup/MK-UNet.

医学图像轻量模型分割多核卷积

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。