The research presents a hybrid approach of regression modeling with data-driven analysis for predicting steel's mechanical properties by analyzing the effects of composition on strength. The study fills the gap of models in accurately predicting steel's performance based on composition since traditional methods cannot fully capture complex relationships between alloying elements and material properties. Various regression models have been used for predicting material properties, such as Linear Regression, Random Forest Regression, Support Vector Regression (SVR), XGBoost Regression, and Neural Networks, and in this paper, Graph Attention Transformer Network (GAT-TransNet) is proposed. Incorporating novel graph attention into the transformer architecture model, GAT-TransNet handles complex data relationships and improves predictive accuracy. Data-driven analyses are also carried out alongside regression analysis to establish how alloying elements, such as carbon (C), manganese (Mn), and chromium (Cr), affect steel's mechanical properties strength, yield strength, hardness, and ductility. The study established that the GAT-TransNet model outperformed other regression models, with an R² score of 0.95, the lowest MAE of 1.40, and an MSE of 4.41, thus underscoring its superior predictive capability compared to existing models. Data-driven insights show that manganese hardens and increases wear resistance, while chromium enhances corrosion resistance and increases tensile strength. This has great importance for optimizing specific steel compositions for industrial applications. Combining machine learning methodologies with composition analysis, this study complements predictive modeling for steel properties with material design and promises better efficiency and targeting in steel production.
X. Dong et al., “Heterostructured Metallic Structural Materials: Research Methods, Properties, and Future Perspectives,” Advanced Functional Materials, vol. 34, no. 51, Aug. 2024, doi: 10.1002/adfm.202410521.
P. Gradl et al., “Robust Metal Additive Manufacturing Process Selection and Development for Aerospace Components,” Journal of Materials Engineering and Performance, vol. 31, no. 8, pp. 6013–6044, Apr. 2022, doi: 10.1007/s11665-022-06850-0.
R. Kumpati, W. Skarka, and S. K. Ontipuli, “Current Trends in Integration of Nondestructive Testing Methods for Engineered Materials Testing,” Sensors, vol. 21, no. 18, p. 6175, Sep. 2021, doi: 10.3390/s21186175.
P. Demircioglu, M. Seckin, A. C. Seckin, and I. Bogrekci, “Non-destructive Testing Methods in Composite Materials,” Fracture Behavior of Nanocomposites and Reinforced Laminate Structures, pp. 487–516, 2024, doi: 10.1007/978-3-031-68694-8_21.
R. Pollice et al., “Data-Driven Strategies for Accelerated Materials Design,” Accounts of Chemical Research, vol. 54, no. 4, pp. 849–860, Feb. 2021, doi: 10.1021/acs.accounts.0c00785.
U. K. R. K, “Design of High Security Smart Health Care Monitoring System using IoT,” International Journal of Emerging Trends in Engineering Research, vol. 8, no. 6, pp. 2259–2265, Jun. 2020, doi: 10.30534/ijeter/2020/09862020.
J. Liang, Z. He, W. Du, X. Ruan, E. Guo, and N. Shen, “Tailoring the microstructure and mechanical properties of laser metal-deposited Hastelloy X superalloy sheets via post heat-treatment,” Materials Science and Engineering: A, vol. 884, p. 145546, Sep. 2023, doi: 10.1016/j.msea.2023.145546.
R. K. Madapudi, A. A. Rao, and G. Merugu, “Change requests artifacts to assess impact on structural design of SDLC phases,” Int’l J. Computer Applications, vol. 54, no. 18, pp. 21–26, 2012.
S. V. Suryanarayana and G. N. Balaji, “Using Computer Vision to enhance Safety in a Post COVID World,” 2022 International Conference on Data Science, Agents & Artificial Intelligence (ICDSAAI), pp. 1–6, Dec. 2022, doi: 10.1109/icdsaai55433.2022.10028919.
M. Sudhakara, M. J. Meena, K. R. Madhavi, P. Anjaiah, and L. N. Prakash, “‘Fish classification using deep learning on small scale and low-quality images,” Int. J. Intell. Syst. Appl. Eng, vol. 10, no. 1s, pp. 282–293, 2022.
J. R. Dwaram and R. K. Madapuri, “Crop yield forecasting by long short‐term memory network with Adam optimizer and Huber loss function in Andhra Pradesh, India,” Concurrency and Computation: Practice and Experience, vol. 34, no. 27, Sep. 2022, doi: 10.1002/cpe.7310.
K. Chen et al., “A review of machine learning in additive manufacturing: design and process,” The International Journal of Advanced Manufacturing Technology, vol. 135, no. 3–4, pp. 1051–1087, Oct. 2024, doi: 10.1007/s00170-024-14543-2.
Z. Xia, B. Wu, C. Y. Chan, T. Wu, M. Zhou, and L. B. Kong, “Deep-learning-based pyramid-transformer for localized porosity analysis of hot-press sintered ceramic paste,” PLOS ONE, vol. 19, no. 9, p. e0306385, Sep. 2024, doi: 10.1371/journal.pone.0306385.
D. Venkata Lakshmi, R. Shyama, S. Anila, S. Abbineni, S. A. Al, and A. Al-Hilali, “An Intelligent Framework for Smart Automated House Implementation via Integration of IOT and DL,” 2024 4th International Conference on Advance Computing and Innovative Technologies in Engineering (ICACITE), pp. 225–228, May 2024, doi: 10.1109/icacite60783.2024.10616423.
M. Ashwin Shenoy and N. Thillaiarasu, “Enhancing temple surveillance through human activity recognition: A novel dataset and YOLOv4-ConvLSTM approach,” Journal of Intelligent & Fuzzy Systems, vol. 45, no. 6, pp. 11217–11232, Dec. 2023, doi: 10.3233/jifs-233919.
C. Sanford et al., “Understanding Transformer Reasoning Capabilities via Graph Algorithms,” Adv Neural Inf Process Syst, vol. 37, pp. 78320–78370, Dec. 2024.
T. PlÖtz, “Applying Machine Learning for Sensor Data Analysis in Interactive Systems,” ACM Computing Surveys, vol. 54, no. 6, pp. 1–25, Jul. 2021, doi: 10.1145/3459666.
A. C. Cheloee Darabi, S. Rastgordani, M. Khoshbin, V. Guski, and S. Schmauder, “Hybrid Data-Driven Deep Learning Framework for Material Mechanical Properties Prediction with the Focus on Dual-Phase Steel Microstructures,” Materials, vol. 16, no. 1, p. 447, Jan. 2023, doi: 10.3390/ma16010447.
B. S. Puttaswamy and N. Thillaiarasu, “Fine DenseNet based human personality recognition using english hand writing of non-native speakers,” Biomedical Signal Processing and Control, vol. 99, p. 106910, Jan. 2025, doi: 10.1016/j.bspc.2024.106910.
L. Li, J. Chang, A. Vakanski, Y. Wang, T. Yao, and M. Xian, “Uncertainty quantification in multivariable regression for material property prediction with Bayesian neural networks,” Scientific Reports, vol. 14, no. 1, May 2024, doi: 10.1038/s41598-024-61189-x.
Z. Cao, R. Magar, Y. Wang, and A. Barati Farimani, “MOFormer: Self-Supervised Transformer Model for Metal–Organic Framework Property Prediction,” Journal of the American Chemical Society, vol. 145, no. 5, pp. 2958–2967, Jan. 2023, doi: 10.1021/jacs.2c11420.
A. Jose, E. Devijver, N. Jakse, and R. Poloni, “Informative Training Data for Efficient Property Prediction in Metal–Organic Frameworks by Active Learning,” Journal of the American Chemical Society, vol. 146, no. 9, pp. 6134–6144, Feb. 2024, doi: 10.1021/jacs.3c13687.
M. Ravi Prasad and N. Thillaiarasu, “Multichannel EfficientNet B7 with attention mechanism using multimodal biometric- based authentication for ATM transaction,” Multiagent and Grid Systems, vol. 20, no. 2, pp. 89–108, Aug. 2024, doi: 10.3233/mgs-230118.
A. Stoll and P. Benner, “Machine learning for material characterization with an application for predicting mechanical properties,” GAMM-Mitteilungen, vol. 44, no. 1, Mar. 2021, doi: 10.1002/gamm.202100003.
Z. H. Wang, Y. F. Liu, T. Wang, J. G. Wang, Y. M. Liu, and Q. X. Huang, “Intelligent prediction model of mechanical properties of ultrathin niobium strips based on XGBoost ensemble learning algorithm,” Computational Materials Science, vol. 231, p. 112579, Jan. 2024, doi: 10.1016/j.commatsci.2023.112579.
M. Justi, M. P. de Freitas, J. M. Silla, C. A. Nunes, and C. A. Silva, “Molecular structure features and fast identification of chemical properties of metal carboxylate complexes by FTIR and partial least square regression,” Journal of Molecular Structure, vol. 1237, p. 130405, Aug. 2021, doi: 10.1016/j.molstruc.2021.130405.
“Steel Strength Data 2018.” Accessed: Mar. 14, 2025. [Online]. Available: https://www.kaggle.com/datasets/fuarresvij/steel-test-data
CRediT Author Statement
The authors confirm contribution to the paper as follows:
Conceptualization: Rudra Kumar M, Rama Vasantha Adiraju, LNC Prakash K, Mahalakshmi V, Penubaka Balaji and Jayavardhanarao Sahukaru;
Methodology: Rudra Kumar M, Rama Vasantha Adiraju and LNC Prakash K;
Writing- Original Draft Preparation: Rudra Kumar M, Rama Vasantha Adiraju, LNC Prakash K, Mahalakshmi V, Penubaka Balaji and Jayavardhanarao Sahukaru;
Writing- Reviewing and Editing: Rudra Kumar M, Rama Vasantha Adiraju, LNC Prakash K, Mahalakshmi V, Penubaka Balaji and Jayavardhanarao Sahukaru;
All authors reviewed the results and approved the final version of the manuscript.
Acknowledgements
Authors thank Reviewers for taking the time and effort necessary to review the manuscript.
Funding
No funding was received to assist with the preparation of this manuscript.
Ethics declarations
Conflict of interest
The authors have no conflicts of interest to declare that are relevant to the content of this article.
Availability of data and materials
Data sharing is not applicable to this article as no new data were created or analysed in this study.
Author information
Contributions
All authors have equal contribution in the paper and all authors have read and agreed to the published version of the manuscript.
Corresponding author
Rudra Kumar M
Department of Information Technology, Mahatma Gandhi Institute of Technology, Hyderabad, Telangana, India.
Open Access This article is licensed under a Creative Commons Attribution NoDerivs is a more restrictive license. It allows you to redistribute the material commercially or non-commercially but the user cannot make any changes whatsoever to the original, i.e. no derivatives of the original work. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-nd/4.0/
Cite this article
Rudra Kumar M, Rama Vasantha Adiraju, LNC Prakash K, Mahalakshmi V, Penubaka Balaji and Jayavardhanarao Sahukaru, “Hybrid Data Driven Deep Learning Framework for Material Property Prediction”, Journal of Machine and Computing, pp. 1068-1083, April 2025, doi: 10.53759/7669/jmc202505085.
