VARADE: a Variational-based AutoRegressive model for Anomaly Detection on the Edge
Alessio Mascolini, Sebastiano Gaiardelli, Francesco Ponzio, Nicola Dall'Ora, Enrico Macii, Sara Vinco, Santa Di Cataldo, Franco Fummi
TL;DR
VARADE introduces a variationally grounded autoregressive model tailored for edge-based anomaly detection in Industry 4.0. By predicting the next time step as a Gaussian distribution through a light convolutional architecture, and optimizing via a joint reconstruction and KL-divergence loss, the model provides a robust anomaly score based on predictive uncertainty. In a real industrial case with a KUKA manipulator, VARADE outperformed lightweight baselines in AUC-ROC while maintaining low latency and reasonable power and memory use on both Jetson Xavier NX and Jetson AGX Orin platforms. The approach demonstrates the practicality of on-device, real-time anomaly detection for complex, multivariate streams, with direct implications for edge AI deployment in smart factories.
Abstract
Detecting complex anomalies on massive amounts of data is a crucial task in Industry 4.0, best addressed by deep learning. However, available solutions are computationally demanding, requiring cloud architectures prone to latency and bandwidth issues. This work presents VARADE, a novel solution implementing a light autoregressive framework based on variational inference, which is best suited for real-time execution on the edge. The proposed approach was validated on a robotic arm, part of a pilot production line, and compared with several state-of-the-art algorithms, obtaining the best trade-off between anomaly detection accuracy, power consumption and inference frequency on two different edge platforms.
