Multi-class Neural Additive Models : An Interpretable Supervised Learning Method for Gearbox Degradation Detection - Archive ouverte HAL
Communication Dans Un Congrès Année : 2024

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

Résumé

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
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

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

Identifiants

Citer

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. ⟨10.1109/ICPHM61352.2024.10627522⟩. ⟨hal-04562531⟩
300 Consultations
72 Téléchargements

Altmetric

Partager

More