跳转至

Empirical Stability Analysis of Kolmogorov-Arnold Networks in Hard-Constrained Recurrent Physics-Informed Discovery

会议: ICLR2026
arXiv: 2602.09988
代码: 未开源
领域: 科学计算/物理信息神经网络
关键词: KAN, physics-informed, oscillator, HRPINN, neural ODE, 残差发现

一句话总结

在硬约束递归物理信息架构(HRPINN)中系统评估vanilla KAN替代MLP作为残差分支的效果——通过3项互补研究×100随机种子发现KAN在单变量可分残差(Duffing的 \(-0.3x^3\))上的表现具有竞争力,但在乘法耦合残差(Van der Pol的 \((1-x^2)v\))上系统性失败且超参数极度脆弱,标准MLP在几乎所有配置下稳定性远优。

研究背景与动机

领域现状:硬约束递归物理信息架构(HRPINN)将已知物理嵌入递归积分器,神经网络只学习残差动力学——这确保了物理一致性并已在网络物理系统中验证有效。同时,Kolmogorov-Arnold Networks(KAN)基于Kolmogorov-Arnold表示定理,将多元函数分解为单变量函数之和 \(\Phi(\mathbf{x}) = \sum_q \phi_q(\sum_p \psi_{q,p}(x_p))\),用可学习的B-样条替代MLP的固定激活函数,在科学ML中展现潜力。

现有痛点: - KAN在Neural ODE和灰箱设置中展现了符号发现潜力(KAN-ODEs、SKANODEs),但这些工作使用的是无约束的连续ODE设置 - 没有人在硬约束递归架构中测试过KAN——递归设置中误差会累积,对稳定性要求更高 - KAN的加法归纳偏置(\(\phi(x) + \phi(v)\))理论上适合可分物理定律,但在实际中是否成立?

核心矛盾:KAN的加法结构天然适合加法可分函数,但许多物理定律包含乘法耦合项(如Van der Pol的 \((1-x^2)v\))。理论上KAN可通过深层组合表示乘法(\(xy = \frac{1}{4}((x+y)^2 - (x-y)^2)\)),但这需要更深的层→深层KAN在递归误差累积下是否仍然稳定?

本文目标:为vanilla KAN在硬约束递归物理架构中建立基线评估。

切入角度:选择两个具有对比残差结构的经典振荡器——Duffing(单变量多项式)和Van der Pol(乘法耦合)——作为加法可分性的边界测试。

核心 idea:通过精心对照的实验设计,揭示KAN的加法归纳偏置在递归物理约束下的实际成功边界。

方法详解

整体框架

整篇工作不是提出新模型,而是把 vanilla KAN 当作 HRPINN 残差分支的"待测器件",用受控实验去逼出它的成功边界。HRPINN 把已知物理和数值积分器固定在递归更新规则里,网络只需学习残差流形 \(R_\theta(x, v)\)——它接收归一化状态 \([x, v]\),分别用标准 ReLU MLP 和 B-spline KAN 两套实现做平行对照。评估同时看 Performance(测试 MSE)和 Discovery(在 \(100 \times 100\)\(x,v \in [-2.5, 2.5]\) 的相空间网格上算的 \(R^2\)),并统一用候选拟合而非 KAN 特有的符号剪枝来提取表达式,保证两边在同一杆秤上称。在此之上,作者用三项互补研究——配置消融、规模×训练范式消融、定性流形可视化——从统计、扩展性、机理三个角度交叉验证同一个论断。

关键设计

1. 配置消融:用穷举网格 + 大样本种子,把"碰巧好"和"鲁棒地好"分开

KAN 的麻烦在于它的网格大小 \(G\) 和样条阶数 \(k\) 直接决定表达能力,单看一个配置的好坏说明不了架构本身。作者固定训练设置、铺开 7 种 \((G, k)\) 组合,每种各跑 100 个随机种子做 bootstrap 统计。结果很说明问题:粗网格的 Config F(\(G=3, k=3\))在 Duffing 上能到 \(R^2 = 0.862\),逼近 MLP 的 \(0.957\);但同一批配置换到 Van der Pol 就大面积塌成负 \(R^2\)(发散解),Config C 甚至是 \(-5.229 \pm 5.091\)。作为对照的 337 参数 MLP 稳稳给出 Duffing \(0.957\)、VdP \(0.768\) 且方差极小。也就是说 KAN 的好成绩属于"某个配置在某个系统上碰巧好",而不是架构层面的稳健,超参数稍一偏移就崩。

2. 规模×训练范式消融:定位加法偏置在乘法耦合上的扩展瓶颈

配置之外还要回答"加大参数量或换训练方式能不能救"。作者固定配置、把参数量从 Very Small 120 拉到 Deep 880,并在单步 Teacher Forcing 和全程 BPTT 两种范式下各测一遍:

训练范式 KAN行为 MLP行为
Teacher Forcing 小KAN在Duffing竞争力强、VdP随规模迅速退化 平稳扩展
BPTT 最小KAN达到VdP最佳 \(R^2 \approx 0.74\)(长时域监督有帮助)、深KAN不稳定 所有规模稳定优越

这张表把瓶颈钉死在归纳偏置上:MLP 的稠密矩阵乘法第一层就靠 \(w_i x + w_j v\) 实现了变量交互,而 KAN 的加法结构 \(\phi(x) + \phi(v)\) 想表达乘法只能靠深层组合去逼近 \(xy = \frac{1}{4}((x+y)^2 - (x-y)^2)\)。问题是递归设置里误差会逐步累积,越深的 KAN 越放大这种累积,于是 Deep 配置直接灾难性发散。BPTT 的长时域监督能让最小 KAN 把 VdP 的 \(R^2\) 从 0.464 拉到 0.743,算是缓解,但 MLP 在所有规模、所有范式下仍然全面占优。

