Artificial intelligence (AI) is transforming every scientific field, from biology to materials science. When it comes to some types of X-ray experiments, new AI approaches have enabled researchers to obtain a more accurate analysis of their samples and to do so in a much shorter amount of time.
A group of researchers at the U.S. Department of Energy’s (DOE) Argonne National Laboratory are harnessing AI to perform the challenging task of analyzing data from high-energy X-ray experiments. With a new neural network-based method called BraggNN, the Argonne team can more precisely identify Bragg peaks — data points that indicate positions and orientations of tiny individual crystals — in a fraction of the time they used to.
“Conventional methods work like trying on a suit at a tailor’s. First you have to try on an estimated shape, and then do form-fitting after it. With our technique, using a neural network, it’s like generating the perfect suit instantly from just a photograph of a person.” — Hemant Sharma, Argonne National Laboratory
Neural networks (the NN in BraggNN) are algorithms that look for patterns in data and, over time, learn to predict outcomes, speeding up the analysis of that data.
“BraggNN gives us far greater efficiency and speed than conventional techniques,” said Argonne’s Antonino Miceli, an author on the paper and a group leader at the Advanced Photon Source (APS), a DOE Office of Science user facility at Argonne.
In recent years, a technique called high-energy diffraction microscopy (HEDM) has become one of the most popular ways scientists use to accurately characterize complicated materials with high resolution. Although HEDM has proven to be a great improvement over conventional techniques, it can also be expensive and time-consuming. It involves the collection of enormous datasets, analysis of millions of Bragg diffraction peaks and reconstruction of the sample using those peaks.
The impending upgrade of the APS is expected to vastly improve the speed of HEDM data acquisition to a minute or less. But the computing time to complete analysis of multiple Bragg peaks can extend to hours or weeks, even with the largest supercomputers. Not only do such delays slow research, they also prevent the use of HEDM information to steer experiments. For instance, scientists studying how cracks form in materials could use the analyzed data to track where those cracks are emerging, in close to real time.
To address these challenges, the researchers working at the APS turned to AI to accelerate and streamline Bragg peak analysis. The conventional method involves using a 2D or 3D model and fitting the peak data to it, but the research team’s new model can directly determine the peak positions from the data.
“Conventional methods work like trying on a suit at a tailor’s,” said Argonne computational scientist Hemant Sharma, an author on the study. “First you have to try on an estimated shape, and then do form-fitting after it. With our technique, using a neural network, it’s like generating the perfect suit instantly from just a photograph of a person.”
After the model was trained on data containing diffraction peaks, the researchers were able to dramatically speed up analysis and improve accuracy. “The true accomplishment is that we made peak determinations much faster and also delivered sub-pixel accuracy — the gold standard for making useful conclusions,” said Argonne computer scientist Zhengchun Liu, the first author on the paper.
The advanced computational methods used by BraggNN are especially conducive for use on a graphics processing unit (GPU) chip, which helps further accelerate its performance.
The team used the ThetaGPU system at the Argonne Leadership Computing Facility (ALCF), a DOE Office of Science user facility, and the Cerebras AI platform at the ALCF AI Testbed to rapidly train the model.
A paper based on the study was published in the Journal of the International Union of Crystallography. Other authors of the study include Argonne’s Jun-Sang Park, Peter Kenesei, Jonathan Almer, Rajkumar Kettimuthu and Ian Foster.
About the Advanced Photon Source
The U. S. Department of Energy Office of Science’s Advanced Photon Source (APS) at Argonne National Laboratory is one of the world’s most productive X-ray light source facilities. The APS provides high-brightness X-ray beams to a diverse community of researchers in materials science, chemistry, condensed matter physics, the life and environmental sciences, and applied research. These X-rays are ideally suited for explorations of materials and biological structures; elemental distribution; chemical, magnetic, electronic states; and a wide range of technologically important engineering systems from batteries to fuel injector sprays, all of which are the foundations of our nation’s economic, technological, and physical well-being. Each year, more than 5,000 researchers use the APS to produce over 2,000 publications detailing impactful discoveries, and solve more vital biological protein structures than users of any other X-ray light source research facility. APS scientists and engineers innovate technology that is at the heart of advancing accelerator and light-source operations. This includes the insertion devices that produce extreme-brightness X-rays prized by researchers, lenses that focus the X-rays down to a few nanometers, instrumentation that maximizes the way the X-rays interact with samples being studied, and software that gathers and manages the massive quantity of data resulting from discovery research at the APS.
This research used resources of the Advanced Photon Source, a U.S. DOE Office of Science User Facility operated for the DOE Office of Science by Argonne National Laboratory under Contract No. DE-AC02-06CH11357.
Argonne National Laboratory seeks solutions to pressing national problems in science and technology. The nation’s first national laboratory, Argonne conducts leading-edge basic and applied scientific research in virtually every scientific discipline. Argonne researchers work closely with researchers from hundreds of companies, universities, and federal, state and municipal agencies to help them solve their specific problems, advance America’s scientific leadership and prepare the nation for a better future. With employees from more than 60 nations, Argonne is managed by UChicago Argonne, LLC for the U.S. Department of Energy’s Office of Science.
The U.S. Department of Energy’s Office of Science is the single largest supporter of basic research in the physical sciences in the United States and is working to address some of the most pressing challenges of our time. For more information, visit https://energy.gov/science.