Pick-or-Mix: Dynamic Channel Sampling for ConvNets

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kumar, Ashish, Kim, Daneul, Park, Jaesik, Behera, Laxmidhar
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913392570138624
author Kumar, Ashish
Kim, Daneul
Park, Jaesik
Behera, Laxmidhar
author_facet Kumar, Ashish
Kim, Daneul
Park, Jaesik
Behera, Laxmidhar
contents Channel pruning approaches for convolutional neural networks (ConvNets) deactivate the channels, statically or dynamically, and require special implementation. In addition, channel squeezing in representative ConvNets is carried out via 1x1 convolutions which dominates a large portion of computations and network parameters. Given these challenges, we propose an effective multi-purpose module for dynamic channel sampling, namely Pick-or-Mix (PiX), which does not require special implementation. PiX divides a set of channels into subsets and then picks from them, where the picking decision is dynamically made per each pixel based on the input activations. We plug PiX into prominent ConvNet architectures and verify its multi-purpose utilities. After replacing 1x1 channel squeezing layers in ResNet with PiX, the network becomes 25% faster without losing accuracy. We show that PiX allows ConvNets to learn better data representation than widely adopted approaches to enhance networks' representation power (e.g., SE, CBAM, AFF, SKNet, and DWP). We also show that PiX achieves state-of-the-art performance on network downscaling and dynamic channel pruning applications.
format Preprint
id arxiv_https___arxiv_org_abs_2406_10935
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Pick-or-Mix: Dynamic Channel Sampling for ConvNets
Kumar, Ashish
Kim, Daneul
Park, Jaesik
Behera, Laxmidhar
Computer Vision and Pattern Recognition
Channel pruning approaches for convolutional neural networks (ConvNets) deactivate the channels, statically or dynamically, and require special implementation. In addition, channel squeezing in representative ConvNets is carried out via 1x1 convolutions which dominates a large portion of computations and network parameters. Given these challenges, we propose an effective multi-purpose module for dynamic channel sampling, namely Pick-or-Mix (PiX), which does not require special implementation. PiX divides a set of channels into subsets and then picks from them, where the picking decision is dynamically made per each pixel based on the input activations. We plug PiX into prominent ConvNet architectures and verify its multi-purpose utilities. After replacing 1x1 channel squeezing layers in ResNet with PiX, the network becomes 25% faster without losing accuracy. We show that PiX allows ConvNets to learn better data representation than widely adopted approaches to enhance networks' representation power (e.g., SE, CBAM, AFF, SKNet, and DWP). We also show that PiX achieves state-of-the-art performance on network downscaling and dynamic channel pruning applications.
title Pick-or-Mix: Dynamic Channel Sampling for ConvNets
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.10935