Benchmarking Machine Learning Approaches for Polarization Mapping in Ferroelectrics Using 4D-STEM

• URL: http://arxiv.org/abs/2603.15582v1
• Date: Mon, 16 Mar 2026 17:45:28 GMT
• Title: Benchmarking Machine Learning Approaches for Polarization Mapping in Ferroelectrics Using 4D-STEM
• Abstract summary: We benchmark multiple machine learning models to automate the detection of polarization directions from 4D-STEM diffraction patterns in ferroelectric potassium sodium niobate.<n>Our findings guide the development of robust, transferable machine learning tools for electron microscopy analysis.
• Score: 0.5660019164402329
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Four-dimensional scanning transmission electron microscopy (4D-STEM) provides rich, atomic-scale insights into materials structures. However, extracting specific physical properties - such as polarization directions essential for understanding functional properties of ferroelectrics - remains a significant challenge. In this study, we systematically benchmark multiple machine learning models, namely ResNet, VGG, a custom convolutional neural network, and PCA-informed k-Nearest Neighbors, to automate the detection of polarization directions from 4D-STEM diffraction patterns in ferroelectric potassium sodium niobate. While models trained on synthetic data achieve high accuracy on idealized synthetic diffraction patterns of equivalent thickness, the domain gap between simulation and experiment remains a critical barrier to real-world deployment. In this context, a custom made prototype representation training regime and PCA-based methods, combined with data augmentation and filtering, can better bridge this gap. Error analysis reveals periodic missclassification patterns, indicating that not all diffraction patterns carry enough information for a successful classification. Additionally, our qualitative analysis demonstrates that irregularities in the model's prediction patterns correlate with defects in the crystal structure, suggesting that supervised models could be used for detecting structural defects. These findings guide the development of robust, transferable machine learning tools for electron microscopy analysis.

Related papers

• Multi4D: an end-to-end neural network for structural determination at complex material interfaces [5.9461952475226445]
  We introduce Multi4D, a physics-informed neural network framework for automated multi-component crystallographic identification.<n>By combining a latent-space Transformer for physics-constrained style translation with a rotation-invariant convolutional neural network for orientation-agnostic classification, this approach translates diffraction datasets into deterministic crystallographic maps with 98.82% accuracy.<n>We apply Multi4D to generate high-fidelity structural maps of complex superconducting heterostructures, corroded alloy surfaces, and degraded solid-state battery interfaces down to single-nanometer spatial resolution.
  arXiv  Detail & Related papers  (2026-09-13T07:12:53Z)
• Mapping Order in Semicrystalline Polymers using Machine Learning of Nanobeam Electron Diffraction [0.0]
  We train a machine learning (ML) model to detect polymer diffraction peaks and their intensities from synthetic data.<n>Compared to correlative peak detection algorithms, the ML model is significantly faster and outperforms correlative algorithms in almost all cases.
  arXiv  Detail & Related papers  (2026-07-18T00:54:23Z)
• Why Physics Still Matters: Improving Machine Learning Prediction of Material Properties with Phonon-Informed Datasets [0.32622301272834514]
  We assess the effectiveness of graph neural network (GNN) models trained on two fundamentally different types of datasets.<n>As a case study, we address the challenging task of predicting electronic and mechanical properties of a prototypical family of optoelectronic materials.<n>We find that the phonons-informed model consistently outperforms the randomly trained counterpart, despite relying on fewer data points.
  arXiv  Detail & Related papers  (2025-11-19T08:16:10Z)
• Foundation Models for Discovery and Exploration in Chemical Space [57.97784111110166]
  MIST is a family of molecular foundation models trained on large unlabeled datasets.<n>We demonstrate the ability of these models to solve real-world problems across chemical space.
  arXiv  Detail & Related papers  (2025-10-20T17:56:01Z)
