Machine learning assisted empirical formula augmentation

Abstract

Tuning the martensite transformation temperature through composition design has become an important way to broaden the applicable temperature range of shape memory alloys (SMAs). The empirical formula based on traditional statistics is a key reference for composition design. Due to the lack of experimental data, a large deviation may exist among the prediction results from the empirical formulas obtained by different data sources. In present work, we proposed an augmentation strategy of empirical formula based on a machine learning method to build the relationship between martensite transformation start temperature (Ms) and compositions in Cu-Al-based SMA system. A series of ML models were established by physical and chemical features and a Gaussian radial basis kernel function support vector machine (SVR.rbf) model was screened out based on mathematical and domain knowledge criteria. An augmented empirical formula of Ms as the function of compositions was fitted based on the abundant augmented dataset combined experimental data with predicted data by the SVR.rbf model. Compared with previous empirical formula fitted by small experimental dataset, the accuracy and robustness of the augmented empirical formula was significantly improved without additional experimental cost. This strategy offers a recipe to build empirical formula based on a small experimental dataset.