Abstract:Accurately predicting the performance of a tunnel boring machine (TBM) is important to safe and efficient tunneling. The application of machine learning algorithms to TBM performance prediction creates several challenges. Such prediction is a nontrivial task involving procedures such as data preprocessing, selection of a machine learning algorithm and optimization of the related hyperparameters. The demand for expert knowledge has restricted the application of machine learning methods to TBM performance prediction, and it is meaningful to study predicting TBM performance automatically. In this paper, we explore three approaches to TBM performance prediction using Bayesian optimization and automated machine learning (AutoML). In the first study, Bayesian optimization is used to determine the optimal hyperparameters of various machine learning algorithms, including support vector regression (SVR), decision tree, bagging tree, random forest and AdaBoost. We attain the minimum mean squared error (MSE) values of 3.135 x 10(-2) and 3.177 x 10(-2) for a decision tree and SVR, respectively. In the second approach called the neural architecture search (NAS), the optimal combination of architecture, hyperparameters and the training procedure of an artificial neural network is found in a single operation. We obtain the optimal results of 3.514 x 10(-2) and 3.237 x 10(-2) if complete and simplified NAS are used, respectively. In the third method, the best combination of a data preprocessing method, a machine learning model and the related hyperparameters is found, and an optimal MSE value of 3.148 x 10(-2) is obtained using AutoML. In all three studies, we obtain state-of-the-art prediction results that are superior to a previous best prediction result of 3.500 x 10(-2). The prediction results prove that Bayesian optimization and AutoML are powerful tools that can not only effectively predict TBM performance but also reduce the demand for expert knowledge of machine learning.

HPC Application Performance Prediction with Machine Learning on New Architectures

Learning with Analytical Models

Machine Learning Enabled Scalable Performance Prediction of Scientific Codes

Towards CPU Performance Prediction: New Challenge Benchmark Dataset and Novel Approach

OpenCL Performance Prediction using Architecture-Independent Features

Performance and power modeling and prediction using MuMMI and 10 machine learning methods

BB-ML: Basic Block Performance Prediction using Machine Learning Techniques

Performance Modelling of Deep Learning on Intel Many Integrated Core Architectures

CAE Performance Prediction Using Machine Learning Model Based On Historical Data

Multi-Parameter Performance Modeling Based on Machine Learning with Basic Block Features

MambaCPU: Enhanced Correlation Mining with State Space Models for CPU Performance Prediction

Machine Learning-enabled Performance Model for DNN Applications and AI Accelerator

Accelerating Computer Architecture Simulation through Machine Learning

TBM Performance Prediction with Bayesian Optimization and Automated Machine Learning

Performance and Power Modeling and Prediction Using MuMMI and Ten Machine Learning Methods

Accurate and efficient regression modeling for microarchitectural performance and power prediction

Automatic Multi-Parameter Performance Modeling of HPC Applications on a New Sunway Supercomputer

A PERFORMANCE PREDICTION METHOD BASED ON NAIVE BAYES CLASSIFICATION

Analytics of Longitudinal System Monitoring Data for Performance Prediction

Towards High Performance, Portability, and Productivity: Lightweight Augmented Neural Networks for Performance Prediction