Automated defect identification in coherent diffraction imaging with smart continual learning

Orcun Yildiz, Krishnan Raghavan, Henry Chan, Mathew J. Cherukara, Prasanna Balaprakash, Subramanian Sankaranarayanan, Tom Peterka

Research output: Contribution to journalArticlepeer-review

Abstract

X-ray Bragg coherent diffraction imaging is a powerful technique for 3D materials characterization. However, obtaining X-ray diffraction data is difficult and computationally intensive, motivating the need for automated processing of coherent diffraction images, with the goal of minimizing the number of X-ray datasets needed. We automate a machine learning approach to identify crystalline line defects in samples from the raw coherent diffraction data, in a workflow coupling coherent diffraction data generation with training and inference of deep neural network defect classifiers. In particular, we adopt a continual learning approach, where we generate training data as needed based on the accuracy of the defect classifier instead of generating all training data a priori. Moreover, we develop a novel data generation mechanism to improve the efficiency of defect identification beyond the previously published continual learning approach. We call the improved method smart continual learning. The results show that our approach improves the accuracy of defect classifiers and reduces training data requirements by up to 98% compared with prior approaches.

Original languageEnglish
JournalNeural Computing and Applications
DOIs
StateAccepted/In press - 2024

Funding

This material is based upon work supported by the U.S. Department of Energy, Office of Science, Office of Advanced Scientific Computing Research, under contract number DE-AC02-06CH11357. We gratefully acknowledge the computing resources provided on Swing, a high-performance computing cluster operated by the Laboratory Computing Resource Center at Argonne National Laboratory. Work performed at the Center for Nanoscale Materials and Advanced Photon Source, both U.S. Department of Energy Office of Science User Facilities, was supported by the U.S. DOE, Office of Basic Energy Sciences, under Contract No. DE-AC02-06CH11357.

FundersFunder number
Basic Energy Sciences
Center for Nanoscale Materials and Advanced Photon Source
U.S. Department of Energy
Argonne National Laboratory
Office of Science
Laboratory Computing Resource Center
Advanced Scientific Computing ResearchDE-AC02-06CH11357

    Keywords

    • Catastrophic forgetting
    • Continual learning
    • Controlled experiments
    • Defect identification
    • HPC workflows
    • Smart data generation

    Fingerprint

    Dive into the research topics of 'Automated defect identification in coherent diffraction imaging with smart continual learning'. Together they form a unique fingerprint.

    Cite this