arXiv · 1908.06694
Adaptative Inference Cost With Convolutional Neural Mixture Models
Abstract
Despite the outstanding performance of convolutional neural networks (CNNs) for many vision tasks, the required computational cost during inference is problematic when resources are limited. In this context, we propose Convolutional Neural Mixture Models (CNMMs), a probabilistic model embedding a large number of CNNs that can be jointly trained and evaluated in an efficient manner. Within the proposed framework, we present different mechanisms to prune subsets of CNNs from the mixture, allowing to easily adapt the computational cost required for inference. Image classification and semantic segmentation experiments show that our method achieve excellent accuracy-compute trade-offs. Moreover, unlike most of previous approaches, a single CNMM provides a large range of operating points along this trade-off, without any re-training.
Explore related subjects
Keep this discovery
Adria Ruiz, Jakob Verbeek. 2019-08-19. Adaptative Inference Cost With Convolutional Neural Mixture Models. https://arxiv.org/abs/1908.06694
Cite the original work for its findings. Save a collection to share your selection of sources.