Multi-class Neural Additive Models : An Interpretable Supervised Learning Method for Gearbox Degradation Detection - Archive ouverte HAL Access content directly
Conference Papers Year : 2024

Multi-class Neural Additive Models : An Interpretable Supervised Learning Method for Gearbox Degradation Detection

Abstract

This paper introduces the Multi-class Neural Additive Models (MNAMs), an extension of Neural Additive Models (NAMs) designed to solve multi-class classification problems while remaining interpretable. The paper firstly presents a set of definitions around model interpretability and associated concepts, concepts on which the proposed machine learning method relies on. The core of the contribution lies in the development of MNAM, a model training method designed to minimize the traditional trade-off between accuracy and interpretability. This method is then put to test in a concrete application: the detection of gearbox degradation levels using vibration data, as part of the PHM Society data challenge of 2023. The obtained results demonstrate that a MNAM reaches higher accuracy performance than other interpretable methods such as Decisional Tree (DT) or Generalized Additive Models (GAMs).
Fichier principal
Vignette du fichier
Multi-class Neural Additive Models An Interpretable Supervised Learning Method for Gearbox Degradation Detection.pdf (4.52 Mo) Télécharger le fichier
Origin : Files produced by the author(s)

Dates and versions

hal-04562531 , version 1 (02-05-2024)

Identifiers

  • HAL Id : hal-04562531 , version 1

Cite

Charles-Maxime Gauriat, Yannick Pencolé, Pauline Ribot, Gregory Brouillet. Multi-class Neural Additive Models : An Interpretable Supervised Learning Method for Gearbox Degradation Detection. 2024 IEEE International Conference on Prognostics and Health Management, Jun 2024, Spokane WA, United States. ⟨hal-04562531⟩
0 View
0 Download

Share

Gmail Facebook X LinkedIn More