Skip to main content
SHARE
Publication

Improving Robustness of Spectrogram Classifiers with Neural Stochastic Differential Equations

by Joel R Brogan, Olivera Kotevska, Anibely A Torres Polanco, Sumit Jha, Mark B Adams
Publication Type
Conference Paper
Book Title
2024 IEEE 34th International Workshop on Machine Learning for Signal Processing (MLSP)
Publication Date
Page Numbers
1 to 6
Publisher Location
New Jersey, United States of America
Conference Name
IEEE International Workshop on Machine Learning for Signal Processing (MLSP)
Conference Location
London, United Kingdom
Conference Sponsor
IEEE
Conference Date
-

Signal analysis and classification is fraught with high levels of noise and perturbation. Computer-vision-based deep learning models applied to spectrograms have proven useful in the field of signal classification and detection; however, these methods aren't designed to handle the low signal-to-noise ratios inherent within non-vision signal processing tasks. While they are powerful, they are currently not the method of choice in the inherently noisy and dynamic critical infrastructure domain, such as smart-grid sensing, anomaly detection, and non-intrusive load monitoring. Currently, these models can be brittle, which makes them susceptible to noisy input. This also means they have sub-optimal stability of explanation outputs. Experts and technicians using these models to make decisions in real world scenarios need assurance that a model is performing as it is supposed to. The classification or prediction outputs it generates should be sound and grounded, not likely to change in the presence of shifting noise landscapes. In this work, we explore the idea of Neural Stochastic Differential Equations (NSDE's) to improve the robustness of models trained to classify time series data and the effect of NSDE's on the explainability of outputs. We then test the effectiveness of these approaches by applying them to a non-intrusive load monitoring (NILM) dataset that consists of simulated harmonic signals injected into a real building.