PINN-Heat-Equation-Solver

🔥 PINN Heat Equation Solver

Physics-Informed Neural Network (PINN) for solving the 2D Transient Heat Equation
Built for CV/Portfolio — B.Tech Computer Engineering | AI/ML/DL Track

---

📌 Project Overview

This project implements a Physics-Informed Neural Network (PINN) to solve the 2D transient heat diffusion PDE without requiring any labeled simulation data. The neural network learns to satisfy both the governing PDE and the boundary/initial conditions simultaneously, using automatic differentiation (via PyTorch autograd) to compute spatial and temporal derivatives.

Problem Statement

∂u/∂t = α (∂²u/∂x² + ∂²u/∂y²)    on Ω = [0,1]² × [0,1]
IC:  u(x, y, 0) = sin(πx)·sin(πy)
BC:  u = 0 on all boundaries (Dirichlet)
Exact: u(x,y,t) = e^(−2π²αt) · sin(πx) · sin(πy)

---

🧠 Architecture

Input: (x, y, t) → 3 neurons
[Linear(3→64) → Tanh] → [Linear(64→128) → Tanh]
[Linear(128→128) → Tanh] → [Linear(128→64) → Tanh]
Output: u(x, y, t) → 1 neuron

Loss Function:

L_total = L_pde + λ_bc · L_bc + λ_ic · L_ic

---

📁 Full Repository Structure

PINN-Heat-Equation-Solver/
├── README.md
├── requirements.txt
│
├── pinn_heat_equation.py        # Core: HeatPINN model + autograd PDE residual
├── train.py                     # Two-phase training: Adam → L-BFGS
├── evaluate.py                  # Error metrics (global + per time-slice)
├── visualize.py                 # Prediction vs. analytical plots
├── inverse_problem.py           # ★ Identify unknown α from noisy sensor data
├── benchmark.py                 # ★ PINN vs. Finite Difference Method (FDM)
│
├── configs/
│   └── config.yaml              # All hyperparameters (centralized)
│
├── utils/
│   ├── __init__.py
│   ├── losses.py                # Modular PDE / BC / IC loss functions
│   ├── collocation.py           # Uniform collocation point samplers
│   ├── adaptive_sampling.py     # ★ Residual-adaptive refinement (RAR)
│   └── schedulers.py            # ★ Warmup-cosine & cyclic LR schedulers
│
├── experiments/
│   ├── ablation_study.py        # ★ Depth × Width × Activation × N_col study
│   └── uncertainty_estimation.py # ★ MC-Dropout uncertainty quantification
│
└── notebooks/
    └── PINN_Heat_Equation.ipynb # Full interactive walkthrough

★ = modules added beyond baseline; these are what make this project stand out.

---

🚀 Quick Start

git clone https://huggingface.co/sarvesh74/PINN-Heat-Equation-Solver
cd PINN-Heat-Equation-Solver
pip install -r requirements.txt

# Forward problem
python train.py --config configs/config.yaml
python evaluate.py --checkpoint results/pinn_best.pth
python visualize.py

# Inverse problem (identify unknown α from sensor data)
python inverse_problem.py --alpha_true 0.01 --alpha_init 0.05 --n_obs 200

# Benchmark vs. classical FDM
python benchmark.py --checkpoint results/pinn_best.pth

# Ablation study
python experiments/ablation_study.py

# Uncertainty quantification
python experiments/uncertainty_estimation.py

---

📊 Results

---

🔬 Key Concepts Demonstrated

---

🎓 Academic Context

Course: Deep Learning / Neural Computation (CE 401)
Institution: Pillai College of Engineering, New Panvel
Degree: B.Tech Computer Engineering
Domain: Scientific Machine Learning / Computational Physics

---

📚 References

1. Raissi et al. (2019). Physics-informed neural networks. Journal of Computational Physics, 378, 686–707.
2. Karniadakis et al. (2021). Physics-informed machine learning. Nature Reviews Physics, 3(6), 422–440.
3. Lu et al. (2021). DeepXDE: A deep learning library for solving differential equations. SIAM Review, 63(1), 208–228.
4. Gal & Ghahramani (2016). Dropout as a Bayesian approximation. ICML.

---

📝 License

MIT License — free to use, modify, and distribute with attribution.

---

Made with ❤️ by Sarvesh — B.Tech CSE (AI/ML/DL)