arrow
Return

Composition and structure analyzer/featurizer for explainable machine-learning models to predict solid state structures

delete2025-01-01
delete1
delete
OA
AI
E
Emil I. Jaffal
S
Sangjoon Lee *
D
Danila Shiryaev
A
Alex Vtorov
N
Nikhil K. Barua
H
Holger Kleinke
A
Anton O. Oliynyk *
DOI:10.1039/d4dd00332bdelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Traditional and non-classical machine learning models for solid-state structure prediction have predominantly relied on compositional features (derived from properties of constituent elements) to predict the existence of a structure and its properties. However, the lack of structural information can be a source of suboptimal property mapping and increased predictive uncertainty. To address this challenge, we have introduced a strategy that generates and combines both compositional and structural features with minimal programming expertise required. Our approach utilizes open-source, interactive Python programs named Composition Analyzer Featurizer (CAF) and Structure Analyzer Featurizer (SAF). CAF generates numerical compositional features from a list of formulae provided in an Excel file, while SAF extracts numerical structural features from a .cif file by generating a supercell. 133 features from CAF and 94 features from SAF are used either individually or in combination to cluster nine structure types in equiatomic AB intermetallics. The performance is comparable to those with features from JARVIS, MAGPIE, mat2vec, and OLED datasets in PLS-DA, SVM, and XGBoost models. Our SAF + CAF features provide a cost-efficient and reliable solution, even with the PLS-DA method, where a significant fraction of the most contributing features is the same as those identified in the more computationally intensive XGBoost models.
Keywords:
BAND-GAP PREDICTION
FEATURE-SELECTION
SCALE
ELECTRONEGATIVITY
REGRESSION
AB

Journal

Digital Discovery cover
Digital Discovery
IF:
5.6
Papers:
979
Citations:
1.7K

Organization

C
Columbia University
Scholars:
7.1W
Papers: 6.4W
Citations: 263
C
city university of new york (cuny) system
Scholars:
1.6W
Papers: 1.5W
Citations: 26