Controlling Information Capacity of Binary Neural Network

Abstract

Despite the growing popularity of deep learning technologies, high memory requirements and power consumption are essentially limiting their application in mobile and IoT areas. While binary convolutional networks can alleviate these problems, the limited bitwidth of weights is often leading to significant degradation of prediction accuracy. In this paper, we present a method for training binary networks that maintains a stable predefined level of their information capacity throughout the training process by applying Shannon entropy based penalty to convolutional filters. The results of experiments conducted on SVHN, CIFAR and ImageNet datasets demonstrate that the proposed approach can statistically significantly improve the accuracy of binary networks.

References in corpus (5)

• Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activations
• Towards Accurate Binary Convolutional Neural Network
• Regularizing Neural Networks by Penalizing Confident Output Distributions
• Improving Neural Network Quantization without Retraining using Outlier Channel Splitting
• Reward Augmented Maximum Likelihood for Neural Structured Prediction

Cited by in corpus (4)

• Fast and Accurate Single-Image Depth Estimation on Mobile Devices, Mobile AI 2021 Challenge: Report
• Real-Time Video Super-Resolution on Smartphones with Deep Learning, Mobile AI 2021 Challenge: Report
• Real-Time Quantized Image Super-Resolution on Mobile NPUs, Mobile AI 2021 Challenge: Report
• Fast Camera Image Denoising on Mobile GPUs with Deep Learning, Mobile AI 2021 Challenge: Report