AtomAgents: Alloy design and discovery through physics-aware multi-modal multi-agent artificial intelligence
A. Ghafarollahi, M.J. Buehler*
MIT
The design of new alloys is a multi-scale problem that requires a holistic approach that involves retrieving relevant knowledge, applying advanced computational methods, conducting experimental validations, and analyzing the results, a process that is typically slow and reserved for human experts. Machine learning (ML) can help accelerate this process, for instance, through the use of deep surrogate models that connect structural and chemical features to material properties, or \textit{vice versa}. However, existing data-driven models often target specific material objectives, offering limited flexibility to integrate out-of-domain knowledge and cannot adapt to new, unforeseen challenges.
Here, we overcome these limitations by leveraging the distinct capabilities of multiple AI agents that collaborate autonomously within a dynamic environment to solve complex materials design tasks. The proposed physics-aware generative AI platform, AtomAgents, synergizes the intelligence of large language models (LLM) the dynamic collaboration among AI agents with expertise in various domains, including knowledge retrieval, multi-modal data integration, physics-based simulations, and comprehensive results analysis across modalities that includes numerical data and images of physical simulation results.
The concerted effort of the multi-agent system allows for addressing complex materials design problems, as demonstrated by examples that include autonomously designing metallic alloys with enhanced properties compared to their pure counterparts. Our results enable accurate prediction of key characteristics across alloys and highlight the crucial role of solid solution alloying to steer the development of advanced metallic alloys. Our framework enhances the efficiency of complex multi-objective design tasks and opens new avenues in fields such as biomedical materials engineering, renewable energy, and environmental sustainability.
Figure 1: Overview of the model and approach.
This repository contains the codes to solve complex alloy design and analysis problems using AtomAgents, an LLM-based multi-agent framework. The file named AtomAgents.ipynb is the main file to present your query as text input. The files named AtomAgents_exp_2 and AtomAgents_exp_3 in the repository, are two examples corresponding to the experiments II, and III, in the corresponding paper, respectively.
The current version supports the following atomistic simulations (in unary and binary systems):
- Lattice constant, elastic constants, and surface energy calculations in FCC and BCC materials
- Create a 1/2<111> screw dislocation in BCC materials
- Perform nudged elastic band (NEB) simulations to compute the Peierls barrier against 1/2<111> screw dislocation in BCC materials
- OpenAI API key is required to run the codes and must be provided in the config_list file.
- To utilize this code, LAMMPS must be compiled with Python support. For more information, visit the LAMMPS documentation.
- The interatomic potential files must be provided in the potential_repository directory.
Figure 1: Example result, showing an overview of the multi-agent work to solve a multi-scale alloy design task.
Please see the paper on arXiv for further details.
@article{ghafarollahi2024atomagentsalloydesigndiscovery,
title={AtomAgents: Alloy design and discovery through physics-aware multi-modal multi-agent artificial intelligence},
author={Alireza Ghafarollahi and Markus J. Buehler},
year={2024},
eprint={2407.10022},
archivePrefix={arXiv},
primaryClass={cs.AI},
url={https://arxiv.org/abs/2407.10022},
month={July}
}