In image segmentation tasks based on deep learning methods
it is common that foreground pixels occur significantly more frequently than background pixels
and consequently bias the trained network towards them. In this paper
based on cost-sensitive learning
a design method of loss function for end-to-end image segmentation network is proposed
where two improvements are provided as follows: 1) Inspired by the conception of "hard examples mining"
focal loss is introduced and extended to work for ineffective learning reduction. 2) Inspired by human visual systems
adaptive weights of receptive field are added to further consider the context information. In order to verify the validity and expansibility
the proposed method has been evaluated on severalmedicalimage datasets. The results show that the proposed method can improve the detection performance of the network for small objects
and obtain segmentation results that are more suitable for object contour.
关键词
Keywords
references
RONNEBERGER O, FISCHER P, BROX T. U-Net: convolutional networks for biomedical image segmentation[C]// International Conference on MICCAI. Munich, Germany: Springer, 2015: 234-241.
ENROTH-CUGELL C, ROBSON J G. The contrast sensitivity of retinal ganglion cells of the cat[J]. The Journal of physiology, 1966, 187(3): 517-552. DOI: 517
CHAURASIA A, CULURCIELLO E. Linknet: Exploiting encoder representations for efficient semantic segmentation[C]//Visual Communications and Image Processing 2017. St. Petersburg, FL, USA: IEEE, 2017: 1-4.
CAELLES S, MANINIS K K, PONT-TUSET J, et al. One-shot video object segmentation[C]//Proceedings of the IEEE conference on CVPR. Honolulu, HI, USA: IEEE, 2017: 221-230.
LONG J, SHELHAMER E, DARRELL T. Fully convolutional networks for semantic segmentation[C]//Proceedings of the IEEE conference on CVPR.Boston, MA, USA: IEEE, 2015: 3431-3440.
WU Z, SHEN C, HENGEL A. High-performance semantic segmentation using very deep fully convolutional networks[J]. arXiv preprint arXiv: 1604.04339, 2016.
MILLETARI F, NAVAB N, AHMADI S A. V-net: Fully convolutional neural networks for volumetric medical image segmentation[C]//2016 Fourth International Conference on 3D Vision. Stanford, California, USA: IEEE, 2016: 565-571.
BADRINARAYANAN V, KENDALL A, CIPOLLA R. Segnet: A deep convolutional encoder-decoder architecture for image segmentation[J]. IEEE transactions on pattern analysis and machine intelligence, 2017, 39(12): 2481-2495. DOI: 2481
LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection[C]//Proceedings of the IEEE international conference on computer vision.Venice, Italy: IEEE, 2017: 2980-2988.
LITJENS G, TOTH R, VAN DE VEN W, et al. Evaluation of prostate segmentation algorithms for MRI: the PROMISE12 challenge[J].Medical image analysis, 2014, 18(2): 359-373. DOI: 359