Automated machine learning pipeline for robust project cost and duration forecasting
Filippo Maria Ottaviani, Pablo Ballesteros‐Pérez, Timur Narbaev
Abstract
Projects often face issues that trigger project controls, where Estimates at Completion (EACs) play a crucial role in determining the scope of corrective actions. Recent studies have applied supervised Machine Learning (ML) regression techniques to develop EAC models, utilizing features derived from Earned Value Management (EVM) and Earned Schedule Management (ESM) methodologies. However, these studies overlook several underfitting and overfitting issues that could compromise model robustness, leading to biased results. This paper introduces an ML pipeline designed to address these issues through automated procedures for data balancing and augmentation, feature engineering, and model training and evaluation. The pipeline was tested with 30 ML techniques on a dataset of 50 real-world construction projects. Results show that the EAC models developed through the pipeline achieve superior accuracy, precision, and timeliness to EVM and ESM ones. These findings validate the pipeline and offer practitioners an automated framework for developing robust, ML-based EAC models.