# Convergence of inverse problem of heat equation with PINN

**URL:** <https://discourse.julialang.org/t/convergence-of-inverse-problem-of-heat-equation-with-pinn/119866>\
**Category:** Machine Learning\
**Created:** [September 25, 2024, 1:41pm UTC](https://discourse.julialang.org/t/convergence-of-inverse-problem-of-heat-equation-with-pinn/119866 "2024-09-25T13:41:49Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![AliaNajwaMY](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/alianajwamy/32/212187_2.png) [@AliaNajwaMY](https://discourse.julialang.org/u/AliaNajwaMY)\
**Post date:** [September 25, 2024, 1:41pm UTC](https://discourse.julialang.org/t/convergence-of-inverse-problem-of-heat-equation-with-pinn/119866/1 "2024-09-25T13:41:49Z")

</div>

I’m having trouble getting my inverse problem of heat equation with multiple parameters to converge using the neuromancer package (which encodes the PINN model). I am coding using python but was told that I still could ask for feedback here anyway. My code was based on this: [Google Colab](https://colab.research.google.com/github/pnnl/neuromancer/blob/master/examples/PDEs/Part_2_PINN_BurgersEquation.ipynb#scrollTo=OdYMzuSDi7Js) except I changed the data generation to my own method of data generation and my goal is to recover the parameter a, b and c from the equation dy\_t - ady\_xx -b dy\_x -c y =0.

Through the module deepxde which also uses the PINN model using the same data generation code, I was able to recover the forward solution of this heat equation (still having trouble with the inverse problem) but with neuromancer, I haven’t been able to get convergence with both forward and inverse problem. I used Pytorch backends for both packages.

I am new to machine learning so my issues might be trivial/ overlooked but I would appreciate any pointers! Thanks!