Hybrid Quantum Neural Network and Shapely Additive Explanations in Railway Track Geometry Modeling

Abstract

This research investigates the application of explainable quantum machine learning (QML) for predictive maintenance in the railroad industry. By utilizing ground-penetrating radar (GPR) data to characterize subsurface track conditions (ballast fouling index, ballast thickness index, layer roughness index, and moisture likelihood index), a quantum neural network (QNN) model was developed to predict track geometry (profile and alignment) defects in a Class 3 railroad track. Shapley additive explanations (SHAP) were employed to analyze the feature importance and the model’s decision-making processes to ensure model interpretability. The QNN model correctly predicted 42 out of 55 test data points. SHAP analysis identified the ballast fouling index and layer roughness index as the most important parameters, aligning with engineering expectations.