|
|
an efficient knowledge distillation architecture for real-time semantic segmentation
|
|
|
|
|
نویسنده
|
mansourian amir m. ,karimi bavandpour nader ,kasaei shohreh
|
منبع
|
aut journal of modeling and simulation - 2023 - دوره : 55 - شماره : 1 - صفحه:99 -108
|
چکیده
|
In recent years, convolutional neural networks (cnns) have made significant strides in the field of segmentation, particularly in semantic segmentation where both accuracy and efficiency are crucial. however, despite their high accuracy, these deep networks are not practical for real-time use due to their low inference speed. this issue has prompted researchers to explore various techniques to improve the efficiency of cnns. one such technique is knowledge distillation, which involves transferring knowledge from a larger, cumbersome (teacher) model to a smaller, more compact (student) model. this paper proposes a simple yet efficient approach to address the issue of low inference speed in cnns using knowledge distillation. the proposed method involves distilling knowledge from the feature maps of the teacher model to guide the learning of the student model. the approach uses a straightforward technique known as pixel-wise distillation to transfer the feature maps of the last convolution layer of the teacher model to the student model. additionally, a pair-wise distillation technique is used to transfer pair-wise similarities of the intermediate layers. to validate the effectiveness of the proposed method, extensive experiments were conducted on the pascalvoc 2012 dataset using a state-of-the-art deeplabv3+ segmentation network with different backbone architectures. the results showed that the proposed method achieved a balanced mean intersection over union (miou) and training time.
|
کلیدواژه
|
computer vision ,convolutional neural networks ,deep learning ,semantic segmentation ,knowledge distillation ,deep neural networks
|
آدرس
|
sharif university of technology, department of computer engineering, iran, sharif university of technology, department of computer engineering, iran, sharif university of technology, department of computer engineering, iran
|
پست الکترونیکی
|
kasaei@sharif.edu
|
|
|
|
|
|
|
|
|
|
|
|
Authors
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|