arXiv:2606.17927cs.LGcs.AI2026-06

KANLib让可解释的神经网络更易用,支持高效开发与实验。

KANLib -- A Modular, Extensible and Fast Kolmogorov-Arnold Network Implementation

  • 整合多种KAN实现,提供灵活可扩展的统一框架
  • 在房价预测任务中表现媲美现有模型,计算效率高
  • 适合想快速试验KAN变体的研究者

Kolmogorov-Arnold网络(KANs)作为传统多层感知机的潜在替代方案,用可学习的单变量函数取代线性权重,理论上具备更强的可解释性和表达能力。然而,由于计算成本高、现有框架功能不一致,实际研究仍面临困难。本文提出KANLib,一个模块化、可扩展且高效的KAN实现框架,整合了PyKAN、EfficientKAN和FastKAN的核心思想,在保持与标准PyTorch工作流兼容的同时,支持两种基函数类型、自适应网格重缩放、网格扩展及细粒度架构定制。在California Housing基准测试上,KANLib重现了已有参考实现的预测性能,同时具备竞争力的计算效率。此外,该框架能以较小性能损失探索超越标准KAN结构的变体。整体而言,KANLib为可扩展、可扩展的KAN研究提供了坚实基础。

原文摘要 · Abstract (English)

Kolmogorov-Arnold Networks (KANs) have recently emerged as a promising alternative to traditional multilayer perceptrons by replacing linear weights with learnable univariate functions. Despite their theoretical advantages in interpretability and expressiveness, practical research of KANs remains difficult due to high computational costs and inconsistent feature support across existing frameworks. This paper introduces KANLib, a modular, extensible, and computationally efficient framework for developing and evaluating KAN architectures. KANLib unifies core concepts from existing implementations, including PyKAN, EfficientKAN, and FastKAN, within a consistent software architecture that emphasizes flexibility, feature parity, and high performance. The framework supports two basis function types, adaptive grid rescaling, grid extension, and fine-grained architectural customization while maintaining compatibility with standard PyTorch workflows. Experimental evaluation on the California Housing benchmark demonstrates that KANLib reproduces the predictive behavior of established reference KAN implementations while achieving competitive computational efficiency. Furthermore, the framework enables the exploration of architectural variations beyond standard KAN formulations with only minor impacts on predictive performance. Overall, KANLib provides a robust foundation for future research on scalable and extensible KAN architectures.

KAN神经网络可解释性PyTorch

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