Multi-Class Leak Detection in Water Pipelines Using a Wavelet-Guided Frequency-Informed Transformer

Abstract

Water utilities continue to lose a lot of Non-Revenue Water (NRW) because of leaks that go undetected. This makes it necessary to find accurate but easy-to-use monitoring solutions. This paper presents FiT-WST+, a wavelet-guided Frequency-Informed Transformer (FiT) designed for the classification of five distinct leak types utilising accelerometer measurements. The proposed architecture combines the spectral modelling ability of a FIT with the stable translation-invariant representation of the Wavelet Scattering Transform (WST). The model uses a guided attention mechanism to combine spectral and scattering cues that work well together to make classes more distinct, especially for fault types that are similar. On the held-out test set, FiT-WST+ achieves 99.6% accuracy, 99.6% balanced accuracy, and a 99.6% macro-averaged F1-score. Comparative benchmarking against recent methods tested on the same dataset shows that this method works at a low sampling rate (1 kHz), which greatly lowers bandwidth needs and allows for scalable deployment on edge devices with limited resources for real-time monitoring of important water infrastructure.