Search papers, labs, and topics across Lattice.
This paper introduces KANLib, a modular and efficient framework for developing Kolmogorov-Arnold Networks (KANs), addressing the challenges of high computational costs and inconsistent feature support in existing implementations. By unifying concepts from previous frameworks and offering extensive customization options, KANLib allows researchers to explore various architectural variations while maintaining compatibility with PyTorch. Experimental results on the California Housing benchmark confirm that KANLib not only reproduces the predictive performance of established KANs but also enhances computational efficiency, paving the way for future advancements in KAN research.
KANLib achieves competitive efficiency and flexibility in KAN architectures, enabling researchers to explore innovative designs with minimal performance trade-offs.
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.
