KAN 2.0: Kolmogorov-Arnold Networks Meet Science

Ziming Liu,Pingchuan Ma,Yixuan Wang,Wojciech Matusik,Max Tegmark
2024-08-20
Abstract:A major challenge of AI + Science lies in their inherent incompatibility: today's AI is primarily based on connectionism, while science depends on symbolism. To bridge the two worlds, we propose a framework to seamlessly synergize Kolmogorov-Arnold Networks (KANs) and science. The framework highlights KANs' usage for three aspects of scientific discovery: identifying relevant features, revealing modular structures, and discovering symbolic formulas. The synergy is bidirectional: science to KAN (incorporating scientific knowledge into KANs), and KAN to science (extracting scientific insights from KANs). We highlight major new functionalities in the pykan package: (1) MultKAN: KANs with multiplication nodes. (2) kanpiler: a KAN compiler that compiles symbolic formulas into KANs. (3) tree converter: convert KANs (or any neural networks) to tree graphs. Based on these tools, we demonstrate KANs' capability to discover various types of physical laws, including conserved quantities, Lagrangians, symmetries, and constitutive laws.
Machine Learning,Artificial Intelligence,Computational Physics,Data Analysis, Statistics and Probability
What problem does this paper attempt to address?
The main goal of this paper is to address a significant challenge between artificial intelligence (AI) and scientific research—the inherent incompatibility between connectionism (primarily based on neural network methods) and symbolism (relying on explicit mathematical formulas and logical rules). Specifically, the authors propose a framework aimed at seamlessly integrating Kolmogorov-Arnold Networks (KANs) with scientific research to support curiosity-driven scientific exploration. Several key points mentioned in the paper include: 1. **Integration of KANs with Scientific Research**: By combining KANs with scientific research, the embedding of scientific knowledge into KANs (science to KANs) and the extraction of scientific insights from KANs (KANs to science) are achieved. 2. **Enhancement of KANs**: - The introduction of Multiplicative KANs (MultKANs), which add multiplicative nodes to the original KANs, helps to better reveal multiplicative structures in the data and improve interpretability and capacity. - The proposal of a KAN compiler (kanpiler) that can convert symbolic formulas into KANs, thereby utilizing known scientific knowledge to initialize KANs and further fine-tune these KANs with data to discover new physical laws. 3. **Embedding Scientific Knowledge**: - Adding important features to KANs, such as introducing auxiliary variables, can increase the network's expressive power and improve interpretability. - Building modular structures into KANs, such as separability and symmetry, can help simplify the understanding of the network. - Compiling symbolic formulas into KANs by first encoding scientific knowledge into KANs and then fine-tuning with data to combine the advantages of both. 4. **Extracting Scientific Knowledge from KANs**: - Identifying important features by using improved attribution score methods to assess the importance of input variables. - Identifying modular structures, including anatomical modules and functional modules, to reveal the internal organization of the network and the overall function representation. In summary, this paper aims to support curiosity-driven scientific research by enhancing the functionality and interpretability of KANs and developing a series of tools and techniques to closely integrate KANs with scientific research. This approach not only helps scientists better understand complex systems but also overcomes, to some extent, the limitations of traditional machine learning methods as black-box models.