pysipfenn 0.11.0

Easily extensible Python package for running Structure-Informed Prediction of Formation Energy using Neural Networks (SIPFENN)

pySIPFENN

Summary

This repository contains py(Structure-Informed Prediction of Formation Energy using Neural Networks) software package allowing efficient predictions of the energetics of atomic configurations. The underlying methodology and implementation is given in

• Adam M. Krajewski, Jonathan W. Siegel, Jinchao Xu, Zi-Kui Liu, Extensible Structure-Informed Prediction of Formation Energy with improved accuracy and usability employing neural networks, Computational Materials Science, Volume 208, 2022, 111254 https://doi.org/10.1016/j.commatsci.2022.111254

While functionalities are similar to the software released along the paper, this package contains improved methods for featurizing atomic configurations. Notably, all of them are now written completely in Python, removing reliance on Java and making extensions of the software much easier thanks to improved readability. A fuller description of capabilities is given in documentation at https://pysipfenn.org and at PSU Phases Research Lab webpage under https://phaseslab.com/sipfenn.

Applications

pySIPFENN in a very flexible tool that can, in principle be used, for prediction of any property of interest that depends on an atomic configuration, with very little modifications. The models shipped by default are trained to predict formation energy because that is what our research group is interested in; however, if one wanted to predict Poisson's ratio and trained a model based on the same features, adding it would take minutes. Simply add the model in open ONNX format and link it using models.json file, as described in documentation.

Real-World Examples

In oru line of work, pySIPFENN and formation energies it predicts are usually used as a computational engine that generates proto-data for creation of thermodynamic databases (TDBs) using ESPEI (https://espei.org). The TDBs are then used through pycalphad (https://pycalphad.org) to predict phase diagrams and other thermodynamic properties.

Another of its uses in our research is guiding the Density Functional Theory (DFT) calculations as a low-cost screening tool. Their efficient conjunction then drives experiments leading to discovery of new materials as presented in these two papers:

• Sanghyeok Im, Shun-Li Shang, Nathan D. Smith, Adam M. Krajewski, Timothy Lichtenstein, Hui Sun, Brandon J. Bocklund, Zi-Kui Liu, Hojong Kim, Thermodynamic properties of the Nd-Bi system via emf measurements, DFT calculations, machine learning, and CALPHAD modeling, Acta Materialia, Volume 223, 2022, 117448, https://doi.org/10.1016/j.actamat.2021.117448.

• Shun-Li Shang, Hui Sun, Bo Pan, Yi Wang, Adam M. Krajewski, Mihaela Banu, Jingjing Li & Zi-Kui Liu, Forming mechanism of equilibrium and non-equilibrium metallurgical phases in dissimilar aluminum/steel (Al–Fe) joints. Nature Scientific Reports 11, 24251 (2021). https://doi.org/10.1038/s41598-021-03578-0

Install pySIPFENN

Installing pySIPFENN is simple and easy utilizing either PyPI package repository or cloning from GitHub. While not required, it is recommended to first set up a virtual environment using venv or Conda. This ensures that one of the required versions of Python (3.9+) is used and there are no dependency conflicts. If you have Conda installed on your system (see instructions at https://docs.conda.io/en/latest/miniconda.html), you can create a new environment with:

conda create -n pysipfenn-workshop python=3.9 jupyter
conda activate pysipfenn-workshop

And then simply install pySIPFENN from PyPI with

pip install pysipfenn

Alternatively, you can also install pySIPFENN in editable mode if you cloned it from GitHub like

git clone https://github.com/PhasesResearchLab/pySIPFENN.git

Or by downloading a ZIP file. Please note, this will by default download the latest development version of the software, which may not be stable. For a stable version, you can specify a version tag after the URL with --branch <tag_name> --single-branch.

Then, move to the pySIPFENN folder and install in editable (-e) mode

cd pySIPFENN
pip install -e .