Skip to Main content Skip to Navigation
Conference papers

Attention Based Pruning for Shift Networks

Abstract : In many application domains such as computer vision, Convolutional Layers (CLs) are key to the accuracy of deep learning methods. However, it is often required to assemble a large number of CLs, each containing thousands of parameters, in order to reach state-of-the-art accuracy, thus resulting in complex and demanding systems that are poorly fitted to resource-limited devices. Recently, methods have been proposed to replace the generic convolution operator by the combination of a shift operation and a simpler 1x1 convolution. The resulting block, called Shift Layer (SL), is an efficient alternative to CLs in the sense it allows to reach similar accuracies on various tasks with faster computations and fewer parameters. In this contribution, we introduce Shift Attention Layers (SALs), which extend SLs by using an attention mechanism that learns which shifts are the best at the same time the network function is trained. We demonstrate SALs are able to outperform vanilla SLs (and CLs) on various object recognition benchmarks while significantly reducing the number of float operations and parameters for the inference.
Complete list of metadata

https://hal.archives-ouvertes.fr/hal-03238771
Contributor : Vincent Gripon Connect in order to contact the contributor
Submitted on : Friday, July 30, 2021 - 9:17:27 AM
Last modification on : Friday, August 5, 2022 - 2:54:52 PM
Long-term archiving on: : Sunday, October 31, 2021 - 6:05:35 PM

File

Attention Based Pruning for Sh...
Files produced by the author(s)

Identifiers

Citation

Ghouthi Boukli Hacene, Carlos Lassance, Vincent Gripon, Matthieu Courbariaux, Yoshua Bengio. Attention Based Pruning for Shift Networks. ICPR 2020: 25th International Conference on Pattern Recognition, Jan 2021, Milan (virtual), Italy. pp.4054-4061, ⟨10.1109/ICPR48806.2021.9412859⟩. ⟨hal-03238771⟩

Share

Metrics

Record views

83

Files downloads

34