Chengyuan Zhu;Yanyun Pu;Zhuoling Lyu;Aonan Wu;Kaixiang Yang;Qinmin Yang
{"title":"基于异构知识蒸馏的轻量级管道边缘检测模型","authors":"Chengyuan Zhu;Yanyun Pu;Zhuoling Lyu;Aonan Wu;Kaixiang Yang;Qinmin Yang","doi":"10.1109/TCSII.2024.3439361","DOIUrl":null,"url":null,"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.","PeriodicalId":13101,"journal":{"name":"IEEE Transactions on Circuits and Systems II: Express Briefs","volume":"71 12","pages":"5059-5063"},"PeriodicalIF":4.6000,"publicationDate":"2024-08-06","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"A Lightweight Pipeline Edge Detection Model Based on Heterogeneous Knowledge Distillation\",\"authors\":\"Chengyuan Zhu;Yanyun Pu;Zhuoling Lyu;Aonan Wu;Kaixiang Yang;Qinmin Yang\",\"doi\":\"10.1109/TCSII.2024.3439361\",\"DOIUrl\":null,\"url\":null,\"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.\",\"PeriodicalId\":13101,\"journal\":{\"name\":\"IEEE Transactions on Circuits and Systems II: Express Briefs\",\"volume\":\"71 12\",\"pages\":\"5059-5063\"},\"PeriodicalIF\":4.6000,\"publicationDate\":\"2024-08-06\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"IEEE Transactions on Circuits and Systems II: Express Briefs\",\"FirstCategoryId\":\"5\",\"ListUrlMain\":\"https://ieeexplore.ieee.org/document/10623816/\",\"RegionNum\":2,\"RegionCategory\":\"工程技术\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q2\",\"JCRName\":\"ENGINEERING, ELECTRICAL & ELECTRONIC\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"IEEE Transactions on Circuits and Systems II: Express Briefs","FirstCategoryId":"5","ListUrlMain":"https://ieeexplore.ieee.org/document/10623816/","RegionNum":2,"RegionCategory":"工程技术","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q2","JCRName":"ENGINEERING, ELECTRICAL & ELECTRONIC","Score":null,"Total":0}
A Lightweight Pipeline Edge Detection Model Based on Heterogeneous Knowledge Distillation
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.
期刊介绍:
TCAS II publishes brief papers in the field specified by the theory, analysis, design, and practical implementations of circuits, and the application of circuit techniques to systems and to signal processing. Included is the whole spectrum from basic scientific theory to industrial applications. The field of interest covered includes:
Circuits: Analog, Digital and Mixed Signal Circuits and Systems
Nonlinear Circuits and Systems, Integrated Sensors, MEMS and Systems on Chip, Nanoscale Circuits and Systems, Optoelectronic
Circuits and Systems, Power Electronics and Systems
Software for Analog-and-Logic Circuits and Systems
Control aspects of Circuits and Systems.