Skip to content
Book Open access

From Noisy STEM to Crystal Structure: Evidence-Structure CoDiffusion under Composition Constraints

Aug 2026 · Proceedings of the 32nd ACM SIGKDD Conference on Knowledge Discovery and Data Mining V.2 · pp. 10578-10588 · 0 citations · 18 references

Abstract

By directly resolving lattice periodicity and atomic columns, scanning transmission electron microscopy (STEM) offers rich structural cues, yet recovering a simulation-ready crystal structure from a single noisy STEM image remains an ill-posed inverse problem. In realistic acquisitions, corruption can obscure geometry-aligned cues, and composition-only structure generation is highly multimodal, yielding many plausible 2D slab candidates. We cast this task as a coupled inference problem that separates evidence recovery from structure inference under a minimal forward model. We introduce STEM2Crystal CoDiffusion (SCCD), a dual-diffusion framework that explicitly separates evidence recovery from structure inference and couples them via bidirectional feature exchange. SCCD comprises (i) an evidence diffusion branch that denoises a structure-aligned, mask-like evidence map conditioned on the noisy STEM image, and (ii) a crystal diffusion branch that jointly denoises the lattice and atomic coordinates under the given composition constraint. A bidirectional co-diffusion update enables iterative refinement: crystallographic context regularizes evidence denoising, while denoised image evidence provides noise-robust geometric cues that sharpen structure inference. For controlled evaluation, we release a large-scale synthetic benchmark with explicit composition constraints, controlled noise regimes, and projection-derived supervision signals. Across multiple complementary metrics and all noise levels, SCCD consistently outperforms strong baselines.

Read PDF

Similar papers

Book Open access Aug 2026

TGD-CSP: Reliable Crystal Structure Prediction with Template-Guided Diffusion and Energy-Based Optimization

Crystal Structure Prediction (CSP), the task of determining stable atomic arrangements from chemical composition alone, remains a central challenge in computational materials science with direct implications for accelerating materials discovery. While recent diffusion-based generative models achieve impressive results by conditioning on space-group symmetry information, this paradigm exposes three fundamental challenges: (1) unreliable symmetry inference from composition, (2) reliance on symmetry-only priors without comprehensive structural geometric guidance, and (3) prior-induced distribution shift caused by inaccurate or overly strong constraints. To address these challenges, we propose TGD-CSP, a three-stage generative framework that: (1) learns a cross-modal embedding space to retrieve structurally relevant templates directly from composition, thereby providing reliable symmetry priors; (2) guides diffusion-based generation via score-based conditioning that explicitly incorporates comprehensive geometric information from retrieved templates; and (3) fine-tunes the generative policy via reinforcement learning with an energy-based reward to alleviate prior-induced distribution shift and mitigate biased generation. % Evidence: Concrete results TGD-CSP achieves match rates of 74.56% and 75.25% on Perov-5 and MP-20, respectively, including a 42% relative improvement over state-of-the-art methods on Perov-5. It reduces RMSE to 0.0259 and 0.0258, respectively, while yielding structures with formation energies closely matching those of ground-truth structures. Our results demonstrate that TGD-CSP enables reliable generation from composition alone, significantly improving the practicality of generative models for real materials discovery.

Lu Yang, Tiantian Xu, X. Liu et al. · 0 citations
Preprint Aug 2026

Composition-Driven Phase Evolution in Sm-Doped BiFeO3 via Latent-Field Reconstruction of Atomically Resolved STEM Data

Functionalities of ferroelectric materials are governed by the spatial organization and coupling of polarization, strain, lattice rotation, and structural order accessible via atomically resolved scanning transmission electron microscopy (STEM) images. Quantitative interpretation of atomic-resolution STEM data has conventionally relied on locating atomic columns and converting their fitted coordinates into local structural descriptors. Here, we develop a field-based approach in which atomic-resolution images are represented by spatially varying latent Bragg fields, whose amplitudes and phases provide continuous maps of crystalline order, lattice displacement, strain, rotation, and mode-specific residual structure. The observed atomically resolved images are decoded from the latent fields. We apply this framework to image series of Sm-substituted BiFeO3 spanning 0-20% Sm and crossing the composition-driven boundary between the R3c ferroelectric phase and the orthorhombic, nonpolar Pnma phase. Conventional atom-resolved parameterization is used as an independent validation, showing that reconstructed Bragg amplitude tracks local atomic-column intensity and that field-derived shear reproduces unit-cell angular distortions obtained from atom fitting. The combined analysis reveals a systematic evolution from extended ferroelectric domains at low Sm concentration, through the appearance and growth of localized regions with period-doubled Pnma order at intermediate compositions, to a connected Pnma-dominated state at high Sm content. The period-doubled order is accompanied by enhanced shear and lattice rotation and by progressive reorganization of the ferroelectric domain structure. These results establish latent-field reconstruction as a physically interpretable complement to atom finding and provide a unified framework for resolving composition-driven phase evolution in ferroic materials.

