Skip to content

Add refined MatrixKAN with mean-based spline path and MNIST benchmark#2

Open
mehizelali wants to merge 1 commit into
OSU-STARLAB:mainfrom
mehizelali:speedupMatrixKAN
Open

Add refined MatrixKAN with mean-based spline path and MNIST benchmark#2
mehizelali wants to merge 1 commit into
OSU-STARLAB:mainfrom
mehizelali:speedupMatrixKAN

Conversation

@mehizelali

Copy link
Copy Markdown

Summary

  • Add 'refinedMatrixKAN.py' with mean-aggregated spline path and faster forward pass
  • Add MNIST comparison experiment vs original layer (~1.5× faster, similar/better accuracy)

Introduce refinedMatrixKAN.py: mean over (num+k) basis/coef axes, nn.Linear
spline projection, and faster forward pass. Add MNIST compare experiment;
refined matches original test accuracy (92.6% vs 92.1%) at ~1.5× speed.
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.

1 participant