#Densely Connected Convolutional Networks (DenseNets) This repository contains the code for the paper Densely Connected Convolutional Networks.
The code is based on Facebook's implementation of ResNet (https://github.com/facebook/fb.resnet.torch).
Also, see
- Our [Caffe Implementation] (https://github.com/liuzhuang13/DenseNetCaffe)
- Our more memory-efficient [Torch Implementation] (https://github.com/gaohuang/DenseNet_lite).
- [Tensorflow Implementation] (https://github.com/YixuanLi/densenet-tensorflow) by Yixuan Li.
- [Tensorflow Implementation] (https://github.com/LaurentMazare/deep-models/tree/master/densenet) by Laurent Mazare.
- [Lasagne Implementation] (https://github.com/Lasagne/Recipes/tree/master/papers/densenet) by Jan Schlüter.
- [Keras Implementation] (https://github.com/tdeboissiere/DeepLearningImplementations/tree/master/DenseNet) by tdeboissiere.
- [Keras Implementation] (https://github.com/robertomest/convnet-study) by Roberto de Moura Estevão Filho.
- [Chainer Implementation] (https://github.com/t-hanya/chainer-DenseNet) by Toshinori Hanya.
- [Chainer Implementation] (https://github.com/yasunorikudo/chainer-DenseNet) by Yasunori Kudo.
If you find this helps your research, please consider citing:
@article{Huang2016Densely,
author = {Huang, Gao and Liu, Zhuang and Weinberger, Kilian Q.},
title = {Densely Connected Convolutional Networks},
journal = {arXiv preprint arXiv:1608.06993},
year = {2016}
}
##Table of Contents 0. Introduction 0. Results 0. Usage 0. Contact
##Introduction DenseNet is a network architecture where each layer is directly connected to every other layer in a feed-forward fashion (within each dense block). For each layer, the feature maps of all preceding layers are treated as separate inputs whereas its own feature maps are passed on as inputs to all subsequent layers. This connectivity pattern yields state-of-the-art accuracies on CIFAR10/100 (with or without data augmentation) and SVHN.
Figure 1: A dense block with 5 layers and growth rate 4.
Figure 2: A deep DenseNet with three dense blocks.
##Results The table below shows the results of DenseNets on CIFAR datasets. The "+" mark at the end denotes standard data augmentation (crop after zero-padding, and horizontal flip). For a DenseNet model, L denotes its depth and k denotes its growth rate. On CIFAR-10 and CIFAR-100 (without augmentation), Dropout with 0.2 drop rate is adopted.
Method | Parameters | CIFAR-10 | CIFAR-10+ | CIFAR-100 | CIFAR-100+ |
---|---|---|---|---|---|
DenseNet (L=40, k=12) | 1.0M | 7.00 | 5.24 | 27.55 | 24.42 |
DenseNet (L=100, k=12) | 7.0M | 5.77 | 4.10 | 23.79 | 20.20 |
DenseNet (L=100, k=24) | 27.2M | 5.83 | 3.74 | 23.42 | 19.25 |
DenseNet-BC (L=100, k=12) | 0.8M | 5.92 | 4.51 | 24.15 | 22.27 |
DenseNet-BC (L=250, k=24) | 15.3M | 5.19 | 3.62 | 19.64 | 17.60 |
DenseNet-BC (L=190, k=40) | 25.6M | - | 3.46 | - | 17.18 |
##Usage 0. Install Torch ResNet (https://github.com/facebook/fb.resnet.torch) following the instructions there. To reduce memory consumption, we recommend to install the optnet package.
- Add the file densenet.lua to the folder models/.
- Change the learning rate schedule in the file train.lua: inside function learningRate(), change line 171/173
from
decay = epoch >= 122 and 2 or epoch >= 81 and 1 or 0
todecay = epoch >= 225 and 2 or epoch >= 150 and 1 or 0
- Train a DenseNet-BC (L=100, k=12) on CIFAR-10+ using
th main.lua -netType densenet -depth 100 -dataset cifar10 -batchSize 64 -nEpochs 300 -optnet true
###Note By default, the growth rate k is set to 12, bottleneck transformation is used, compression rate at transiton layers is 0.5, dropout is disabled. To experiment with other settings, please change densenet.lua accordingly (see the comments in the code).
##Contact
liuzhuangthu at gmail.com
gh349 at cornell.edu
Any discussions, suggestions and questions are welcome!