动态堆叠KAN网络,高效精准分割肝脏肿瘤。
ProKAN: Progressive Stacking of Kolmogorov-Arnold Networks for Efficient Liver Segmentation
- 按过拟合情况逐步添加KAN模块,自动调节模型复杂度。
- 在3D肝脏分割任务中提升精度,Dice得分显著优于MLP和固定KAN。
- 可解释性强,通过可学习系数揭示决策过程,适合医疗影像分析。
随着对肝脏肿瘤精确高效3D识别需求的增长,深度学习模型的研究日益活跃。尽管现有架构性能优异,但仍面临过拟合和计算成本过高的问题。本文提出proKAN,一种针对柯尔莫哥洛夫-阿诺德网络(KAN)的渐进式堆叠方法,以应对上述挑战。与传统架构不同,proKAN在训练过程中根据过拟合行为动态增加KAN模块,当检测到过拟合时停止扩展,从而避免不必要的计算开销,同时保持高精度。该方法利用基于B样条的可学习激活函数,增强对3D医学数据中复杂关系的建模能力。实验表明,proKAN在肝脏分割任务中达到领先性能,显著优于标准多层感知机(MLP)和固定结构的KAN架构,在准确性、Dice分数和时间效率上均有明显提升。此外,其可解释性通过可学习系数提供决策过程洞察,为3D医学图像分割提供了高效且可靠的解决方案。
原文摘要 · Abstract (English)
The growing need for accurate and efficient 3D identification of tumors, particularly in liver segmentation, has spurred considerable research into deep learning models. While many existing architectures offer strong performance, they often face challenges such as overfitting and excessive computational costs. An adjustable and flexible architecture that strikes a balance between time efficiency and model complexity remains an unmet requirement. In this paper, we introduce proKAN, a progressive stacking methodology for Kolmogorov-Arnold Networks (KANs) designed to address these challenges. Unlike traditional architectures, proKAN dynamically adjusts its complexity by progressively adding KAN blocks during training, based on overfitting behavior. This approach allows the network to stop growing when overfitting is detected, preventing unnecessary computational overhead while maintaining high accuracy. Additionally, proKAN utilizes KAN's learnable activation functions modeled through B-splines, which provide enhanced flexibility in learning complex relationships in 3D medical data. Our proposed architecture achieves state-of-the-art performance in liver segmentation tasks, outperforming standard Multi-Layer Perceptrons (MLPs) and fixed KAN architectures. The dynamic nature of proKAN ensures efficient training times and high accuracy without the risk of overfitting. Furthermore, proKAN provides better interpretability by allowing insight into the decision-making process through its learnable coefficients. The experimental results demonstrate a significant improvement in accuracy, Dice score, and time efficiency, making proKAN a compelling solution for 3D medical image segmentation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。