Classification of sewer pipe defects based on an automatically designed convolutional neural network

Yu Wang,Jiahao Fan,Yanan Sun
DOI: https://doi.org/10.1016/j.eswa.2024.125806
IF: 8.5
2024-12-01
Expert Systems with Applications
Abstract:Accurate classification of sewer pipe defects allows for timely prevention of system failures, preventing environmental pollution caused by sewage leakage, thereby safeguarding public health and ensuring urban infrastructure operates normally. However, existing methods with deep convolutional neural networks not only require a significant amount of time to design network architectures, but also demand substantial computational resources. Therefore, this paper automatically designed high-performance and lightweight AutoSewerNet based on neural architecture search (NAS) for classification of sewer pipe defects. First, Super-net is designed to enrich the diversity of search space and include high-performance network architecture. Second, to reduce the search time of NAS, a gradient-based search strategy is designed. Third, AutoSewerNet is designed to be lightweight and perform real-time inspections of sewer pipes. Fourth, weight balance is introduced to solve the imbalanced dataset problem. Experimental results demonstrate that AutoSewerNet achieved an F1-score of 0.6251 on the benchmark dataset, which is far superior to that of ResNet-50 (F1-score: 0.4523) and InceptionV3 (F1-score: 0.4611). Moreover, AutoSewerNet requires only 11.6% of VGG-16. Thus, AutoSewerNet is better than state-of-the-art methods. Our source code and models are anonymously available at https://anonymous.4open.science/r/ASN-26AB/ .
computer science, artificial intelligence,engineering, electrical & electronic,operations research & management science
What problem does this paper attempt to address?