arXiv:2411.11926cs.CV2024-11被引 9

融合非线性建模与长程依赖,提升医学图像分割精度

KAN-Mamba FusionNet: Redefining Medical Image Segmentation with Non-Linear Modeling

  • 设计KAMBA模块,结合KAN的非线性建模与Mamba的长程依赖捕捉能力
  • 在BUSI、Kvasir-Seg、GlaS三个数据集上,IoU和F1得分均超越现有方法
  • 适用于需要高精度分割的临床场景,如手术机器人辅助与疾病诊断

医学图像分割对机器人手术、疾病诊断和治疗规划至关重要。尽管近期深度学习模型有所进展,但现有方法在处理复杂医学图像时仍受限于对非线性特征的捕捉或长程依赖的建模。本文提出KAN-Mamba FusionNet,通过新设计的KAMBA模块,有效结合了柯尔莫戈洛夫-阿诺德网络(KAN)对非线性特性的强表达能力与Mamba架构对长程依赖的高效建模。我们在BUSI、Kvasir-Seg和GlaS三个医学图像分割数据集上评估该模型,结果表明其在交并比(IoU)和F1分数上持续优于当前最优方法。消融实验进一步验证了各组件对性能的贡献,证明该方法在处理复杂视觉数据方面具有显著优势,为医疗影像分析提供了可靠的新范式。

原文摘要 · Abstract (English)

Medical image segmentation is essential for applications like robotic surgeries, disease diagnosis, and treatment planning. Recently, various deep-learning models have been proposed to enhance medical image segmentation. One promising approach utilizes Kolmogorov-Arnold Networks (KANs), which better capture non-linearity in input data. However, they are unable to effectively capture long-range dependencies, which are required to accurately segment complex medical images and, by that, improve diagnostic accuracy in clinical settings. Neural networks such as Mamba can handle long-range dependencies. However, they have a limited ability to accurately capture non-linearities in the images as compared to KANs. Thus, we propose a novel architecture, the KAN-Mamba FusionNet, which improves segmentation accuracy by effectively capturing the non-linearities from input and handling long-range dependencies with the newly proposed KAMBA block. We evaluated the proposed KAN-Mamba FusionNet on three distinct medical image segmentation datasets: BUSI, Kvasir-Seg, and GlaS - and found it consistently outperforms state-of-the-art methods in IoU and F1 scores. Further, we examined the effects of various components and assessed their contributions to the overall model performance via ablation studies. The findings highlight the effectiveness of this methodology for reliable medical image segmentation, providing a unique approach to address intricate visual data issues in healthcare.

医学图像分割非线性建模长程依赖

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