Learning Rewards, Not Labels: Adversarial Inverse Reinforcement Learning for Machinery Fault Detection
Organizations: Deakin University Waurn Ponds, Victoria, Australia
Abstract
Reinforcement learning (RL) offers significant promise for machinery fault detection (MFD). However, most existing RL-based MFD approaches do not fully exploit RL's sequential decision-making strengths, often treating MFD as a simple guessing game (Contextual Bandits). To bridge this gap, we formulate MFD as an offline inverse reinforcement learning problem, where the agent learns the reward dynamics directly from healthy operational sequences, thereby bypassing the need for manual reward engineering and fault labels. Our framework employs Adversarial Inverse Reinforcement Learning to train a discriminator that distinguishes between normal (expert) and policy-generated transitions. The discriminator's learned reward serves as an anomaly score, indicating deviations from normal operating behaviour. When evaluated on three run-to-failure benchmark datasets (HUMS2023, IMS, and XJTU-SY), the model consistently assigns low anomaly scores to normal samples and high scores to faulty ones, enabling early and robust fault detection. By aligning RL's sequential reasoning with MFD's temporal structure, this work opens a path toward RL-based diagnostics in data-driven industrial settings.
Figures & tables
| Model | Detection | Model | Detection |
| IF | Day 21 (#9) | VAE | Day 21 (#37) |
| OCSVM | Day 21 (#37) | LSTM-VAE | Day 22 (#131) |
| AE | Day 21 (#1) | SS-AD ( Neupane et al., 2024a ) | Day 21 (#50) |
| LSTM-AE | Day 22 (#131) | FRESH filter ( Vaerenberg et al., 2025 ) | Day 22 (#127) |
| CTQN (CB) | No Fault | CW ( Peeters et al., 2024 ) | Day 23 (#175) |
| Committee GT | Day 24 (#264) | AIRL (Ours) | Day 22 (#163) |