Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys
This study developed machine learning (ML) models to predict the mechanical properties of Ni-free β-type titanium shape memory alloys (SMAs). Using a dataset of 107 entries derived from both literature and laboratory experiments, we focused on predicting ultimate tensile strength (UTS) and elongatio...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Elsevier
2025-01-01
|
| Series: | Journal of Materials Research and Technology |
| Subjects: | |
| Online Access: | http://www.sciencedirect.com/science/article/pii/S2238785424030473 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832595345862819840 |
|---|---|
| author | Naoki Nohira Taichi Ichisawa Masaki Tahara Itsuo Kumazawa Hideki Hosoda |
| author_facet | Naoki Nohira Taichi Ichisawa Masaki Tahara Itsuo Kumazawa Hideki Hosoda |
| author_sort | Naoki Nohira |
| collection | DOAJ |
| description | This study developed machine learning (ML) models to predict the mechanical properties of Ni-free β-type titanium shape memory alloys (SMAs). Using a dataset of 107 entries derived from both literature and laboratory experiments, we focused on predicting ultimate tensile strength (UTS) and elongation (EL). Key features, including Mo equivalent, bond order, and d-orbital energy level, were selected for the models through Pearson correlation maps and subset selection methods. Four ML algorithms—Linear Regression (LIN), Support Vector Regression (SVR), Random Forest Regression (RFR), and Gradient Boosting Regression (GBR)—were employed and evaluated using metrics like mean absolute error (MAE), mean squared error (MSE), and coefficient of determination (R2). The GBR model for EL showed the highest prediction accuracy (R2 = 0.998 for training and R2 = 0.817 for testing), whereas UTS predictions were less accurate (R2 < 0.6 for testing). Although the models were also adapted to predict yield stress (YS), their accuracy was reduced, with improvements seen when incorporating phase constitution information reflecting phase stability. The primary reasons for the discrepancy in this study include the small dataset size and the absence of microstructural features. This research demonstrates the potential of ML models in predicting the mechanical properties of β-type titanium SMAs, highlighting the importance of integrating domain-specific knowledge through feature engineering to overcome the challenge of small data sets, and to enhance accuracy and robustness. |
| format | Article |
| id | doaj-art-aa635ee3d0264ae5a439c8344d5ce3a5 |
| institution | Kabale University |
| issn | 2238-7854 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Elsevier |
| record_format | Article |
| series | Journal of Materials Research and Technology |
| spelling | doaj-art-aa635ee3d0264ae5a439c8344d5ce3a52025-01-19T06:25:56ZengElsevierJournal of Materials Research and Technology2238-78542025-01-013426342644Machine learning-based prediction of the mechanical properties of β titanium shape memory alloysNaoki Nohira0Taichi Ichisawa1Masaki Tahara2Itsuo Kumazawa3Hideki Hosoda4Corresponding author.; Institute of Integrated Research (IIR), Institute of Science Tokyo, 4259 Nagatsuta-cho, Midori-ku, Yokohama 226-8501, JapanInstitute of Integrated Research (IIR), Institute of Science Tokyo, 4259 Nagatsuta-cho, Midori-ku, Yokohama 226-8501, JapanInstitute of Integrated Research (IIR), Institute of Science Tokyo, 4259 Nagatsuta-cho, Midori-ku, Yokohama 226-8501, JapanInstitute of Integrated Research (IIR), Institute of Science Tokyo, 4259 Nagatsuta-cho, Midori-ku, Yokohama 226-8501, JapanInstitute of Integrated Research (IIR), Institute of Science Tokyo, 4259 Nagatsuta-cho, Midori-ku, Yokohama 226-8501, JapanThis study developed machine learning (ML) models to predict the mechanical properties of Ni-free β-type titanium shape memory alloys (SMAs). Using a dataset of 107 entries derived from both literature and laboratory experiments, we focused on predicting ultimate tensile strength (UTS) and elongation (EL). Key features, including Mo equivalent, bond order, and d-orbital energy level, were selected for the models through Pearson correlation maps and subset selection methods. Four ML algorithms—Linear Regression (LIN), Support Vector Regression (SVR), Random Forest Regression (RFR), and Gradient Boosting Regression (GBR)—were employed and evaluated using metrics like mean absolute error (MAE), mean squared error (MSE), and coefficient of determination (R2). The GBR model for EL showed the highest prediction accuracy (R2 = 0.998 for training and R2 = 0.817 for testing), whereas UTS predictions were less accurate (R2 < 0.6 for testing). Although the models were also adapted to predict yield stress (YS), their accuracy was reduced, with improvements seen when incorporating phase constitution information reflecting phase stability. The primary reasons for the discrepancy in this study include the small dataset size and the absence of microstructural features. This research demonstrates the potential of ML models in predicting the mechanical properties of β-type titanium SMAs, highlighting the importance of integrating domain-specific knowledge through feature engineering to overcome the challenge of small data sets, and to enhance accuracy and robustness.http://www.sciencedirect.com/science/article/pii/S2238785424030473β-Ti alloyMachine learningMechanical propertyRegression modelShape memory alloy |
| spellingShingle | Naoki Nohira Taichi Ichisawa Masaki Tahara Itsuo Kumazawa Hideki Hosoda Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys Journal of Materials Research and Technology β-Ti alloy Machine learning Mechanical property Regression model Shape memory alloy |
| title | Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys |
| title_full | Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys |
| title_fullStr | Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys |
| title_full_unstemmed | Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys |
| title_short | Machine learning-based prediction of the mechanical properties of β titanium shape memory alloys |
| title_sort | machine learning based prediction of the mechanical properties of β titanium shape memory alloys |
| topic | β-Ti alloy Machine learning Mechanical property Regression model Shape memory alloy |
| url | http://www.sciencedirect.com/science/article/pii/S2238785424030473 |
| work_keys_str_mv | AT naokinohira machinelearningbasedpredictionofthemechanicalpropertiesofbtitaniumshapememoryalloys AT taichiichisawa machinelearningbasedpredictionofthemechanicalpropertiesofbtitaniumshapememoryalloys AT masakitahara machinelearningbasedpredictionofthemechanicalpropertiesofbtitaniumshapememoryalloys AT itsuokumazawa machinelearningbasedpredictionofthemechanicalpropertiesofbtitaniumshapememoryalloys AT hidekihosoda machinelearningbasedpredictionofthemechanicalpropertiesofbtitaniumshapememoryalloys |