arXiv:2605.06740cs.LGcs.AI2026-05

让神经网络学会根据数据几何形状自适应调整坐标,提升复杂科学问题建模能力。

Geometric Kolmogorov--Arnold Network (GeoKAN)

论文配图:Geometric Kolmogorov--Arnold Network (GeoKAN)
图 1 · 摘自论文原文
  • 在可学习的几何坐标系中进行函数逼近,而非固定欧氏空间
  • 通过动态拉伸高变化区域、压缩平滑区域,按需分配表示能力
  • 适用于物理信息学习和微分方程求解等强非均匀问题场景

我们提出几何柯尔莫哥洛夫-阿诺德网络(GeoKAN),一类几何感知的KAN型模型,其近似在学习到的、与几何适配的坐标系中进行,而非固定的欧氏输入坐标。GeoKAN通过学习对角黎曼度量,在基函数展开与特征混合前对输入进行变形。该学习度量提供了局部长度缩放与体积扭曲的几何归纳偏置,在物理信息设置下还影响模型所见的微分结构。在此框架下,我们发展了三种主要变体:GeoKAN-NNMetric、GeoKAN-γ 和 LM-KAN。对于 LM-KAN,进一步考虑三种基函数特定版本:LM-KAN-RBF、LM-KAN-Wav 与 LM-KAN-Fourier。这些变体使我们能够将几何感知的 KAN 模型用于通用函数逼近及物理信息学习中的代理模型。通过拉伸快速变化区域、压缩平滑区域,GeoKAN 以任务相关方式重分配表示分辨率,使模型能在最需要处集中容量。因此,GeoKAN 非常适合科学机器学习与微分方程问题中出现的尖锐、刚性、局部化与强非均匀情形。

原文摘要 · Abstract (English)

We introduce Geometric Kolmogorov--Arnold Networks (GeoKANs), a family of geometry-aware KAN-type models in which approximation is carried out in learned, geometry-adapted coordinates rather than in fixed Euclidean input coordinates. GeoKAN achieves this by learning a diagonal Riemannian metric that warps the input before basis expansion and feature mixing. The learned metric provides a geometric inductive bias through local length scaling and volume distortion, and in physics-informed settings it also affects the differential structure seen by the model. Within this framework, we develop three main variants, namely GeoKAN-NNMetric, GeoKAN-$γ$, and LM-KAN. For LM-KAN, we further consider three basis-specific versions, LM-KAN-RBF, LM-KAN-Wav, and LM-KAN-Fourier. These variants allow us to study geometry-aware KAN models both as general function approximators and as surrogates in physics-informed learning. By stretching regions with rapid variation and compressing smoother regions, GeoKAN reallocates representational resolution in a task-dependent manner, allowing the model to place capacity where it is most needed. As a result, GeoKAN is well suited to sharp, stiff, localized, and strongly non-uniform regimes arising in scientific machine learning and differential-equation problems.

几何深度学习科学机器学习KAN物理信息网络

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