Abstract
In this paper, a new machine learning (ML) model combining conditional generative adversarial networks (CGANs) and active learning (AL) is proposed to predict the body-centered cubic (BCC) phase, face-centered cubic (FCC) phase, and BCC+FCC phase of high-entropy alloys (HEAs). Considering the lack of data, CGANs are introduced for data augmentation, and AL can achieve high prediction accuracy under a small sample size owing to its special sample selection strategy. Therefore, we propose an ML framework combining CGAN and AL to predict the phase of HEAs. The arithmetic optimization algorithm (AOA) is introduced to improve the artificial neural network (ANN). AOA can overcome the problem of falling into the locally optimal solution for the ANN and reduce the number of training iterations. The AOA-optimized ANN model trained by the AL sample selection strategy achieved high prediction accuracy on the test set. To improve the performance and interpretability of the model, domain knowledge is incorporated into the feature selection. Additionally, considering that the proposed method can alleviate the problem caused by the shortage of experimental data, it can be applied to predictions based on small datasets in other fields.
References
Yeh J W, Chen S K, Lin S J, et al. Nanostructured high-entropy alloys with multiple principal elements: Novel alloy design concepts and outcomes. Adv Eng Mater, 2004, 6: 299–303
Wang J, Li J, Wang Q, et al. The incredible excess entropy in high entropy alloys. Scripta Mater, 2019, 168: 19–22
Chen C, Ma L, Zhang Y, et al. Accelerating the design of high-entropy alloys with high hardness by machine learning based on particle swarm optimization. Intermetallics, 2023, 154: 107819
Guo X, Xie X, Ren J, et al. Plastic dynamics of the Al0.5CoCrCuFeNi high entropy alloy at cryogenic temperatures: Jerky flow, stair-like fluctuation, scaling behavior, and non-chaotic state. Appl Phys Lett, 2017, 111: 251905
Wu L, Wei G, Wang G, et al. Creating win-wins from strength-ductility trade-off in multi-principal element alloys by machine learning. Mater Today Commun, 2022, 32: 104010
Ding Z Y, He Q F, Yang Y. Exploring the design of eutectic or near-eutectic multicomponent alloys: From binary to high entropy alloys. Sci China Tech Sci, 2018, 61: 159–167
Shang X L, Wang Z J, He F, et al. The intrinsic mechanism of corrosion resistance for FCC high entropy alloys. Sci China Tech Sci, 2018, 61: 189–196
Zhao D Q, Pan S P, Zhang Y, et al. Structure prediction in high-entropy alloys with machine learning. Appl Phys Lett, 2021, 118: 231904
Liu Z, Lei Y, Gray C, et al. Examination of solid-solution phase formation rules for high entropy alloys from atomistic Monte Carlo simulations. JOM, 2015, 67: 2364–2374
Niu C, Windl W, Ghazisaeidi M. Multi-cell Monte Carlo relaxation method for predicting phase stability of alloys. Scripta Mater, 2017, 132: 9–12
Niu C, Rao Y, Windl W, et al. Multi-cell Monte Carlo method for phase prediction. npj Comput Mater, 2019, 5: 120
Troparevsky M C, Morris J R, Kent P R C, et al. Criteria for predicting the formation of single-phase high-entropy alloys. Phys Rev X, 2015, 5: 011041
Wang W Y, Shang S L, Wang Y, et al. Atomic and electronic basis for the serrations of refractory high-entropy alloys. NPJ Comput Mater, 2017, 3: 1–9
Senkov O N, Miller J D, Miracle D B, et al. Accelerated exploration of multi-principal element alloys with solid solution phases. Nat Commun, 2015, 6: 6529
Choi W M, Jung S, Jo Y H, et al. Design of new face-centered cubic high entropy alloys by thermodynamic calculation. Met Mater Int, 2017, 23: 839–847
Saal J E, Berglund I S, Sebastian J T, et al. Equilibrium high entropy alloy phase stability from experiments and thermodynamic modeling. Scripta Mater, 2017, 146: 5–8
Senkov O N, Zhang C, Pilchak A L, et al. CALPHAD-aided development of quaternary multi-principal element refractory alloys based on NbTiZr. J Alloys Compd, 2018, 783: 729–742
Jiang C, Uberuaga B P. Efficient ab initio modeling of random multicomponent alloys. Phys Rev Lett, 2016, 116: 105501
Feng R, Liaw P K, Gao M C, et al. First-principles prediction of high-entropy-alloy stability. npj Comput Mater, 2017, 3: 50
Lederer Y, Toher C, Vecchio K S, et al. The search for high entropy alloys: A high-throughput ab-initio approach. Acta Mater, 2018, 159: 364–383
Liu Y, Zhao T, Ju W, et al. Materials discovery and design using machine learning. J Materiomics, 2017, 3: 159–177
Liu Y, Guo B, Zou X, et al. Machine learning assisted materials design and discovery for rechargeable batteries. Energy Stor Mater, 2020, 31: 434–450
Zhang H, Fu H, Zhu S, et al. Machine learning assisted composition effective design for precipitation strengthened copper alloys. Acta Mater, 2021, 215: 117118
Li Y, Guo W. Machine-learning model for predicting phase formations of high-entropy alloys. Phys Rev Mater, 2019, 3: 095005
Pei Z, Yin J, Hawk J A, et al. Machine-learning informed prediction of high-entropy solid solution formation: Beyond the hume-rothery rules. NPJ Comput Mater, 2020, 6: 1–8
Zhang Y, Wen C, Wang C, et al. Phase prediction in high entropy alloys with a rational selection of materials descriptors and machine learning models. Acta Mater, 2020, 185: 528–539
Machaka R. Machine learning-based prediction of phases in high-entropy alloys. Comput Mater Sci, 2021, 188: 110244
Qu N, Liu Y, Zhang Y, et al. Machine learning guided phase formation prediction of high entropy alloys. Mater Today Commun, 2022, 32: 104146
Li Z, Nash W T, O’Brien S P, et al. cardiGAN: A generative adversarial network model for design and discovery of multi principal element alloys. J Mater Sci Tech, 2022, 125: 81–96
Chang H, Tao Y, Liaw P K, et al. Phase prediction and effect of intrinsic residual strain on phase stability in high-entropy alloys with machine learning. J Alloys Compd, 2022, 921: 166149
Rao Z, Tung P Y, Xie R, et al. Machine learning-enabled high-entropy alloy discovery. Science, 2022, 378: 78–85
Goodfellow I, Pouget-Abadie J, Mirza M, et al. Generative adversarial nets. In: Proceedings of the 27th International Conference on Neural Information Processing Systems. Montreal, 2014, 2672–2680
Mirza M, Osindero S, Conditional generative adversarial nets. arXiv: 1411.1784
Settles B. Active Learning Literature Survey. Technical Report. Madison: University of Wisconsin, 2010
Abualigah L, Diabat A, Mirjalili S, et al. The arithmetic optimization algorithm. Comput Methods Appl Mech Eng, 2021, 376: 113609
Miracle D B, Senkov O N. A critical review ofhigh entropy alloys and related concepts. Acta Mater, 2017, 122: 448–511
Ye Y F, Wang Q, Lu J, et al. High-entropy alloy: Challenges and prospects. Mater Today, 2016, 19: 349–362
Couzinié J P, Senkov O N, Miracle D B, et al. Comprehensive data compilation on the mechanical properties of refractory high-entropy alloys. Data brief, 2018, 21: 1622–1641
Gorsse S, Nguyen M H, Senkov O N, et al. Database on the mechanical properties of high entropy alloys and complex concentrated alloys. Data Brief, 2018, 21: 2664–2678
Yang X, Zhang Y. Prediction of high-entropy stabilized solid-solution in multi-component alloys. Mater Chem Phys, 2012, 132: 233–238
Liu Y, Wu J, Avdeev M, et al. Multi-layer feature selection incorporating weighted score-based expert knowledge toward modeling materials with targeted properties. Adv Theor Simul, 2020, 3: 1900215
Liu Y, Zou X, Ma S, et al. Feature selection method reducing correlations among features by embedding domain knowledge. Acta Mater, 2022, 238: 118195
Wu H, Xie J, Yang H, et al. A cost-effective eutectic high entropy alloy with an excellent strength-ductility combination designed by VEC criterion. J Mater Sci Technol, 2022, 19: 1759–1765
Juan Y, Zhang J, Dai Y, et al. Designing rules of laser-clad high-entropy alloy coatings with simple solid solution phases. Acta Metall Sin (Engl Lett), 2020, 33: 1064–1076
Chanda B, Das J. Composition dependence on the evolution of nanoeutectic in CoCrFeNiNbx (0.45 ⩽ x ⩽ 0.65) high entropy alloys. Adv Eng Mater, 2018, 20: 1700908
Zhang Y, Zhou Y, Lin J, et al. Solid-solution phase formation rules for multi-component alloys. Adv Eng Mater, 2008, 10: 534–538
Xie L, Brault P, Thomann A L, et al. Molecular dynamics simulation of Al-Co-Cr-Cu-Fe-Ni high entropy alloy thin film growth. Intermetallics, 2016, 68: 78–86
Troparevsky M C, Morris J R, Daene M, et al. Beyond atomic sizes and hume-rothery rules: Understanding and predicting high-entropy alloys. JOM, 2015, 67: 2350–2363
Yao H W, Qiao J W, Hawk J A, et al. Mechanical properties of refractory high-entropy alloys: Experiments and modeling. J Alloys Compd, 2017, 696: 1139–1150
Si S, Fan B, Liu X, et al. Study on strengthening effects of Zr-Ti-Nb-O alloys via high throughput powder metallurgy and data-driven machine learning. Mater Des, 2021, 206: 109777
Li X, Shan G, Shek C H. Machine learning prediction of magnetic properties of Fe-based metallic glasses considering glass forming ability. J Mater Sci Tech, 2022, 103: 113–120
Hume-Rothery W, Mabbott G W, Evans K M C, The freezing points, melting points, and solid solubility limits of the alloys of silver and copper with the elements of the b sub-groups. Philos Trans Royal Soc A, 1934, 233: 1–97
Guo S, Ng C, Lu J, et al. Effect ofvalence electron concentration on stability of fcc or bcc phase in high entropy alloys. J Appl Phys, 2011, 109: 103505
Otto F, Yang Y, Bei H, et al. Relative effects of enthalpy and entropy on the phase stability of equiatomic high-entropy alloys. Acta Mater, 2013, 61: 2628–2638
Troparevsky M C, Morris J R, Kent P R C, et al. Criteria for predicting the formation of single-phase high-entropy alloys. Phys Rev X, 2015, 5: 011041
Abbaschian R, Reed-Hill R E. Physical Metallurgy Principles. New York: Van Nostrand, 1973
George E P, Raabe D, Ritchie R O. High-entropy alloys. Nat Rev Mater, 2019, 4: 515–534
Salimans T, Goodfellow I, Zaremba W, et al. Improved techniques for training gans. In: Proceedings of the 30th International Conference on Neural Information Processing Systems. Barcelona, 2016
Shannon C E. The mathematical theory of communication. M.D. Comput-Comput Medical Pract, 1997, 14: 306–317
Author information
Authors and Affiliations
Corresponding author
Additional information
This work was supported by the Key Scientific and Technological Project of Henan Province (Grant No. 212102210112), the National Natural Science Foundation of China (Grant No. 52071298), and the Strategic Research and Consulting Project of Chinese Academy of Engineering (Grant No. 2022HENYB05).
Rights and permissions
About this article
Cite this article
Chen, C., Zhou, H., Long, W. et al. Phase prediction for high-entropy alloys using generative adversarial network and active learning based on small datasets. Sci. China Technol. Sci. 66, 3615–3627 (2023). https://doi.org/10.1007/s11431-023-2399-2
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11431-023-2399-2