335

Better Together: Resnet-50 accuracy with 13×13 \times fewer parameters and at 3×3\times speed

Abstract

Recent research on compressing deep neural networks has focused on reducing the number of parameters. Smaller networks are easier to export and deploy on edge-devices. We introduce \textit{Adjoined networks} as a training approach that can regularize and compress any CNN-based neural architecture. Our one-shot learning paradigm trains both the original and the smaller networks together. The parameters of the smaller network are shared across both the architectures. We prove strong theoretical guarantees on the regularization behavior of the adjoint training paradigm. We complement our theoretical analysis by an extensive empirical evaluation of both the compression and regularization behavior of adjoint networks. For resnet-50 trained adjointly on Imagenet, we are able to achieve a 13.7x13.7x reduction in the number of parameters (For size comparison, we ignore the parameters in the last linear layer as it varies by dataset and are typically dropped during fine-tuning. Else, the reductions are 11.5x11.5x and 95x95x for imagenet and cifar-100 respectively.) and a 3x3x improvement in inference time without any significant drop in accuracy. For the same architecture on CIFAR-100, we are able to achieve a 99.7x99.7x reduction in the number of parameters and a 5x5x improvement in inference time. On both these datasets, the original network trained in the adjoint fashion gains about 3%3\% in top-1 accuracy as compared to the same network trained in the standard fashion.

View on arXiv
Comments on this paper