arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.07702cs.CV
收藏

一种混合的柯莫戈罗夫-阿诺德网络用于医学图像分割

A hybrid Kolmogorov-Arnold network for medical image segmentation

Concordia Institute for Information Systems Engineering(康科德信息系统工程研究所);Concordia University(康科德大学)

Deep Bhattacharyya, Ali Ayub, A. Ben Hamza

AI总结:

本文提出U-KABS,一种结合KAN和U形架构的混合网络,用于提升医学图像分割的性能,尤其在复杂结构分割中表现优异。

摘要:

医学图像分割在诊断和治疗计划中起着至关重要的作用,但由于医学图像固有的复杂性和变异性,仍面临挑战,尤其是在捕捉数据中的非线性关系方面。我们提出了U-KABS,一种新颖的混合框架,它将Kolmogorov-Arnold Networks(KANs)的表达能力与U形编码器-解码器架构相结合,以提高分割性能。U-KABS模型结合了卷积和挤压与激活阶段,增强了通道特征表示,并结合了KAN伯恩斯坦样条(KABS)阶段,该阶段采用了基于伯恩斯坦多项式和B-样条的可学习激活函数。这种混合设计利用了伯恩斯坦多项式的全局平滑性和B-样条的局部适应性,使模型能够有效捕捉广义的上下文趋势和细粒度的模式,这对区分复杂的医学图像结构至关重要。编码器和解码器层之间的跳接连接支持有效的多尺度特征融合并保留空间细节。在多种医学影像基准数据集上评估,U-KABS在分割复杂解剖结构方面表现出优于强基线的性能。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