Fine-Grained Leakage Detection for Water Supply Pipelines Based on CNN and Selective State-Space Models

The water supply pipeline system is responsible for providing clean drinking water to residents, but pipeline leaks can lead to water resource wastage, increased operational costs, and safety hazards. To effectively detect the leakage level in the water supply pipelines and address the difficulty of accurately distinguishing fine-grained leakage levels using traditional methods, this paper proposes a fine-grained leakage identification method based on Convolutional Neural Networks (CNN) and the Selective State Space Model (Mamba). An experimental platform was built to simulate different leakage conditions, and multi-axis sensors were used to collect data, resulting in the creation of a high-quality dataset. The signals were converted into frequency-domain images using Short-Time Fourier Transform (STFT), and CNN was employed to extract image features. Mamba was integrated to capture the one-dimensional time dynamic characteristics of the leakage signal, and the CosFace loss function was introduced to increase the inter-class distance, thereby improving the fine-grained classification ability. Experimental results show that the proposed method achieves optimal performance across various evaluation metrics. Compared to SVM, BP neural networks, and CNN methods, the accuracy was improved by 17.9%, 15.9%, and 3.0%, respectively. Compared to Support Vector Machine (SVM), Backpropagation neural network (BP), attention mechanism with the LSTM network (LSTM-AM), CNN, and inverted transformers network (iTransformer) methods, the accuracy improved by 17.9%, 15.9%, 7.8%, 3.0%, and 2.3%, respectively. Additionally, the method enhanced intra-class consistency and increased inter-class differences, showing outstanding performance at different leakage levels, which could contribute to improved intelligent management for water pipeline leakage detection.