Neural Field Thermal Tomography: A Differentiable Physics Framework for Non-Destructive Evaluation
Organizations: Princeton University
Abstract
Inverse problems for stiff parabolic partial differential equations (PDEs), such as the inverse heat conduction problem (IHCP), are severely ill-posed: the forward map rapidly damps high-frequency interior structure before it reaches the boundary. Soft-constrained physics-informed neural networks (PINNs), which embed the PDE as a residual penalty, suffer from gradient pathology in this regime and tend to fit boundary measurements while leaving the interior field essentially untouched. We propose Neural Field Thermal Tomography (NeFTY), a hard-constrained neural field framework for label-free three-dimensional inverse heat conduction. NeFTY represents the unknown diffusivity as a continuous coordinate-based neural network, and at every optimization step passes the candidate field through a differentiable implicit-Euler heat solver with harmonic-mean interface flux, so that the governing PDE holds exactly on the discretization rather than as a soft penalty. Adjoint gradients propagate the surface reconstruction error back to the network weights at solver-level memory cost, making test-time inversion tractable on a single GPU. Across synthetic 3D benchmarks, NeFTY substantially outperforms soft-constrained PINN variants and a voxel-grid baseline on label-free volumetric recovery, and it transfers to real thermography data, surpassing classical signal-processing baselines in both defect segmentation and depth estimation. Additional details at https://cab-lab-princeton.github.io/nefty/
Figures & tables
| Homogeneous | Layered | |||||||
| Method | MSE ( ) | PSNR | SSIM | IoU | MSE ( ) | PSNR | SSIM | IoU |
| Data-driven (zero-shot) | ||||||||
| U-Net (Sound-Only) [ 45 ] | ||||||||
| Label-free | ||||||||
| PINN [ 44 , 7 ] | ||||||||
| SPINN [ 8 ] | ||||||||
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Category | Parameter | Value |
| Network architecture | MLP depth | 10 |
| MLP hidden width | 512 | |
| Encoding bandwidth | 12 | |
| Skip-connection layer | 4 | |
| Output activation | scaled sigmoid | |
| Domain and grid | Domain size |
| Homogeneous | Layered Composite | |||||||||||
| Defect | Defects | Defects | Defects | Layers | Layers | |||||||
| Method | PSNR | IoU | PSNR | IoU | PSNR | IoU | PSNR | IoU | PSNR | IoU | PSNR | IoU |
| Supervised | ||||||||||||
| U-Net (Full) | ||||||||||||
| U-Net (Sound-Only) | ||||||||||||
| Label-free | ||||||||||||
| Homogeneous | Layered Composite | |||||||||||
| Defect | Defects | Defects | Defects | Layers | Layers | |||||||
| Method | MSE | PSNR | MSE | PSNR | MSE | PSNR | MSE | PSNR | MSE | PSNR | MSE | PSNR |
| Grid Opt. | ||||||||||||
| PINN | ||||||||||||
| NeFTY (Ours) | ||||||||||||
| Homogeneous | Layered | |||||||
| Variant | MSE ( ) | PSNR | SSIM | IoU | MSE ( ) | PSNR | SSIM | IoU |
| Base | ||||||||
| + PE | ||||||||
| + PE, FA | ||||||||
| + PE, FA, | ||||||||
| + PE, FA, , HM | ||||||||
| Method | Params | Colloc. | Iters | ms/iter | Wall-clock | Peak GPU |
| PINN [ 44 ] | M | min | GB | |||
| Causal-PINN [ 56 ] | M | min | GB | |||
| DCGD [ 25 ] | M | min | GB | |||
| SPINN [ 8 ] | M | min | GB | |||
| Grid Opt. | M | — | min | GB | ||
| NeFTY (Ours) | M | — | min | GB |
| Method | Abs Rel | RMSE (mm) | |||
| Swin-UNETR [ 21 ] | |||||
| PPT [ 32 ] | |||||
| TSR [ 50 ] | |||||
| SPINN [ 8 ] | |||||
| Grid Opt. | |||||
| NeFTY (Ours) |
PDE Inverse Problems
Partial differential equation (PDE) inverse problems recover hidden fields, parameters, or sources from observations governed by physical laws.
See all 106 papers