Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

KAN implementation in the TEDEouS #41

Merged
merged 33 commits into from
Aug 20, 2024
Merged

KAN implementation in the TEDEouS #41

merged 33 commits into from
Aug 20, 2024

Conversation

dimerf99
Copy link
Collaborator

@dimerf99 dimerf99 commented Aug 6, 2024

1. Change models.py file:
KAN implementation was added to the models.py file. The implementation consists of two classes:

  1. class KANLinear - realization of a linear layer using Kolmogorov-Arnold splines. It allows to model nonlinear dependencies between input and output data using splines.
  2. class KAN - realization of multilayer network using Kolmogorov-Arnold splines.

2. Adding examples of equation solutions:

  1. Adding an example of solving the wave equation using KAN.
  2. Adding an example of solving the Bürgers equation using KAN.

KAN implementation was added to the models.py file. The implementation consists of two classes:
1. class KANLinear - realization of a linear layer using Kolmogorov-Arnold splines. It allows to model nonlinear dependencies between input and output data using splines.
2. class KAN - realization of multilayer network using Kolmogorov-Arnold splines.
Adding an example of solving the wave equation using KAN.
Adding an example of solving the Bürgers equation using KAN.
Update model.py
@dimerf99 dimerf99 requested a review from SuperSashka August 6, 2024 15:40
@dimerf99 dimerf99 self-assigned this Aug 6, 2024
@SuperSashka
Copy link
Member

atm we need to find a way out how to properly import efficient-kan to not to take all required code manually and be able to change KAN backbone if it will be the case

Copy link
Member

@SuperSashka SuperSashka left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It seems that we need a fork of efficient-kan with wheels ot something like this. I think, we could make PR to a main efficient-kan repo in the future.

dimerf99 and others added 18 commits August 19, 2024 15:25
Adding git modules: kan, efficient_kan, fast_kan
Create efficient kan example
Update burgers inverse kan example
Create burgers inverse example
Create burgers inverse efficient kan example
Update requirements
Update requirements
Adding condition for fast kan
Adding some attributes for different models
Adding some attributes for different models
Update efficient kan example
Update kan parameters in wave physics kan example
Create wave physics fast kan example
Copy link
Member

@SuperSashka SuperSashka left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We need to fix last one and we're good to go on this PR.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It seems that we do not need code here anymore.

Copy link
Member

@SuperSashka SuperSashka left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Okay, we're now good to go.

@dimerf99 dimerf99 merged commit d574422 into main Aug 20, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants