Tensile strength prediction of steel sheets: An insight into data-driven models, dimensionality reduction, and feature importance
Research output: Contribution to journal › Article › Research › peer-review
Authors
Organisational units
External Organisational units
- Materials Center Leoben Forschungs GmbH
Abstract
In this work we apply data-driven models for predicting tensile strength of steel coils from chemical composition and process parameters. The data originates from steel production and includes a full chemical analysis, as well as many process parameters and the resulting strength properties from tensile tests. We establish a data pre-processing pipeline, where we apply data cleaning and feature engineering to create a machine-readable dataset suitable for various modeling tasks. We compare prediction quality, complexity and interpretability of pure machine learning (ML) models, either with the full feature set or a reduced one. Dimensionality reduction methods are used to reduce the number of features and therefore reduce complexity, either with a smart selection method or feature encoding, where features are combined and the included information is preserved. In order to determine key features of our models, we are investigating feature importance ratings, which can be used as a feature selection criteria. Furthermore, we are highlighting methods to explain predictions and determine the impact of every feature in every observation applicable for any ML model.
Details
Original language | English |
---|---|
Article number | 075004 |
Number of pages | 23 |
Journal | Modelling and simulation in materials science and engineering |
Volume | 32.2024 |
Issue number | 7 |
DOIs | |
Publication status | Published - 24 Aug 2024 |