Physics-Informed Neural Networks for Sparse Strain-Field Reconstruction in 4D-STEM
Abstract
Quantitative strain mapping using four-dimensional scanning transmission electron microscopy (4D-STEM) typically requires densely sampled scans that can damage beam-sensitive specimens. We develop a physics-informed neural network (PINN) for sparse 4D-STEM strain reconstruction that embeds elastic equilibrium and Saint-Venant compatibility in the training loss through automatic differentiation. The architecture combines a coordinate-based implicit representation, sine activations with stable second derivatives, frozen residual-scale normalization, an exponential physics-weight ramp, and residual-based adaptive collocation. We apply a sine-activated residual network to an experimental -pixel strain map of domain-structured PbGeSnSeTe. Across - sampling (- probe positions), for reaches at sampling and saturates near by ; the chevron strain-band morphology is recovered from of probe positions. At sampling, the PINN reduces mean absolute error by approximately relative to compressed sensing and relative to Gaussian-process regression. An ablation against an equal-capacity data-only SIREN shows that the PDE prior improves accuracy at extreme sparsity and consistently improves physical self-consistency, but biases the reconstruction when data are abundant. Monte Carlo dropout and mean-field variational inference provide per-pixel epistemic uncertainty maps correlated with reconstruction error. With an appropriate constitutive model, the framework is adaptable to strain mapping across diverse material systems.
Cite
@article{arxiv.2608.01601,
title = {Physics-Informed Neural Networks for Sparse Strain-Field Reconstruction in 4D-STEM},
author = {Roberto dos Reis and Gabriel T. dos Santos and Yukun Liu and Xiaobing Hu and Vinayak P. Dravid},
journal= {arXiv preprint arXiv:2608.01601},
year = {2026}
}
Comments
18 pages, 7 figures, 1 table, and Supplementary Material. Code and processed data are available at https://github.com/rmsreis/pinns-4dstem