Resource efficient sensor fusion by knowledge-based network pruning

被引:9
作者
Balemans, Dieter [1 ]
Casteels, Wim [1 ]
Vanneste, Simon [1 ]
de Hoog, Jens [1 ]
Mercelis, Siegfried [1 ]
Hellinckx, Peter [1 ]
机构
[1] Univ Antwerp, IMEC, Fac Appl Engn, ID Lab, Sint Pietersvliet 7, B-2000 Antwerp, Belgium
关键词
Sensor fusion; Deep learning; Neural network pruning; Explainable AI; GRADIENT; VISION;
D O I
10.1016/j.iot.2020.100231
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The perception of the environment is key for autonomous driving applications. To increase the accuracy of perception in different environmental contexts vehicles can rely on both camera and LiDAR sensors that provide complementary information about the same features. Therefore, a sensor fusion method can improve the detection accuracy by combining the information of both sensors. Recently, many sensor fusion methods have been proposed that rely on deep neural networks that typically require a lot of resources to be executed in real-time. Therefore, we propose a resource efficient sensor fusion approach with a new neural network optimization method called knowledge-based pruning. The general principle is to prune the neural network guided by the location of the knowledge within the network that is unveiled with explainable AI methods. More specifically, in this work we propose a pruning method that uses layer-wise relevance propagation (LRP) to localize the network knowledge. The considered sensor fusion method uses off-the-shelve pretrained networks which we optimize for our application using the LRP pruning method. This can be used as a form of transfer learning as a pretrained model is optimized to be applied for a subset of the tasks it was originally trained for. (C) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:11
相关论文
共 26 条
[1]   Structured Pruning of Deep Convolutional Neural Networks [J].
Anwar, Sajid ;
Hwang, Kyuyeon ;
Sung, Wonyong .
ACM JOURNAL ON EMERGING TECHNOLOGIES IN COMPUTING SYSTEMS, 2017, 13 (03)
[2]   On Pixel-Wise Explanations for Non-Linear Classifier Decisions by Layer-Wise Relevance Propagation [J].
Bach, Sebastian ;
Binder, Alexander ;
Montavon, Gregoire ;
Klauschen, Frederick ;
Mueller, Klaus-Robert ;
Samek, Wojciech .
PLOS ONE, 2015, 10 (07)
[3]  
Dong Xin., 2017, Advances in Neural Information Processing Systems, V2017-Decem, P4858
[4]  
Du XX, 2018, IEEE INT CONF ROBOT, P3194
[5]   The PASCAL Visual Object Classes Challenge: A Retrospective [J].
Everingham, Mark ;
Eslami, S. M. Ali ;
Van Gool, Luc ;
Williams, Christopher K. I. ;
Winn, John ;
Zisserman, Andrew .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 111 (01) :98-136
[6]  
France J, 2018, HIST WARFARE, V116, P1, DOI 10.1163/9789004349599_002
[7]   Vision meets robotics: The KITTI dataset [J].
Geiger, A. ;
Lenz, P. ;
Stiller, C. ;
Urtasun, R. .
INTERNATIONAL JOURNAL OF ROBOTICS RESEARCH, 2013, 32 (11) :1231-1237
[8]  
Geiger A, 2012, PROC CVPR IEEE, P3354, DOI 10.1109/CVPR.2012.6248074
[9]   Accelerating Convolutional Neural Networks via Activation Map Compression [J].
Georgiadis, Georgios .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :7078-7088
[10]   Licensed-Assisted Access for LTE in Unlicensed Spectrum: A MAC Protocol Design [J].
Han, Shiying ;
Liang, Ying-Chang ;
Chen, Qian ;
Soong, Boon-Hee .
2016 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2016,