NeuralPDE.jl icon indicating copy to clipboard operation
NeuralPDE.jl copied to clipboard

Added i) Inverse Dirichlet Adaptive Loss and ii) Neural Tangent Kernel Loss Implementations

Open sphinx-tech opened this issue 2 years ago • 7 comments
trafficstars

Neural Kernel Tangent Adaptive Loss

Inverse Dirichlet Adaptive Loss

sphinx-tech avatar Apr 01 '23 14:04 sphinx-tech

Instead of using the Jacobians for K (as implemented) by the authors; I have used dot products because we were only concerned with the main diagonal entries of the K matrix for our implementation - in Algorithm 1 (Pg. 10)

sphinx-tech avatar Apr 01 '23 14:04 sphinx-tech

Put Plots and Revise in to your v1.8 environment, and add using Plots, Revise to your juliarc

xtalax avatar Apr 01 '23 14:04 xtalax

Note that for this there is starter code in https://github.com/SciML/NeuralPDE.jl/pull/504 and https://github.com/SciML/NeuralPDE.jl/pull/506. It doesn't make sense to do both in the same PR though.

ChrisRackauckas avatar Apr 02 '23 11:04 ChrisRackauckas

@xtalax @ChrisRackauckas I have made some changes according to your suggestions.

  • Two independent commits have been made that deal with the different loss functions
  • The Inverse Dirichlet is performing well, but the loss in Neural Tangent kernel is high

Pls let me know of any improvements that can be made in the implementations. Also any direction on how to improve the testing part would be very helpful !

sphinx-tech avatar Apr 03 '23 05:04 sphinx-tech

Rebase onto master.

ChrisRackauckas avatar Apr 03 '23 09:04 ChrisRackauckas

The next step would be to setup tutorials and benchmarks with these.

ChrisRackauckas avatar Apr 03 '23 10:04 ChrisRackauckas

Looks like tests failed.

ChrisRackauckas avatar Apr 04 '23 11:04 ChrisRackauckas