arXiv:2608.12050cs.CV2026-08

让神经网络的每条边函数提前学好功能,提升医学图像分割精度。

Predicting Functions, Not Features: KANs with Function-Space Joint-Embedding Predictive Learning for Medical Image Segmentation

论文配图:Predicting Functions, Not Features: KANs with Function-Space Joint-Embedding Predictive Learning for Medical Image Segmentation
图 1 · 摘自论文原文
  • 在函数空间中预训练每条边的特征表示,而非仅靠聚合后目标优化
  • 在5个基准上平均Dice提升2.25个百分点,优于现有最强KAN方法
  • 适合追求高精度医学图像分割的研究者与临床应用开发者

Kolmogorov--Arnold网络(KANs)通过将每条网络边参数化为可学习的一维函数,实现了显式的函数表达。然而,现有基于KAN的分割模型仅在边聚合后通过目标函数优化边函数,导致各函数缺乏预聚合阶段的显式学习目标。为此,我们提出函数空间联合嵌入预测学习(FS-JEPA)框架用于医学图像分割。该框架将预测学习引入KAN的预聚合函数空间:一个带掩码的在线分支预测由全上下文指数移动平均目标分支生成的采样边函数的结构化签名,共享的边索引保持预测与目标之间的对应关系。我们不直接预测孤立的边响应,而是使用围绕输入锚点的多尺度函数值组合成结构化签名,以捕捉单点响应无法表征的局部函数变化,提供更丰富的预测目标。函数空间目标与分割损失联合优化,推理时移除预测分支。在五个医学图像分割基准上的实验表明,我们的方法取得了最佳平均Dice分数,较最强的对比方法高出2.25个百分点。

原文摘要 · Abstract (English)

Kolmogorov--Arnold Networks (KANs) introduce explicit functional representations by parameterizing each network edge as a learnable univariate function. However, existing KAN-based segmentation models optimize edge functions only through objectives defined after edge aggregation, leaving individual functions without an explicit pre-aggregation learning target. To address this limitation, we propose Function-Space Joint-Embedding Predictive Learning (FS-JEPA) for medical image segmentation. Our FS-JEPA framework moves predictive learning into the pre-aggregation function space of KANs. A masked online branch predicts structured signatures of sampled KAN edge functions generated by a full-context exponential moving average target branch, while shared edge indices preserve correspondence between predictions and targets. Rather than predicting an isolated edge response, we represent each sampled edge function using a multi-radius signature composed of function evaluations around its input anchor. This structured representation captures local functional variations that cannot be characterized by a single response and provides a more informative predictive target. The function-space objective is jointly optimized with the segmentation loss during training, while the predictive branch is removed at inference. Experiments on five medical image segmentation benchmarks show that our FS-JEPA achieves the best average Dice and outperforms the strongest competing KAN-based method by +2.25 percentage points.

医学图像KAN网络函数空间分割精度

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