Abstract
Convolutional neural network (CNN) has expanded rapidly, and has been widely used in medical image classification. The large number of parameters in a neural network makes CNN models computationally expensive. This leads to slow inference speed, especially for 3D data such as optical coherence tomography (OCT) for retinal images. A volume OCT scan of retina often contains hundreds of 2D images which needs to be analyzed sequentially in a local computer with limited computational resources. We introduce network pruning to OCT images classification and propose an algorithm to prune networks. We compress the popular classification models, such as ResNet and VGG. For example, within 1% accuracy loss, we compress ResNet-18 from 44.8 MB to 69 KB and VGG-16 from 537.1 MB to 194 KB. These pruned models are much smaller and easier to deploy on the OCT devices. As for the inference speed, the pruned models are 10 to 20 times faster than original models for ResNet and VGG in CPU.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Cheng, J., Wang, P., Li, G., Hu, Q., Lu, H.: Recent advances in efficient computation of deep convolutional neural networks. Front. Inf. Technol. Electron. Eng. 19(1), 64–77 (2018)
LeCun, Y.: Optimal brain damage. In: International Conference on Neural Information Processing Systems (1989)
Denton, E., Zaremba, W., Bruna, J., LeCun, Y., Fergus, R.: Exploiting linear structure within convolutional networks for efficient evaluation. In: Proceedings of the 27th International Conference on Neural Information Processing Systems, NIPS 2014, vol. 1, pp. 1269–1277. MIT Press, Cambridge, MA, USA (2014). http://dl.acm.org/citation.cfm?id=2968826.2968968
Frankle, J., Carbin, M.: The lottery ticket hypothesis: finding sparse, trainable neural networks. In: ICLR (2019)
He, K., Zhang, X., Ren, S., Sun, J.: Delving deep into rectifiers: surpassing human-level performance on imagenet classification. In: Proceedings of the IEEE International Conference on Computer Vision 2015 Inter, pp. 1026–1034 (2015). https://doi.org/10.1109/ICCV.2015.123
Hee, M.R., et al.: Optical coherence tomography of the human retina. Arch. Ophthalmol. 113(3), 325–332 (1995)
Kermany, D.S., et al.: Identifying medical diagnoses and treatable diseases by image-based deep learning. Cell 172(5), 1122–1131.e9 (2018)
Ba, J., Caurana, R.: Do deep nets really need to be deep ? In: Advances in Neural Information Processing Systems, pp. 2654–2662 (2013)
Liu, Z., Sun, M., Zhou, T., Huang, G., Darrell, T.: Rethinking the value of network pruning. In: ICLR (2019)
Mao, H., et al.: Exploring the granularity of sparsity in convolutional neural networks. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), pp. 1927–1934, July 2017. https://doi.org/10.1109/CVPRW.2017.241
Perdomo, O., Otálora, S., González, F.A., Meriaudeau, F., Müller, H.: OCT-NET: a convolutional network for automatic classification of normal and diabetic macular edema using sd-oct volumes. In: 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018), pp. 1423–1426, April 2018. https://doi.org/10.1109/ISBI.2018.8363839
Qiu, J., Sun, Y.: Self-supervised iterative refinement learning for macular OCT volumetric data classification. Comput. Biol. Med. 103327 (2019). https://doi.org/10.1016/j.compbiomed.2019.103327
Selvaraju, R.R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., Batra, D.: Grad-CAM: visual explanations from deep networks via gradient-based localization. In: Proceedings of the IEEE International Conference on Computer Vision (2016). https://doi.org/10.1109/ICCV.2017.74
Zhao, H., Shi, J., Qi, X., Wang, X., Jia, J.: Pyramid scene parsing network. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 6230–6239, July 2017. https://doi.org/10.1109/CVPR.2017.660
Acknowledgment
This work was supported National Natural Science Foundation of China (61601029, 61602322), Grant of Ningbo 3315 Innovation Team, and China Association for Science and Technology (2016QNRC001).
Author information
Authors and Affiliations
Corresponding authors
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Yang, B. et al. (2019). Network Pruning for OCT Image Classification. In: Fu, H., Garvin, M., MacGillivray, T., Xu, Y., Zheng, Y. (eds) Ophthalmic Medical Image Analysis. OMIA 2019. Lecture Notes in Computer Science(), vol 11855. Springer, Cham. https://doi.org/10.1007/978-3-030-32956-3_15
Download citation
DOI: https://doi.org/10.1007/978-3-030-32956-3_15
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-32955-6
Online ISBN: 978-3-030-32956-3
eBook Packages: Computer ScienceComputer Science (R0)