In simulation of how water freezes, artificial intelligence breaks the ice

A team based at Princeton University has accurately simulated the initial steps of ice formation by applying artificial intelligence (AI) to solving equations that govern the quantum behavior of individual atoms and molecules.

The resulting simulation describes how water molecules transition into solid ice with quantum accuracy. This level of accuracy, once thought unreachable due to the amount of computing power it would require, became possible when the researchers incorporated deep neural networks, a form of artificial intelligence, into their methods. The study was published in the journal Proceedings of the National Academy of Sciences.

“In a sense, this is like a dream come true,” said Roberto Car, Princeton’s Ralph W. *31 Dornte Professor in Chemistry, who co-pioneered the approach of simulating molecular behaviors based on the underlying quantum laws more than 35 years ago. “Our hope then was that eventually we would be able to study systems like this one, but it was not possible without further conceptual development, and that development came via a completely different field, that of artificial intelligence and data science.”

The ability to model the initial steps in freezing water, a process called ice nucleation, could improve accuracy of weather and climate modeling as well as other processing like flash-freezing food.

The new approach enables the researchers to track the activity of hundreds of thousands of atoms over time periods that are thousands of times longer, albeit still just fractions of a second, than in early studies.

Car co-invented the approach to using underlying quantum mechanical laws to predict the physical movements of atoms and molecules. Quantum mechanical laws dictate how atoms bind to each other to form molecules, and how molecules join with each other to form everyday objects.

Car and Michele Parrinello, a physicist now at the Istituto Italiano di Tecnologia in Italy, published their approach, known as “ab initio” (Latin for “from the beginning”) molecular dynamics, in a groundbreaking paper in 1985.

But quantum mechanical calculations are complex and take tremendous amounts of computing power. In the 1980’s, computers could simulate just a hundred atoms over spans of a few trillionths of a second. Subsequent advances in computing and the advent of modern supercomputers boosted the number of atoms and timespan of the simulation, but the result fell far short of the number of atoms needed to observe complex processes such as ice nucleation.

AI provided an attractive potential solution. Researchers train a neural network, named for its similarities to the workings of the human brain, to recognize a comparatively small number of selected quantum calculations. Once trained, the neural network can calculate the forces between atoms that it has never seen before with quantum mechanical accuracy. This “machine learning” approach is already in use in everyday applications such as voice recognition and self-driving automobiles.

In the case of AI applied to molecular modeling, a major contribution came in 2018 when Princeton graduate student Linfeng Zhang, working with Car and Princeton professor of mathematics Weinan E, found a way to apply deep neural networks to modeling quantum-mechanical interatomic forces. Zhang, who earned his Ph.D. in 2020 and is now a research scientist at the Beijing Institute of Big Data Research, called the approach “deep potential molecular dynamics.”

In the current paper, Car and postdoctoral researcher Pablo Piaggi along with colleagues applied these techniques to the challenge of simulating ice nucleation. Using deep potential molecular dynamics, they were able to run simulations of up to 300,000 atoms using significantly less computing power, for much longer timespans than were previously possible. They carried out the simulations on Summit, one of the world’s fastest supercomputers, located at Oak Ridge National Laboratory.

This work provides one of the best studies of ice nucleation, said Pablo Debenedetti, Princeton’s dean for research and the Class of 1950 Professor of Engineering and Applied Science, and a co-author of the new study.

“Ice nucleation is one of the major unknown quantities in weather prediction models,” Debenedetti said. “This is a quite significant step forward because we see very good agreement with experiments. We’ve been able to simulate very large systems, which was previously unthinkable for quantum calculations.”

Currently, climate models obtain estimates of how fast ice nucleates primarily from observations made in laboratory experiments, but these correlations are descriptive, not predictive, and are valid over a limited range of experimental conditions. In contrast, molecular simulations of the type done in this study can produce simulations that are predictive of future situations, and can estimate ice formation under extreme conditions of temperature and pressure, such as on other planets.

“The deep potential methodology used in our study will help realize the promise of ab initio molecular dynamics to produce valuable predictions of complex phenomena, such as chemical reactions and the design of new materials,” said Athanassios Panagiotopoulos, the Susan Dod Brown Professor of Chemical and Biological Engineering and a co-author of the study.

“The fact that we are studying very complex phenomena from the fundamental laws of nature, to me that is very exciting,” said Piaggi, the study’s first author and a postdoctoral research associate in chemistry at Princeton. Piaggi earned his Ph.D. working with Parrinello on the development of new techniques to study rare events, such as nucleation, using computer simulation. Rare events take place over timescales that are longer than the simulation times that can be afforded, even with the help of AI, and specialized techniques are needed to accelerate them.

Jack Weis, a graduate student in chemical and biological engineering, helped increase the likelihood of observing nucleation by “seeding” tiny ice crystals into the simulation. “The goal of seeding is to increase the likelihood that water will form ice crystals during the simulation, allowing us to measure the nucleation rate,” said Weis, who is advised by Debenedetti and Panagiotopoulos.

Water molecules consist of two hydrogen atoms and an oxygen atom. The electrons around each atom determine how atoms can bond with each other to form molecules.

“We start with the equation that describes how electrons behave,” Piaggi said. “Electrons determine how atoms interact, how they form chemical bonds, and virtually the whole of chemistry.”

The atoms can exist in literally millions of different arrangements, said Car, who is director of the Chemistry in Solution and at Interfaces center, funded by the U.S. Department of Energy Office of Science and including regional universities.

“The magic is that because of some physical principles, the machine is able to extrapolate what happens in a relatively small number of configurations of a small collection of atoms to the countless arrangements of a much bigger system,” Car said.

Although AI approaches have been available for some years, researchers have been cautious about applying them to calculations of physical systems, Piaggi said. “When machine learning algorithms started to become popular, a big part of the scientific community was skeptical, because these algorithms are a black box. Machine learning algorithms don’t know anything about the physics, so why would we use them?”

In the last couple of years, however, there has been a significant change in this attitude, Piaggi said, not only because the algorithms work but also because researchers are using their knowledge of physics to inform the machine learning models.

For Car, it is satisfying to see the work started three decades ago come to fruition. “The development came via something that was developed in a different field, that of data science and applied mathematics,” Car said. “Having this kind of cross interaction between different fields is very important.”

This work was supported by the U.S. Department of Energy (grant DE-731 SC0019394) and used resources of the Oak Ridge Leadership Computing Facility (grant DE-AC05-00OR22725) at the Oak Ridge National Laboratory. Simulations were substantially performed using the Princeton Research Computing resources at Princeton University. Pablo Piaggi was supported by an Early Postdoc.Mobility fellowship from the Swiss National Science Foundation.

The study, “Homogeneous ice nucleation in an ab initio machine learning model of water,” by Pablo M. Piaggi, Jack Weis, Athanassios Z. Panagiotopoulos, Pablo G. Debenedetti, and Roberto Car, was published in the journal Proceedings of the National Academy of Sciences the week of August 8, 2022. DOI: /10.1073/pnas.2207294119.