TY - JOUR
T1 - Multi-Scale Semantic Map Distillation for Lightweight Pavement Crack Detection
AU - Wang, Xin
AU - Mao, Zhaoyong
AU - Liang, Zhiwei
AU - Shen, Junge
N1 - Publisher Copyright:
© 2024 IEEE.
PY - 2024
Y1 - 2024
N2 - Timely and accurate pavement crack detection plays a crucial role in urban transportation and pavement management. Recently, deep learning-based pavement crack detection approaches have witnessed significant advancements. However, high-performance networks heavily rely on complex network structures and massive model parameters, making their application in practical scenarios challenging. To achieve efficient yet accurate pavement crack detection, this paper presents a novel Lightweight Pavement Crack Detection based on Multi-Scale Semantic Map Distillation (LPCD-MSMD). Specifically, we improve the U-Net by designing a novel Cascaded U-Net (C-UNet) structure and incorporating additional branches to capture more subtle features. The proposed C-UNet is optimized by reducing the number of convolutional layers and compressing the channel dimensions, benefiting to a more lightweight model while maintaining comparable performance. Additionally, we utilize the original C-UNet as the teacher network and the simplified network as the student network for a lightweight pavement crack detection. Through the proposed multi-scale semantic map distillation strategy, the student network can acquire multi-scale output knowledge from the cascaded output of the teacher network, benefiting significantly improved performance of the CU-Net(s). Thorough experimental results demonstrate that the elaborately designed student network possesses a remarkably small parameter size of only 0.54 MB. Moreover, the performance evaluated on the Crack500 dataset and the GAPS384 dataset indicate the superiority of proposed method over several typical deep learning-based approaches including SegNet, FCN, BiSeNet, and U-Net. In particular, our method outperforms the second-best method by 0.09 and 0.124 in terms of F1 score evaluated on the Crack500 dataset and the GAPS384 dataset, respectively.
AB - Timely and accurate pavement crack detection plays a crucial role in urban transportation and pavement management. Recently, deep learning-based pavement crack detection approaches have witnessed significant advancements. However, high-performance networks heavily rely on complex network structures and massive model parameters, making their application in practical scenarios challenging. To achieve efficient yet accurate pavement crack detection, this paper presents a novel Lightweight Pavement Crack Detection based on Multi-Scale Semantic Map Distillation (LPCD-MSMD). Specifically, we improve the U-Net by designing a novel Cascaded U-Net (C-UNet) structure and incorporating additional branches to capture more subtle features. The proposed C-UNet is optimized by reducing the number of convolutional layers and compressing the channel dimensions, benefiting to a more lightweight model while maintaining comparable performance. Additionally, we utilize the original C-UNet as the teacher network and the simplified network as the student network for a lightweight pavement crack detection. Through the proposed multi-scale semantic map distillation strategy, the student network can acquire multi-scale output knowledge from the cascaded output of the teacher network, benefiting significantly improved performance of the CU-Net(s). Thorough experimental results demonstrate that the elaborately designed student network possesses a remarkably small parameter size of only 0.54 MB. Moreover, the performance evaluated on the Crack500 dataset and the GAPS384 dataset indicate the superiority of proposed method over several typical deep learning-based approaches including SegNet, FCN, BiSeNet, and U-Net. In particular, our method outperforms the second-best method by 0.09 and 0.124 in terms of F1 score evaluated on the Crack500 dataset and the GAPS384 dataset, respectively.
KW - Cascaded U-Net
KW - coordinate attention
KW - multi-scale semantic map distillation
KW - pavement crack detection
UR - http://www.scopus.com/inward/record.url?scp=85196735892&partnerID=8YFLogxK
U2 - 10.1109/TITS.2024.3405477
DO - 10.1109/TITS.2024.3405477
M3 - 文章
AN - SCOPUS:85196735892
SN - 1524-9050
VL - 25
SP - 15081
EP - 15093
JO - IEEE Transactions on Intelligent Transportation Systems
JF - IEEE Transactions on Intelligent Transportation Systems
IS - 10
ER -