Distilled Center and Scale Prediction: Distilling anchor-free pedestrian detector for edge computing

被引:0
作者
Wang, Jianyuan [1 ]
She, Liang [2 ,3 ]
Wang, Wei [2 ]
Liu, Xinyue [4 ]
Zeng, Yangyan [5 ]
机构
[1] Univ Sci & Technol Beijing, Sch Intelligence Sci & Technol, Beijing 100183, Peoples R China
[2] Cent South Univ, Sch Comp Sci & Engn, Changsha 410083, Peoples R China
[3] Hunan Univ Technol & Business, Sch Comp Sci, Changsha 410205, Peoples R China
[4] Beihang Univ, Sch Comp Sci & Engn, Beijing 100191, Peoples R China
[5] Hunan Univ Technol & Business, Sch Frontier Crossover Studies, Changsha 410205, Peoples R China
基金
中国国家自然科学基金;
关键词
Pedestrian detection; Knowledge distillation; Internet of Things; Edge computing; NETWORK;
D O I
10.1016/j.iot.2024.101444
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
As an important task of the Internet of Things, pedestrian detection has attained good detection results powered by deep learning. However, common pedestrian detectors based on deep learning usually require lots of computing resources and consume large amounts of energy, making them not suitable for edge devices in the edge computing paradigm. Therefore, for the pedestrian detection task in edge computing, we consider knowledge distillation on the anchor-free detector Center and Scale Prediction to ensure the pedestrian detection performance while reducing the parameter number and inference time of the detector as much as possible. We propose a distillation framework Distilled Center and Scale Prediction, in which we implement feature-based and response-based distillation for transferring knowledge from the larger model to the smaller model. In order to transmit information useful for detection in distillation as much as possible, Multi-Reference Distillation is designed to filter the transferred knowledge. Moreover, Cross-Module Distillation is proposed to enhance the transfer of relational information during distillation. We perform related experiments on the CityPersons dataset. Our proposed distilled detector achieves 10.45% MR-2 with ResNet18 as backbone and 10.27% MR-2 with ResNet50 as backbone, even outperforming the original teacher detectors. At the same time, the inference time per image is reduced by more than 10% compared to the original teacher detector.
引用
收藏
页数:15
相关论文
共 74 条
[1]   A Novel Resource Sharing Scheme for Vehicular Communication in 5G Cellular Networks for Smart Cities [J].
Akhter, Javed ;
Hazra, Ranjay ;
Mihovska, Albena ;
Prasad, Ramjee .
IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2024, 70 (03) :5848-5856
[2]   Toward Efficient Sensor Deployment in Internet of Things for Target Coverage and Sensor Connectivity [J].
Balaji, S. ;
Pavithra, R. ;
Arivudainambi, D. ;
Varunkumar, K. A. ;
Suresh, A. ;
Omar, Marwan ;
Bashir, Ali Kashif .
IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2024, 70 (03) :5432-5440
[3]   Cascade R-CNN: Delving into High Quality Object Detection [J].
Cai, Zhaowei ;
Vasconcelos, Nuno .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6154-6162
[4]   Multicamera edge-computing system for persons indoor location and tracking [J].
Carro-Lagoa, Angel ;
Barral, Valentin ;
Gonzalez-Lopez, Miguel ;
Escudero, Carlos J. ;
Castedo, Luis .
INTERNET OF THINGS, 2023, 24
[5]   Enhancing pedestrian mobility in Smart Cities using Big Data [J].
Carter, Ebony ;
Adam, Patrick ;
Tsakis, Deon ;
Shaw, Stephanie ;
Watson, Richard ;
Ryan, Peter .
JOURNAL OF MANAGEMENT ANALYTICS, 2020, 7 (02) :173-188
[6]  
Chen GB, 2017, ADV NEUR IN, V30
[7]   A deep neural network compression algorithm based on knowledge transfer for edge devices [J].
Chen, Yanming ;
Li, Chao ;
Gong, Luqi ;
Wen, Xiang ;
Zhang, Yiwen ;
Shi, Weisong .
COMPUTER COMMUNICATIONS, 2020, 163 :186-194
[8]   The Cityscapes Dataset for Semantic Urban Scene Understanding [J].
Cordts, Marius ;
Omran, Mohamed ;
Ramos, Sebastian ;
Rehfeld, Timo ;
Enzweiler, Markus ;
Benenson, Rodrigo ;
Franke, Uwe ;
Roth, Stefan ;
Schiele, Bernt .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :3213-3223
[9]   General Instance Distillation for Object Detection [J].
Dai, Xing ;
Jiang, Zeren ;
Wu, Zhao ;
Bao, Yiping ;
Wang, Zhicheng ;
Liu, Si ;
Zhou, Erjin .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :7838-7847
[10]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893