Heat Treatment of Metals ›› 2025, Vol. 50 ›› Issue (1): 266-271.DOI: 10.13251/j.issn.0254-6051.2025.01.041

• COMPUTER APPLICATION • Previous Articles     Next Articles

Machine learning to predict effect of annealing temperature and time on mechanical properties of SUS321 stainless steel

Wang Huanhuan1,2, Lu Sujun1,2, Li Yuan3, Xu Ning1,2, Zhu Tingxian3, Wang Xu1,2, Wei Ning4, Wu Dali1,2, Peng Weizhong1,2   

  1. 1. National Key Laboratory of Ni & Co Associated Minerals Resources Development and Comprehensive Utilization, Jinchang Gansu 737100, China;
    2. Jinchuan Nickel & Cobalt Research and Engineering Institute, Jinchang Gansu 737100, China;
    3. Jinchuan Group Nickel Alloy Co., Ltd., Jinchang Gansu 737100, China;
    4. Jinchuan Group Information and Automation Engineering Co., Ltd., Jinchang Gansu 737100, China
  • Received:2024-07-29 Revised:2024-11-12 Online:2025-01-25 Published:2025-03-12

Abstract: In order to reveal the complex relationship between mechanical properties and heat treatment process of SUS321 stainless steel, a prediction model between annealing temperature, annealing time and mechanical properties of the SUS321 stainless steel was established by machine learning method based on random forest model. The results show that the prediction validity parameter R2 of the model for the tensile strength, yield strength and elongation of the SUS321 stainless steel exceeds 0.8, showing a good prediction effect. The analysis based on partial dependence plots and individual conditional expectation plots shows that with the increase of annealing temperature and annealing time, the tensile strength and yield strength of the SUS321 stainless steel decrease, while the elongation increases, and the annealing temperature is the main feature parameter.

Key words: SUS321 stainless steel, annealing treatment, mechanical properties, machine learning, model interpretation

CLC Number: