English

Physics-Informed Neural Networks for Sparse Strain-Field Reconstruction in 4D-STEM

Materials Science 2026-08-03 v1 Instrumentation and Detectors

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 180×400180\times400-pixel strain map of domain-structured PbGeSnSe1.5_{1.5}Te1.5_{1.5}. Across 11-7575% sampling (720720-54,00054{,}000 probe positions), R2R^2 for εxx\varepsilon_{xx} reaches 0.800.80 at 1010% sampling and saturates near 0.860.86 by 2525%; the chevron strain-band morphology is recovered from 1010% of probe positions. At 1010% sampling, the PINN reduces mean absolute error by approximately 2626% relative to compressed sensing and 2222% 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