Leveraging KANs For Enhanced Deep Koopman Operator Discovery

George Nehma,Madhur Tiwari
2024-08-12
Abstract:Multi-layer perceptrons (MLP's) have been extensively utilized in discovering Deep Koopman operators for linearizing nonlinear dynamics. With the emergence of Kolmogorov-Arnold Networks (KANs) as a more efficient and accurate alternative to the MLP Neural Network, we propose a comparison of the performance of each network type in the context of learning Koopman operators with control. In this work, we propose a KANs-based deep Koopman framework with applications to an orbital Two-Body Problem (2BP) and the pendulum for data-driven discovery of linear system dynamics. KANs were found to be superior in nearly all aspects of training; learning 31 times faster, being 15 times more parameter efficiency, and predicting 1.25 times more accurately as compared to the MLP Deep Neural Networks (DNNs) in the case of the 2BP. Thus, KANs shows potential for being an efficient tool in the development of Deep Koopman Theory.
Machine Learning,Dynamical Systems,Applied Physics,Computational Physics
What problem does this paper attempt to address?
The paper aims to address the following issues: 1. **Improving the efficiency and accuracy of linearizing nonlinear systems**: By using Kolmogorov-Arnold Networks (KANs) instead of traditional multilayer perceptron (MLP) neural networks to discover Deep Koopman operators, the efficiency and accuracy of linearizing nonlinear systems can be improved. 2. **Reducing training time and data requirements**: Compared to MLPs, KANs can learn Koopman operators faster and require less training data, which is significant in practical applications, especially in data-scarce situations. 3. **Validating the effectiveness of KANs in complex systems**: By validating the effectiveness and superiority of the KANs framework in different scenarios such as the pendulum system, cart-pole system, and two-body problem, its potential applications in control, state estimation, and other areas are demonstrated. 4. **Developing efficient controllers**: Utilizing the Koopman operators learned by KANs to develop linear systems and designing effective linear quadratic regulator (LQR) controllers based on this to achieve precise control of nonlinear systems. In summary, the main goal of this paper is to improve the efficiency and accuracy of linearizing nonlinear systems by introducing the KANs framework to enhance the existing Deep Koopman theory and demonstrate its potential in practical applications.