• Direct Dual-Energy CT Material Decomposition using Model-based Denoising Diffusion Model [105.95160543743984]
  We propose a deep learning procedure called Dual-Energy Decomposition Model-based Diffusion (DEcomp-MoD) for quantitative material decomposition.<n>We show that DEcomp-MoD outperform state-of-the-art unsupervised score-based model and supervised deep learning networks.
  arXiv  Detail & Related papers  (2025-07-24T01:00:06Z)
• STEM Diffraction Pattern Analysis with Deep Learning Networks [0.0]
  This work presents a machine learning-based approach for predicting Euler angles directly from scanning transmission electron microscopy (STEM) diffraction patterns (DPs)<n>It enables the automated generation of high-resolution crystal orientation maps, facilitating the analysis of internal microstructures at the nanoscale.<n>Three deep learning architectures--convolutional neural networks (CNNs), Dense Convolutional Networks (DenseNets), and Shifted Windows (Swin) Transformers--are evaluated, using an experimentally acquired dataset labelled via a commercial TM algorithm.
  arXiv  Detail & Related papers  (2025-07-02T16:58:09Z)
• Causal Discovery from Data Assisted by Large Language Models [50.193740129296245]
  It is essential to integrate experimental data with prior domain knowledge for knowledge driven discovery.<n>Here we demonstrate this approach by combining high-resolution scanning transmission electron microscopy (STEM) data with insights derived from large language models (LLMs)<n>By fine-tuning ChatGPT on domain-specific literature, we construct adjacency matrices for Directed Acyclic Graphs (DAGs) that map the causal relationships between structural, chemical, and polarization degrees of freedom in Sm-doped BiFeO3 (SmBFO)
  arXiv  Detail & Related papers  (2025-03-18T02:14:49Z)
• Exploring structure diversity in atomic resolution microscopy with graph neural networks [18.903519247639355]
  Deep learning is a powerful tool to explore structure diversity in a fast, accurate, and intelligent manner. This work provides a powerful tool to explore structure diversity in a fast, accurate, and intelligent manner.
  arXiv  Detail & Related papers  (2024-10-23T07:48:35Z)
• Capturing dynamical correlations using implicit neural representations [85.66456606776552]
  We develop an artificial intelligence framework which combines a neural network trained to mimic simulated data from a model Hamiltonian with automatic differentiation to recover unknown parameters from experimental data. In doing so, we illustrate the ability to build and train a differentiable model only once, which then can be applied in real-time to multi-dimensional scattering data.
  arXiv  Detail & Related papers  (2023-04-08T07:55:36Z)
• Machine Learning Benchmarks for the Classification of Equivalent Circuit Models from Electrochemical Impedance Spectra [0.0]
  We showcase machine learning methods to classify the ECMs of 9,300 impedance spectra provided by QuantumScape for the BatteryDEV hackathon. A key remaining challenge is the identifiability of the labels, underlined by the model performances and the comparison of misclassified spectra.
  arXiv  Detail & Related papers  (2023-02-07T10:08:35Z)
• Disentangling multiple scattering with deep learning: application to strain mapping from electron diffraction patterns [48.53244254413104]
  We implement a deep neural network called FCU-Net to invert highly nonlinear electron diffraction patterns into quantitative structure factor images. We trained the FCU-Net using over 200,000 unique dynamical diffraction patterns which include many different combinations of crystal structures. Our simulated diffraction pattern library, implementation of FCU-Net, and trained model weights are freely available in open source repositories.
  arXiv  Detail & Related papers  (2022-02-01T03:53:39Z)
• Learning Bijective Feature Maps for Linear ICA [73.85904548374575]
  We show that existing probabilistic deep generative models (DGMs) which are tailor-made for image data, underperform on non-linear ICA tasks. To address this, we propose a DGM which combines bijective feature maps with a linear ICA model to learn interpretable latent structures for high-dimensional data. We create models that converge quickly, are easy to train, and achieve better unsupervised latent factor discovery than flow-based models, linear ICA, and Variational Autoencoders on images.
  arXiv  Detail & Related papers  (2020-02-18T17:58:07Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.