Materials Learning Algorithms (MALA): Scalable Machine Learning for Electronic Structure Calculations in Large-Scale Atomistic Simulations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cangi, Attila, Fiedler, Lenz, Brzoza, Bartosz, Shah, Karan, Callow, Timothy J., Kotik, Daniel, Schmerler, Steve, Barry, Matthew C., Goff, James M., Rohskopf, Andrew, Vogel, Dayton J., Modine, Normand, Thompson, Aidan P., Rajamanickam, Sivasankaran
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910720517472256
author Cangi, Attila
Fiedler, Lenz
Brzoza, Bartosz
Shah, Karan
Callow, Timothy J.
Kotik, Daniel
Schmerler, Steve
Barry, Matthew C.
Goff, James M.
Rohskopf, Andrew
Vogel, Dayton J.
Modine, Normand
Thompson, Aidan P.
Rajamanickam, Sivasankaran
author_facet Cangi, Attila
Fiedler, Lenz
Brzoza, Bartosz
Shah, Karan
Callow, Timothy J.
Kotik, Daniel
Schmerler, Steve
Barry, Matthew C.
Goff, James M.
Rohskopf, Andrew
Vogel, Dayton J.
Modine, Normand
Thompson, Aidan P.
Rajamanickam, Sivasankaran
contents We present the Materials Learning Algorithms (MALA) package, a scalable machine learning framework designed to accelerate density functional theory (DFT) calculations suitable for large-scale atomistic simulations. Using local descriptors of the atomic environment, MALA models efficiently predict key electronic observables, including local density of states, electronic density, density of states, and total energy. The package integrates data sampling, model training and scalable inference into a unified library, while ensuring compatibility with standard DFT and molecular dynamics codes. We demonstrate MALA's capabilities with examples including boron clusters, aluminum across its solid-liquid phase boundary, and predicting the electronic structure of a stacking fault in a large beryllium slab. Scaling analyses reveal MALA's computational efficiency and identify bottlenecks for future optimization. With its ability to model electronic structures at scales far beyond standard DFT, MALA is well suited for modeling complex material systems, making it a versatile tool for advanced materials research.
format Preprint
id arxiv_https___arxiv_org_abs_2411_19617
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Materials Learning Algorithms (MALA): Scalable Machine Learning for Electronic Structure Calculations in Large-Scale Atomistic Simulations
Cangi, Attila
Fiedler, Lenz
Brzoza, Bartosz
Shah, Karan
Callow, Timothy J.
Kotik, Daniel
Schmerler, Steve
Barry, Matthew C.
Goff, James M.
Rohskopf, Andrew
Vogel, Dayton J.
Modine, Normand
Thompson, Aidan P.
Rajamanickam, Sivasankaran
Materials Science
Machine Learning
We present the Materials Learning Algorithms (MALA) package, a scalable machine learning framework designed to accelerate density functional theory (DFT) calculations suitable for large-scale atomistic simulations. Using local descriptors of the atomic environment, MALA models efficiently predict key electronic observables, including local density of states, electronic density, density of states, and total energy. The package integrates data sampling, model training and scalable inference into a unified library, while ensuring compatibility with standard DFT and molecular dynamics codes. We demonstrate MALA's capabilities with examples including boron clusters, aluminum across its solid-liquid phase boundary, and predicting the electronic structure of a stacking fault in a large beryllium slab. Scaling analyses reveal MALA's computational efficiency and identify bottlenecks for future optimization. With its ability to model electronic structures at scales far beyond standard DFT, MALA is well suited for modeling complex material systems, making it a versatile tool for advanced materials research.
title Materials Learning Algorithms (MALA): Scalable Machine Learning for Electronic Structure Calculations in Large-Scale Atomistic Simulations
topic Materials Science
Machine Learning
url https://arxiv.org/abs/2411.19617