Newsha Javanmardi, C. Nelson, Anna N. Morozovska et al. · 0 citations
#artificial intelligence Preprint Aug 2026

ED-CSP: Crystal Structure Prediction from Electron Diffraction

This work introduces ED-CSP, a machine learning framework that predicts crystal structures from chemical composition, atom count, and multiple detector-plane ED spot sets and establishes a benchmark for generative crystal structure prediction from sparse ED observations and provides a foundation for future transfer to experimental data.

Germain Poloudenny, Arnaud Demortière, Yael Fregier · 0 citations
Open access Aug 2026

Tackling real-world crystal structure prediction from powder X-ray diffraction data

Crystal structure prediction (CSP) from powder diffraction data is a central challenge in materials chemistry. Machine learning (ML) models show promise, but most are trained on idealized simulated data, limiting reliability on real experiments. Here, we assess real-world behaviour using the previously published deCIFer model as an example of PXRD-conditioned generative CSP. deCIFer is an autoregressive transformer that conditions each step of structure generation on encoded PXRD data, guiding token-wise predictions of space group, lattice parameters, and atomic positions. Using controlled robustness tests, we quantify performance under realistic artefacts (noise, background, peak asymmetry, and Scherrer broadening) and introduce metrics for accuracy and predictive uncertainty. deCIFer adapts smoothly to signal distortions and improves over unconditioned baselines when diffraction features remain informative, while expressing appropriate uncertainty as the PXRD pattern becomes underdetermined. Experimental PXRD tests recover the known structures of Si and CeO2 and expose the expected limitations for lower-symmetry Fe2O3 and nanocrystalline CeO2. Overall, ML-based CSP is fundamentally limited by the information content of PXRD, but can accelerate expert workflows by rapidly generating chemically plausible candidates and quantifying uncertainty, making such models valuable human-in-the-loop tools for real-world structure determination.

Frederik L. Johansen, A. Sapnik, Erik B. Dam et al. · 0 citations
Preprint Aug 2026

DynaCrys: Crystal Generation with Dynamic Space-Group Diffusion

The search for new crystalline materials spans an enormous compositional and structural space. Generating candidates in this space requires jointly modeling discrete crystallographic symmetry, elemental composition, and continuous geometry. We introduce DynaCrys, a generative model for crystals in which the space group co-evolves with Wyckoff occupations and elements through a coupled symbolic diffusion process. The structured space-group transitions follow crystallographic group-subgroup relations. As the space group changes, a shared, pretrained symmetry codebook provides both the legality-constrained stochastic decoder and the symmetry-constrained crystal-geometry model with a common representation of the corresponding Wyckoff vocabulary. Across large-scale evaluations using two independent relaxation-and-evaluation engines, DynaCrys achieves best-in-class performance in symmetry-aware discovery of stable, unique, and novel crystals, both overall and under the additional requirement of nontrivial post-relaxation symmetry. It also enables fast sampling while generating structures with consistently low relaxation-induced structural displacements.

Zhuotao Jin, Xiaoyun Wang, Nicholas Brawand et al. · 0 citations
Jul 2026

Local Geometry Recovers but Cooperative Structure Does Not: Residue-Resolved Limits of All-Atom Reconstruction from Single-Bead Coarse-Grained Disordered Protein Ensembles.

The benchmark quantifies which atomistic properties can be recovered after projecting a single-bead coarse-grained force field ensemble into all-atom space and identifies improved CG geometric encoding and sequence-conditioned AI priors as the directions for further progress.

Jianxiang Huang, Xin Qiao, Ning Liu et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.