A color prediction model for mending materials of the Yuquan Iron Pagoda based on machine learning

Abstract During the restoration of iron cultural relics, the removal of rust from these artifacts is necessary. However, this rust removal process may lead to inconsistent local color on the iron relics. To address this, mending materials are applied to treat the surface, ensuring consistent local color. In the surface treatment of iron cultural relics, a significant challenge lies in modulating the color of these mending materials. This study employs machine learning methods to predict the color of mending materials corresponding to varying contents of Fe2O3, Fe3O4, and epoxy resin. The Artificial Neural Network (ANN), eXtreme Gradient Boosting (XGBoost), and Light Gradient Boost Machine (LightGBM) algorithms are utilized to develop the model, and the predictive performance of these three algorithms is compared. XGBoost exhibits the best prediction performance, achieving a square correlation coefficient (R2) of 0.94238 and a mean absolute error (MAE) of 0.68485. Additionally, the SHapley Additive exPlanations (SHAP) method is employed to analyze the most crucial raw material affecting the color of mending materials, which is identified as Fe3O4. The study illustrates the specific process of employing this model by applying it to the surface treatment of the Yuquan Iron Pagoda, demonstrating the practicality of the model. This model can be applied to assist in the surface treatment of other iron cultural relics.