Contributed Session 2:  PDEs and Applications
Understanding the dynamics of reaction diffusion equation using Transformer-based Koopman Autoencoder
Nitu Kumari
Indian Institute of Technology Mandi
India
  Co-Author(s):    
  Abstract:
 

A Transformer based Koopman autoencoder is proposed for linearizing reaction diffusion equation. The primary focus of this study is on using deep learning techniques to find complex spatiotemporal patterns in the reaction diffusion system. The emphasis is not just solving the equation but also transforming the system dynamics into a more comprehensible, linear form. Global coordinate transformations are achieved through the autoencoder, which learns to capture the underlying dynamics by training on a dataset with 60,000 initial conditions. Extensive testing on multiple datasets was used to assess the efficacy of the proposed model, demonstrating its ability to accurately predict the system evolution as well as to generalize. We provide a thorough comparison study, comparing our suggested design to a few other comparable methods using experiments on various PDEs. Results show improved accuracy, highlighting the capabilities of the Transformer based Koopman autoencoder. The proposed architecture is significantly ahead of other architectures, in terms of solving different types of PDEs using a single architecture. Our method relies entirely on the data, without requiring any knowledge of the underlying equations. This makes it applicable to even the datasets where the governing equations are not known.