3. 定性流形可视化:让残差曲面替统计结论作证

纯数字容易让人怀疑是不是评估口径问题,所以作者把 KAN 和 MLP 学到的残差曲面直接画出来和真值比。Duffing 上 KAN 准确复现了立方流形,候选拟合得到 \(-0.234x^3\)(真值 \(-0.3x^3\)\(R^2 = 0.91\),38% 种子成功);Van der Pol 上 KAN 曲面却坍缩成近似线性,完全没抓住 \((1-x^2)v\) 那个抛物线调制结构。可视化和前两项的定量统计严丝合缝地对上:加法偏置在单变量残差上是优势,一碰到变量乘法耦合就成了硬瓶颈。值得强调的是,这个失败发生在已用候选拟合喂入正确函数形式的前提下——说明卡住 KAN 的是递归误差累积下的优化稳定性,而非表达能力或符号提取环节本身。

实验关键数据

配置消融主表(95% Bootstrap CI, N=100 seeds)

配置 Duffing \(R^2\) Van der Pol \(R^2\)
KAN Config A (\(G=5, k=3\)) 0.835 ± 0.030 0.667 ± 0.037
KAN Config C (Sparse-Low) 0.595 ± 0.033 -5.229 ± 5.091
KAN Config E (Aggressive-Grid) 0.794 ± 0.067 0.699 ± 0.065
KAN Config F (Coarse-Grid) 0.862 ± 0.037 0.639 ± 0.302
KAN Config G (Fine-Grid) 0.745 ± 0.099 -0.174 ± 0.691
MLP (337 params) 0.957 ± 0.009 0.768 ± 0.015

参数规模消融(Mean ± 95% CI, N=100 seeds)

架构 参数 Duffing(TF) VdP(TF) Duffing(BPTT) VdP(BPTT)
KAN Very Small 120 0.836±0.032 0.464±0.166 0.914±0.061 0.743±0.061
KAN Small 240 0.777±0.079 0.322±0.292 0.874±0.080 0.785±0.073
KAN Wide 480 0.845±0.025 0.232±0.570 0.468±0.773 -0.602±2.842
KAN Deep 880 -3.146±7.106 -0.303±1.579 (不稳定) 0.754±0.079
MLP Tiny 105 0.914±0.026 0.593±0.048 0.906±0.092 0.622±0.173
MLP Small 337 0.957±0.009 0.768±0.015 0.937±0.047 0.879±0.032
MLP Medium 1185 0.960±0.013 0.805±0.014 0.951±0.033 0.879±0.019
MLP Large 4417 0.965±0.009 0.843±0.010 0.932±0.063 0.898±0.017

关键发现

  • KAN在Duffing上可发现立方结构(\(-0.234x^3\),真值 \(-0.3x^3\)\(R^2=0.91\)),38%种子成功→有潜力但不可靠
  • KAN在Van der Pol上系统性失败→加法偏置无法稳定学习乘法耦合
  • BPTT的长时域监督帮助最小KAN缓解VdP问题(\(R^2\) 从0.464升至0.743),但MLP仍全面占优
  • KAN超参数敏感度远高于MLP——VdP上从0.699到-5.229→实践中不实用
  • 深层KAN(880参数)在递归设置中灾难性不稳定(\(R^2 = -3.146\)

亮点与洞察

  • 诚实的"负面结果"——清楚展示了当前vanilla KAN在物理约束递归架构中的实际边界,为KAN社区提供了重要警示
  • 加法偏置vs乘法耦合的精准诊断:选择Duffing和Van der Pol作为恰好跨越加法可分性边界的测试对→诊断直击KAN设计核心假设
  • 大规模种子统计的可信度:每项实验100个随机种子+95%置信区间→结论不依赖于幸运的初始化
  • 递归误差累积的独特洞察:KAN在无约束ODE中可能表现尚可,但在硬约束递归设置中误差快速放大→揭示了设置依赖性

局限与展望

  • 仅测试vanilla KAN——改进变体(SKANODEs、Hybrid KAN-MLP、DeepOKAN)可能克服乘法限制
  • 仅两个振荡器系统——更复杂/混沌系统(Lorenz吸引子)待测试
  • 未与SINDy等成熟符号发现方法对比
  • 未探究KAN独有的符号剪枝能力——直接通过样条结构提取符号表达式
  • 未分析梯度条件数/优化景观——仅展示了"什么失败"但未完全解释"为什么失败"

相关工作与启发

  • vs KAN-ODEs (Koenig et al., 2024):在无约束连续ODE中表现好→本文揭示硬约束递归设置下的脆弱性→设置依赖性是关键
  • vs SKANODEs (Liu et al., 2025):结构化KAN可能通过算子链接(分别表示 \(1-x^2\) 再与 \(v\) 交互)缓解乘法问题→启发混合方案
  • 启发:可否设计"乘法感知KAN"→在KAN基础层引入显式乘法门→保留加法偏置的可解释性同时处理耦合项?

评分

⭐⭐⭐⭐ (4/5)

综合评价:系统性的负面结果论文,100种子×3项研究的极其充分实证支撑了关于KAN加法偏置边界的精确论断——虽非新方法,但为KAN在物理信息应用中的实践提供了不可或缺的校准参考。