Abstract
The medical image is often noisy, which makes it difficult to extract the image features from the medical image segmentation model. Because the noise is often generated randomly, it is difficult to use supervised information for denoising. In this paper, we focus on this challenging problem and propose an IM-Net algorithm for medical image segmentation based on mutual information maximization. The IM-Net can remove the noise and therefore improve the quality of the extracted feature by maximizing the mutual information between the extracted feature and the input image. IM-Net uses the Binary Cross Entropy with Logits Estimation to approach the true value of mutual information and uses a bilinear interpolation function as a discriminator to maximize the mutual information estimator. Extensive experiments are conducted and the IM-Net is compared with different methods to demonstrate the effectiveness of our model. Experimental results show that the training efficiency and segmentation precision are greatly improved.
This work was supported in part by National Natural Science Foundation of China (No. 61806097); College Students Entrepreneurship Training Program 2019 (No. S20190025). The study was funded by Shanghai Science and Technology Committee (17511104200, 18411952100, 17411953500).
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Shen, D., Ip, H.H.S.: Discriminative wavelet shape descriptors for recognition of 2-D patterns. Pattern Recogn. 32(2), 151–165 (1999)
Tamura, H., Mori, S., Yamawaki, T.: Textural features corresponding to visual perception. IEEE Trans. Syst. Man Cybern. 8(6), 460–473 (1978)
Hong, Z.-Q.: Algebraic feature extraction of image for recognition. Pattern Recogn. 24(3), 211–219 (1991)
Goupillaud, P., Grossmann, A., Morlet, J.: Cycle-octave and related transforms in seismic signal analysis. Geoexploration 23(1), 85–102 (1984)
Hjelm, R.D., et al.: Learning deep representations by mutual information estimation and maximization. arXiv preprint arXiv:1808.06670 (2018)
Yang, S., Ramanan, D.: Multi-scale recognition with DAG-CNNs. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1215–1223 (2015)
Herranz, L., Jiang, S., Li, X.: Scene recognition with CNNs: objects, scales and dataset bias. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 571–579 (2016)
Gao, H., Ji, S.: Graph u-nets. In: Proceedings of the 36th International Conference on Machine Learning, ICML 2019, vol. 97, pp. 2083–2092. PMLR (2019)
Yoo, D., Park, S., Lee, J.-Y., Kweon, I.S.: Multi-scale pyramid pooling for deep convolutional representation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, pp. 71–80 (2015)
Chen, C., Liu, X., Ding, M., Zheng, J., Li, J.: 3D dilated multi-fiber network for real-time brain tumor segmentation in MRI. In: Shen, D., et al. (eds.) MICCAI 2019. LNCS, vol. 11766, pp. 184–192. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32248-9_21
Huang, C., Han, H., Yao, Q., Zhu, S., Zhou, S.K.: 3D U\(^2\)-net: a 3D universal U-net for multi-domain medical image segmentation. In: Shen, D., Liu, T., Peters, T.M., Staib, L.H., Essert, C., Zhou, S., Yap, P.-T., Khan, A. (eds.) MICCAI 2019. LNCS, vol. 11765, pp. 291–299. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32245-8_33
Brügger, R., Baumgartner, C.F., Konukoglu, E.: A partially reversible U-net for memory-efficient volumetric image segmentation. In: Shen, D., Liu, T., Peters, T.M., Staib, L.H., Essert, C., Zhou, S., Yap, P.-T., Khan, A. (eds.) MICCAI 2019. LNCS, vol. 11766, pp. 429–437. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32248-9_48
Orlando, J.I., et al.: U2-net: a Bayesian U-net model with epistemic uncertainty feedback for photoreceptor layer segmentation in pathological OCT scans. In: 16th IEEE International Symposium on Biomedical Imaging, ISBI 2019, pp. 1441–1445. IEEE (2019)
Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28
Zhou, Z., Rahman Siddiquee, M.M., Tajbakhsh, N., Liang, J.: UNet++: a nested U-net architecture for medical image segmentation. In: Stoyanov, D., et al. (eds.) DLMIA/ML-CDS -2018. LNCS, vol. 11045, pp. 3–11. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-00889-5_1
Milletari, F., Navab, N., Ahmadi, S.: V-net: fully convolutional neural networks for volumetric medical image segmentation. In: 2016 Fourth International Conference on 3D Vision (3DV), pp. 565–571, October 2016
Oktay, O., et al.: Attention U-net: learning where to look for the pancreas. arXiv preprint arXiv:1804.03999, 2018
Jin, Q., Meng, Z., Sun, C., Wei, L., Su, R.: RA-UNET: a hybrid deep attention-aware network to extract liver and tumor in CT scans. arXiv preprint arXiv:1811.01328 (2018)
Belghazi, M.I., et al.: Mutual information neural estimation. International Conference on Machine Learning, pp. 531–540 (2018)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
Sun, Y., Yuan, P. (2020). IM-Net: Semantic Segmentation Algorithm for Medical Images Based on Mutual Information Maximization. In: Li, G., Shen, H., Yuan, Y., Wang, X., Liu, H., Zhao, X. (eds) Knowledge Science, Engineering and Management. KSEM 2020. Lecture Notes in Computer Science(), vol 12274. Springer, Cham. https://doi.org/10.1007/978-3-030-55130-8_35
Download citation
DOI: https://doi.org/10.1007/978-3-030-55130-8_35
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-55129-2
Online ISBN: 978-3-030-55130-8
eBook Packages: Computer ScienceComputer Science (R0)