AutoML for Feature Selection and Model Tuning Applied to Fault Severity Diagnosis in Spur Gearboxes

Abstract

Gearboxes are widely used in industrial processes as mechanical power transmission systems. Then, gearbox failures can affect other parts of the system and produce economic loss. The early detection of the possible failure modes and their severity assessment in such devices is an important field of research. Data-driven approaches usually require an exhaustive development of pipelines including models’ parameter optimization and feature selection. This paper takes advantage of the recent Auto Machine Learning (AutoML) tools to propose proper feature and model selection for three failure modes under different severity levels: broken tooth, pitting and crack. The performance of 64 statistical condition indicators (SCI) extracted from vibration signals under the three failure modes were analyzed by two AutoML systems, namely the H2O Driverless AI platform and TPOT, both of which include feature engineering and feature selection mechanisms. In both cases, the systems converged to different types of decision tree methods, with ensembles of XGBoost models preferred by H2O while TPOT generated different types of stacked models. The models produced by both systems achieved very high, and practically equivalent, performances on all problems. Both AutoML systems converged to pipelines that focus on very similar subsets of features across all problems, indicating that several problems in this domain can be solved by a rather small set of 10 common features, with accuracy up to 90%. This latter result is important in the research of useful feature selection for gearbox fault diagnosis.

Exploring Factors That Affected Student Well-Being during the COVID-19 Pandemic: A Comparison of Data-Mining Approaches

“…Regardless of the frequency of using the mentioned learning methods, it should be noted that according to the “No-Free-Lunch” theorem, there is no single superior classifier that perfectly fits every dataset. Therefore, a comprehensive pipeline development such as parameter optimization and variable selection is required for such data-specific techniques [ 40 ]. Once the models are built, some fundamental properties (i.e., predictive accuracy, speed, robustness, scalability, interpretability, and simplicity [ 41 ]) assist in the selection of the final model that will be used for different tasks, as stated in Bakhshinategh et al [ 32 ] and Hernandez-Blanco et al [ 35 ].…”

Attention Measurement of an Autism Spectrum Disorder User Using EEG Signals: A Case Study

“…Nowadays, intelligent systems that incorporate artificial intelligence (AI) frequently rely on machine learning (ML) [21,22]. ML is a term that refers to a system's ability to learn from problem-specific training data in order to automate the process of developing analytical models and completing associated tasks [23,24].…”

Evaluation of Machine Learning Algorithms for Early Diagnosis of Deep Venous Thrombosis

“…The ML algorithms have advanced in the early diagnosis of DVT and other applications [41][42][43], moving from binary Decision Trees developed by the team of [44] to more sophisticated algorithms that integrate image analysis by AI [18] and are also very complex in that they go into up to 68 variables to give a final verdict of this disease [45]. In some investigations with very big datasets, the predictors have an area under the receiver operating characteristic (AU-ROC) of 0.83 to 0.85 [46].…”