A Lightweight Pipeline Edge Detection Model Based on Heterogeneous Knowledge Distillation

Chengyuan Zhu,Yanyun Pu,Zhuoling Lyu,Aonan Wu,Kaixiang Yang,Qinmin Yang
DOI: https://doi.org/10.1109/tcsii.2024.3439361
2024-01-01
Abstract:The pipeline safety warning system (PSEW) is an important guarantee for the safe transportation of energy pipelines. Given the constraints of deploying detection models at resource-limited pipeline stations, there is a compelling need to develop efficient, lightweight models suitable for edge device applications. This brief introduces an adaptive heterogeneous model knowledge distillation network (AHKDnet) for edge deployment of pipeline network detection models. The global information and long-distance dependency relationships from the ViT-based teacher network are transferred to the CNN-based shallow student network. We introduce the learnable modulation parameters to optimize target information enhancement, reducing the impact of irrelevant information. By embedding the model selection at each stage of knowledge distillation, the performance collapse of student models caused by misleading cross-architecture knowledge is avoided, and model convergence is accelerated. Experiments on three actual scene datasets of pipeline networks show that AHKDnet outperforms the state-of-the-art KD methods and has strong generalization ability. Notably, AHKDnet enhances the recognition performance of shallow student networks by an average of 10%, highlighting its efficacy and potential for practical applications. Our method can provide a new reference for edge deployment of PSEW.
What problem does this paper attempt to address?