KANLib:一个模块化、可扩展且快速的Kolmogorov-Arnold网络实现

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

精选理由

想试KAN但被碎片化实现劝退?KANLib把PyKAN、EfficientKAN、FastKAN统一成一个高效框架,直接上手跑基准。

AI 摘要

KANLib是一个模块化、可扩展且计算高效的Kolmogorov-Arnold网络(KAN)框架,统一了PyKAN、EfficientKAN和FastKAN等现有实现的核心概念。它支持两种基函数类型、自适应网格缩放、网格扩展及细粒度架构定制,并保持与PyTorch工作流的兼容性。在California Housing基准上,KANLib再现了参考KAN实现的预测行为,同时实现了有竞争力的计算效率。该框架允许探索超出标准KAN公式的架构变体,对预测性能影响微小。

AI 翻译 · 中文

KANLib是一个模块化、可扩展且计算高效的Kolmogorov-Arnold网络(KAN)框架,统一了PyKAN、EfficientKAN和FastKAN等现有实现的核心概念。它支持两种基函数类型、自适应网格缩放、网格扩展及细粒度架构定制,并保持与PyTorch工作流的兼容性。在California Housing基准上,KANLib再现了参考KAN实现的预测行为,同时实现了有竞争力的计算效率。该框架允许探索超出标准KAN公式的架构变体,对预测性能影响微小。

arXiv cs.AIKolmogorov-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 advantag