Skip to main content
Log in

Predictions and mechanism analyses of the fatigue strength of steel based on machine learning

  • Metals & corrosion
  • Published:
Journal of Materials Science Aims and scope Submit manuscript

Abstract

It is not completely understood fatigue strength at this time due to its complex formation mechanism. Therefore, in order to address this issue, machine learning has been used to examine the important factors involved in predicting fatigue strength. In this study, a hybrid model was proposed based on the modified bagging method by combining XGBoost and LightGBM, in which the hyperparameters of the models were optimized by a grey wolf algorithm. Moreover, an interpretable method, referred to as Shapley additive explanations (SHAP), was introduced to explain the fatigue strength predictions made by ML models. The SHAP values were calculated, and feature importance of fatigue strength by XGBoost, LightGBM and the hybrid model was discussed. The final results demonstrated that the SHAP method had major potential for interpreting fatigue strength predictions, which would provide constructive guidance for the development of antifatigue steel material in the future.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Figure 1
Figure 2
Figure  3
Figure 4
Figure 5
Figure 6
Figure 7
Figure 8
Figure 9
Figure 10
Figure 11
Figure 12

Similar content being viewed by others

References

  1. Jose R, Ramakrishna S (2018) Materials 4.0: materials big data enabled materials discovery. Appl Mater Today 10:127–132. https://doi.org/10.1016/j.apmt.2017.12.015

    Article  Google Scholar 

  2. Agrawal A, Choudhary A (2018) An online tool for predicting fatigue strength of steel alloys based on ensemble data mining. Int J Fatigue 113:389–400. https://doi.org/10.1016/j.ijfatigue.2018.04.017

    Article  CAS  Google Scholar 

  3. Seshadri R, Sparks TD (2016) Perspective: interactive material property databases through aggregation of literature data. APL Mater 4:053206. https://doi.org/10.1063/1.4944682

    Article  CAS  Google Scholar 

  4. Sukumar N, Krein M, Luo Q, Breneman C (2012) MQSPR modeling in materials informatics: a way to shorten design cycles? J Mater Sci 47:7703–7715. https://doi.org/10.1007/s10853-012-6639-0

    Article  CAS  Google Scholar 

  5. Pilania G, Liu XY (2018) Machine learning properties of binary wurtzite superlattices. J Mater Sci 53:6652–6664. https://doi.org/10.1007/s10853-018-1987-z

    Article  CAS  Google Scholar 

  6. Sun YT, Bai HY, Li MZ, Wang WH (2017) Machine learning approach for prediction and understanding of glass-forming ability. J Phys Chem Lett 8:3434–3439. https://doi.org/10.1021/acs.jpclett.7b01046

    Article  CAS  Google Scholar 

  7. Feng S, Zhou H, Dong H (2019) Using deep neural network with small dataset to predict material defects. Mater Des 162:300–310. https://doi.org/10.1016/j.matdes.2018.11.060

    Article  Google Scholar 

  8. Ward L, Agrawal A, Choudhary A, Wolverton C (2016) A general-purpose machine learning framework for predicting properties of inorganic materials. Npj Comput Mater 2:1–7. https://doi.org/10.1038/npjcompumats.2016.28

    Article  Google Scholar 

  9. Sendek AD, Cubuk ED, Antoniuk ER, Cheon G, Cui Y, Reed EJ (2018) Machine learning-assisted discovery of solid li-ion conducting materials. Chem Mater 31:342–352. https://doi.org/10.1021/acs.chemmater.8b03272

    Article  CAS  Google Scholar 

  10. Sokolov AN, Atahan-Evrenk S, Mondal R, Akkerman HB, Sánchez-Carrera RS, Granados-Focil S et al (2011) From computational discovery to experimental characterization of a high hole mobility organic crystal. Nat Commun 2:1–8. https://doi.org/10.1038/ncomms1451

    Article  CAS  Google Scholar 

  11. Hachmann J, Olivares-Amaya R, Jinich A, Appleton AL, Blood-Forsythe MA, Seress LR et al (2014) Lead candidates for high-performance organic photovoltaics from high-throughput quantum chemistry-the Harvard clean energy project. Energy Environ Sci 7:698–704. https://doi.org/10.1039/c3ee42756k

    Article  CAS  Google Scholar 

  12. Gomez IJ, Arnaiz B, Cacioppo M, Arcudi F, Prato M (2018) Nitrogen-doped Carbon Nanodots for bioimaging and delivery of paclitaxel. J Mater Chem B 6:5540–5548. https://doi.org/10.1039/x0xx00000x

    Article  CAS  Google Scholar 

  13. Hautier G, Fischer CC, Jain A, Mueller T, Ceder G (2010) Finding natures missing ternary oxide compounds using machine learning and density functional theory. Chem Mater 22:3762–3767. https://doi.org/10.1021/cm100795d

    Article  CAS  Google Scholar 

  14. Kreethi R, Sivateja C, Mondal AK, Dutta K (2019) Ratcheting life prediction of quenched–tempered 42CrMo4 steel. J Mater Sci 54:11703–11712. https://doi.org/10.1007/s10853-019-03705-3

    Article  CAS  Google Scholar 

  15. Nikitin I, Fedoseeva A, Kaibyshev R (2020) Strengthening mechanisms of creep-resistant 12%Cr–3%Co steel with low N and high B contents. J Mater Sci 55:7530–7545. https://doi.org/10.1007/s10853-020-04508-7

    Article  CAS  Google Scholar 

  16. Ye XW, Su YH, Han JP (2014) A state-of-the-art review on fatigue life assessment of steel bridges. Math Probl Eng 2014:1–13. https://doi.org/10.1155/2014/956473

  17. Chen X, Sun Y, Wu Z, Yao L, Zhang Y, Zhou S et al (2019) An investigation on residual strength and failure probability prediction for plain weave composite under random fatigue loading. Int J Fatigue 120:267–282. https://doi.org/10.1016/j.ijfatigue.2018.10.013

    Article  CAS  Google Scholar 

  18. Wang Y, Chai P, Ma H, Cao X, Zhang Y (2020) Formation mechanism and fracture behavior in extra-filling refill friction stir spot weld for Al–Cu–Mg aluminum alloy. J Mater Sci 55:358–374. https://doi.org/10.1007/s10853-019-03940-8

    Article  CAS  Google Scholar 

  19. Furuya Y, Hirukawa H, Takeuchi E (2019) Gigacycle fatigue in high strength steels. Sci Technol Adv Mater 20:643–656. https://doi.org/10.1080/14686996.2019.1610904

    Article  CAS  Google Scholar 

  20. Liu T, Zhu X, Chen L, Bi H, Lin Y, Long J (2020) Fatigue–creep behavior of two ferritic stainless steels in simulated automotive exhaust gas and argon. J Mater Sci 55:3684–3699. https://doi.org/10.1007/s10853-019-04233-w

    Article  CAS  Google Scholar 

  21. Zhang M, Sun CN, Zhang X, Goh PC, Wei J, Hardacre D et al (2019) High cycle fatigue life prediction of laser additive manufactured stainless steel: a machine learning approach. Int J Fatigue 128:105194. https://doi.org/10.1016/j.ijfatigue.2019.105194

    Article  CAS  Google Scholar 

  22. Wan HY, Chen GF, Li CP, Qi XB, Zhang GP (2019) Data-driven evaluation of fatigue performance of additive manufactured parts using miniature specimens. J Mater Sci Technol 35:1137–1146. https://doi.org/10.1016/j.jmst.2018.12.011

    Article  Google Scholar 

  23. Kobayashi S (2019) Low-angle boundary engineering for improving high- cycle fatigue property of 430 ferritic stainless steel. J Mater Sci 55:1–13. https://doi.org/10.1007/s10853-020-04555-0

    Article  CAS  Google Scholar 

  24. Rathi S (2019) Generating counterfactual and contrastive explanations using SHAP.  arXiv:1906.09293

  25. Mirjalili S, Mirjalili SM, Lewis A (2014) Grey wolf optimizer. Adv Eng Softw 69:46–61. https://doi.org/10.1016/j.advengsoft.2013.12.007

    Article  Google Scholar 

  26. Chen T, Guestrin C. XGBoost: A scalable tree boosting system. In: Proc ACM SIGKDD Int Conf Knowl Discov Data Min 2016:785–94. https://doi.org/10.1145/2939672.2939785

  27. Ke G, Meng Q, Finley T, Wang T, Chen W, Ma W et al (2017) LightGBM: a highly efficient gradient boosting decision tree. Adv Neural Inf Process. Syst. 2017:3147–3155

    Google Scholar 

  28. Zapotecas-Martínez S, García-Nájera A, López-Jaimes A (2019) Multi-objective grey wolf optimizer based on decomposition. Expert Syst Appl 120:357–371. https://doi.org/10.1016/j.eswa.2018.12.003

    Article  Google Scholar 

  29. Holland JH (1992) Genetic algorithms. Sci Am 267:66–72. https://doi.org/10.1038/scientificamerican0792-66

    Article  Google Scholar 

  30. Tian D, Shi Z (2018) MPSO: modified particle swarm optimization and its applications. Swarm Evol Comput 41:49–68. https://doi.org/10.1016/j.swevo.2018.01.011

    Article  Google Scholar 

  31. Wu Y, Ma W, Miao Q, Wang S (2019) Multimodal continuous ant colony optimization for multisensor remote sensing image registration with local search. Swarm Evol Comput 47:89–95. https://doi.org/10.1016/j.swevo.2017.07.004

    Article  Google Scholar 

  32. Choong SS, Wong LP, Lim CP (2019) An artificial bee colony algorithm with a modified choice function for the traveling salesman problem. Swarm Evol Comput 44:622–635. https://doi.org/10.1016/j.swevo.2018.08.004

    Article  Google Scholar 

  33. Breiman LP (1996) Bagging Predictors. Mach Learn 24:123–140. https://doi.org/10.1007/BF00058655

    Article  Google Scholar 

  34. Choi DK (2019) Data-driven materials modeling with XGBoost algorithm and statistical inference analysis for prediction of fatigue strength of steels. Int J Precis Eng Manuf 20:129–138. https://doi.org/10.1007/s12541-019-00048-6

    Article  Google Scholar 

  35. Fan J, Wu L, Zhang F, Cai H, Ma X, Bai H (2019) Evaluation and development of empirical models for estimating daily and monthly mean daily diffuse horizontal solar radiation for different climatic regions of China. Renew Sustain Energy Rev 105:168–186. https://doi.org/10.1016/j.rser.2019.01.040

    Article  Google Scholar 

  36. Papadopoulos S, Kontokosta CE (2019) Grading buildings on energy performance using city benchmarking data. Appl Energy 233:244–253. https://doi.org/10.1016/j.apenergy.2018.10.053

    Article  Google Scholar 

  37. Lundberg SM, Erion GG, Lee S-I (2018) Consistent individualized feature attribution for tree ensembles.  arXiv:1802.03888

  38. Lundberg SM, Lee SI (2017) A unified approach to interpreting model predictions. In: Adv. Neural Inf. Process. Syst., 2017, pp. 4766–75

  39. Lim S, Chi S (2019) Xgboost application on bridge management systems for proactive damage estimation. Adv Eng Inform 41:100922. https://doi.org/10.1016/j.aei.2019.100922

    Article  Google Scholar 

Download references

Acknowledgements

This work was supported by the project of the National Key R&D Program “Joint Research on Advanced Technology and Application of Electric Vehicles based on China-US Cooperation” (2016YFE0102200). It was also supported by the National Natural Science Foundation of China (grant number U1864207). The authors also sincerely gave thanks to the support of Technology Innovation and Entrepreneurship Team of Hunan Province

Author information

Authors and Affiliations

Authors

Contributions

FY was involved in methodology, writing—original draft preparation, and software. KS contributed to data curation, conceptualization, and formal analysis. YL helped in visualization and investigation. SC was involved in writing—reviewing and editing, supervision, and validation. JC contributed to resources, project administration, and funding acquisition.

Corresponding author

Correspondence to Kai Song.

Ethics declarations

Conflict of interest

The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Yan, F., Song, K., Liu, Y. et al. Predictions and mechanism analyses of the fatigue strength of steel based on machine learning. J Mater Sci 55, 15334–15349 (2020). https://doi.org/10.1007/s10853-020-05091-7

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10853-020-05091-7

Navigation