In Defense of Fully Connected Layers in Visual Representation Transfer.

Chen-Lin Zhang,Jian-Hao Luo,Xiu-Shen Wei,Jianxin Wu
DOI: https://doi.org/10.1007/978-3-319-77383-4_79
2017-01-01
Abstract:Pre-trained convolutional neural network (CNN) models have been widely applied in many computer vision tasks, especially in transfer learning tasks. In transfer learning, the target domain may be in a different feature space or follow a different data distribution, compared to the source domain. In CNN transfer tasks, we often transfer visual representations from a source domain (e.g., ImageNet) to target domains with fewer training images or have different image properties. It is natural to explore which CNN model performs better in visual representation transfer. Through visualization analyses and extensive experiments, we show that when either image properties or task objective in the target domain is far away from those in the source domain, having the fully connected layers in the source domain pre-trained model is essential in achieving high accuracy after transferring to the target domain.
What problem does this paper attempt to address?