中文

GroupKAN:基于分组样条建模的高效Kolmogorov-Arnold网络

计算机视觉与模式识别 2026-04-15 v2

摘要

医学图像分割需要实现高精度的同时保持计算效率和临床可解释性。虽然最近的Kolmogorov-Arnold网络(KAN)提供了强大的自适应非线性函数,但其全通道样条变换相对于通道维度C会产生二次参数增长,复杂度为 O(C2(G+k))\mathcal{O}(C^{2}(G+k)),其中G和k分别表示网格区间数和样条多项式阶数。此外,受限于无约束样条映射缺乏结构约束,导致功能自由度过大,在有限的医学标注数据下可能引发过拟合。为此,我们提出GroupKAN(分组Kolmogorov-Arnold网络),一种基于分组结构样条建模的高效架构。具体包括:(1)分组KAN变换(GKT),通过跨g个组的组内通道映射限制样条相互作用,将由样条引起的二次膨胀有效降低为 \textbf{O(C2(G+kg+1))\mathcal{O}(C^2(\frac{G+k}{g} + 1))}$,从而显著降低有效二次系数;以及(2)分组KAN激活(GKA),在每个组内应用共享样条函数,以实现高效的token级非线性变换。通过对通道相互作用施加结构约束,GroupKAN在不牺牲表达能力的前提下显著减少参数冗余。对三个医学基准(BUSI、GlaS和CVC)的广泛评估表明,GroupKAN实现了79.80%的平均IoU,超越强U-KAN基线+1.11%,而仅需47.6%的参数(3.02M vs. 6.35M)。定性结果进一步显示,GroupKAN生成的激活图更精确地局部化,与真实标签的对齐性优于MLP和KAN,显著提升了临床可解释性。

关键词

引用

@article{arxiv.2511.05477,
  title  = {GroupKAN: Efficient Kolmogorov-Arnold Networks via Grouped Spline Modeling},
  author = {Guojie Li and Tianyi Liu and Anwar P. P. Abdul Majeed and Muhammad Ateeq and Anh Nguyen and Fan Zhang},
  journal= {arXiv preprint arXiv:2511.05477},
  year   = {2026}
}