U-Net Neural Network Optimization Method Based on Deconvolution Algorithm

U-net deep neural network has shown good performances in medical image segmentation analysis. Most of the existing works are a single use of upsampling algorithm or deconvolution algorithm in the expansion path, but they are not opposites. In this paper, we proposed a U-net network optimization strategy, in order to use the available annotation samples more effectively. One deconvolution layer and upsampling output layer were added in the splicing process of the high-resolution features in the contraction path, and then the obtained "feature map" was combined with the high-resolution features in the contraction path in the way that broaden the channel. The training data used in the experiment is the pathological section image of prostate tumor. The average Dice scores for models based on our optimization strategy improve from 0.749 to 0.813. It proves that the deconvolution algorithm can extract feature information different from the upsampling algorithm, and the complementarity can achieve a better data enhancement effect.