An ensemble learning model for asphalt pavement performance prediction based on gradient boosting decision tree

ABSTRACT

This paper proposes an ensemble learning model that deploys a Gradient Boosting Decision Tree (GBDT) to predict two relevant functional indices, the International roughness index (IRI) and the rut depth (RD), considering multiple influence factors. To train and validate the proposed models, more than 1600 different records were extracted from Long-Term Pavement Performance database. The most suitable hyper parameters for the GBDT model are determined through a grid search and 5-fold cross-validation. Then, a sensitivity analysis is performed to determine the final input variables among the initial considered factors. Further, the optimized models utilise SHAP (Shapley Additive explanation) to interpret the results and analyse the importance of influencing factors. Finally, a comparison experiment with reference artificial intelligence approaches demonstrates that, the GBDT model can outperform the artificial neural network (ANN) and the random forest regression (RFR) methods in terms of quality of prediction results, reaching a coefficient of determination (R²) equal to 0.9. The proposed model can provide more precise pavement performance values and may be useful for providing accurate reference for pavement maintenance and optimising the available budget for road administrations.

KEYWORDS:

• Gradient Boosting Decision Tree
• pavement performance prediction
• IRI
• rutting
• hyper parameter choice
• SHAP

Acknowledgments

Partially sponsored by Tsinghua-Toyota Joint Research Institute Cross-discipline Program.

Disclosure statement

No potential conflict of interest was reported by the author(s).