U-Module: Better Parameters Initialization of Convolutional Neural Network for Medical Image Classification.

Kailai Zhang,Xuesi Zhou,Ji Wu
DOI: https://doi.org/10.1109/icip.2019.8803799
2019-01-01
Abstract:In this paper, we present a novel U-module for better parameters initialization of convolutional neural network(CNN) on medical image classification tasks. CNN-based models such as VGG and Resnet have achieved great success on natural image classification tasks. However, these methods need a large number of training data to get good performance. In medical image field, the performance of the CNN-based models is always limited by the lack of annotated medical images. In this case, the parameters initialization plays an important role for the model performance. Addressing this problem, we design an up-sample structure with unsupervised loss function named U-module, which can be easily inserted into different CNN-based models. In addition, we propose a special training method for the CNN with our U-modules. Two challenging datasets of medical images are chosen for the evaluation, and the experiment results show consistent improvement on different models. Particularly, our method is even better than using the pre-train parameters on large natural image dataset.
What problem does this paper attempt to address?