用新型混合网络提升医学图像分割精度
A hybrid Kolmogorov-Arnold network for medical image segmentation
- 结合柯尔莫哥洛夫-阿诺德网络与U形结构,提升特征表达能力
- 在多个医学影像数据集上优于现有方法,尤其擅长复杂结构分割
- 适合需要高精度分割的医学影像分析研究者使用
医学图像分割在诊断与治疗规划中至关重要,但因图像内在复杂性和变异性,仍具挑战性,尤其在捕捉数据中的非线性关系方面。我们提出U-KABS,一种融合柯尔莫哥洛夫-阿诺德网络(KANs)与U形编码器-解码器架构的新型混合框架,以增强分割性能。该模型结合卷积与挤压-激励模块,强化通道特征表示;引入基于伯恩斯坦多项式和B样条的可学习激活函数(KABS),利用伯恩斯坦多项式的全局平滑性与B样条的局部自适应性,有效捕捉整体上下文趋势与细微结构特征。编码器与解码器间的跳跃连接实现多尺度特征融合并保留空间细节。在多个医学影像基准数据集上的评估显示,U-KABS显著优于强基线模型,尤其在复杂解剖结构分割任务中表现突出。
原文摘要 · Abstract (English)
Medical image segmentation plays a vital role in diagnosis and treatment planning, but remains challenging due to the inherent complexity and variability of medical images, especially in capturing non-linear relationships within the data. We propose U-KABS, a novel hybrid framework that integrates the expressive power of Kolmogorov-Arnold Networks (KANs) with a U-shaped encoder-decoder architecture to enhance segmentation performance. The U-KABS model combines the convolutional and squeeze-and-excitation stage, which enhances channel-wise feature representations, and the KAN Bernstein Spline (KABS) stage, which employs learnable activation functions based on Bernstein polynomials and B-splines. This hybrid design leverages the global smoothness of Bernstein polynomials and the local adaptability of B-splines, enabling the model to effectively capture both broad contextual trends and fine-grained patterns critical for delineating complex structures in medical images. Skip connections between encoder and decoder layers support effective multi-scale feature fusion and preserve spatial details. Evaluated across diverse medical imaging benchmark datasets, U-KABS demonstrates superior performance compared to strong baselines, particularly in segmenting complex anatomical structures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。