投审稿入口

基于联邦学习的热轧带钢力学性能预报

Prediction of mechanical properties of hot-rolled strip steel based on federated learning

  • 摘要: 热轧带钢作为工业领域的关键材料,其力学性能预报精度对下游制造质量具有直接影响。然而,在实际应用中,受限于数据样本量不足,模型的预测性能难以提升;若考虑多产线或多工厂协同建模,又会面临数据异构性与隐私保护的双重挑战。针对上述问题,本文提出一种基于联邦学习的热轧带钢力学性能预报方法。该方法首先完成特征维度对齐,在确保数据隐私安全的前提下,实现对热轧带钢抗拉强度、屈服强度及伸长率的多方协同预测,并将其结果与基于单产线数据训练的模型预测结果展开对比。实验结果表明,该方法在各项力学性能预报任务中均表现出良好性能。此外,为进一步优化方案,本文引入基于特征重要性的联邦维数优化方法,协同筛选出对力学性能影响显著的关键因素。

     

    Abstract: As a critical material in industry, hot-rolled strip steel has a direct impact on downstream manufacturing quality due to the accuracy of its mechanical property prediction. However, in practical applications, the predictive performance of models is difficult to improve due to insufficient data sample sizes. When considering collaborative modeling across multiple production lines or factories, dual challenges of data heterogeneity and privacy protection arise. To address these issues, this paper proposes a method for predicting the mechanical properties of hot-rolled strip based on federated learning. The method first achieves feature dimension alignment and enables multi-party collaborative prediction of tensile strength, yield strength, and elongation of hot-rolled strip under the premise of ensuring data privacy and security. Its results are compared with those of models trained on single production line data. Experimental results demonstrate that this method exhibits good performance in all mechanical property prediction tasks. Additionally, to further optimize the scheme, this paper introduces a federated dimension optimization method based on feature importance to collaboratively screen out key factors that significantly affect mechanical properties.

     

/

返回文章
返回