Abstract
This project aims to recognize a group of rare retinal diseases, the hereditary macular dystrophies, based on Optical Coherence Tomography (OCT) images, whose primary manifestation is the interruption, disruption, and loss of the layers of the retina. The challenge of using machine learning models to recognize those diseases arises from the limited number of collected images due to their rareness. We formulate the problems caused by lacking labeled data as a Student-Teacher learning task with a discriminative feature space and knowledge distillation (KD). OCT images have large variations due to different types of macular structural changes, capturing devices, and angles. To alleviate such issues, a pipeline of preprocessing is first utilized for image alignment. Tissue images at different angles can be roughly calibrated to a horizontal state for better feature representation. Extensive experiments on our dataset demonstrate the effectiveness of the proposed approach.
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Anantrasirichai, N., Achim, A., Morgan, J.E., Erchova, I., Nicholson, L.: SVM-based texture classification in Optical Coherence Tomography. In: 2013 IEEE 10th International Symposium on Biomedical Imaging, San Francisco, CA, USA, pp. 1332–1335. IEEE (2013)
Chen, X., He, K.: Exploring simple Siamese representation learning. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 15750–15758 (2021)
Dabov, K., Foi, A., Katkovnik, V., Egiazarian, K.: Image denoising by sparse 3-D transform-domain collaborative filtering. IEEE Trans. Image Proces. 16(8), 2080–2095 (2007)
Fang, L., Wang, C., Li, S., Yan, J., Chen, X., Rabbani, H.: Automatic classification of retinal three-dimensional optical coherence tomography images using principal component analysis network with composite kernels. J. Biomed. Opt. 22(11), 116011 (2017)
Frosst, N., Papernot, N., Hinton, G.: Analyzing and improving representations with the soft nearest neighbor loss. In: International Conference on Machine Learning, pp. 2012–2020. PMLR (2019)
Hinton, G., Vinyals, O., Dean, J.: Distilling the knowledge in a neural network. stat 1050, 9 (2015)
Hussain, M.A., et al.: Classification of healthy and diseased retina using SD-OCT imaging and random forest algorithm. PloS One 13(6), e0198281 (2018)
Ji, Q., He, W., Huang, J., Sun, Y.: Efficient deep learning-based automated pathology identification in retinal optical coherence tomography images. Algorithms 11(6), 88 (2018)
Ji, Q., Huang, J., He, W., Sun, Y.: Optimized deep convolutional neural networks for identification of macular diseases from optical coherence tomography images. Algorithms 12(3), 51 (2019)
Karri, S.P.K., Chakraborty, D., Chatterjee, J.: Transfer learning based classification of optical coherence tomography images with diabetic macular edema and dry age-related macular degeneration. Biomed. Opt. Express 8(2), 579–592 (2017)
Kermany, D., Zhang, K., Goldbaum, M., et al.: Labeled optical coherence tomography (OCT) and chest X-Ray images for classification (2018)
Liu, Y.Y., Chen, M., Ishikawa, H., Wollstein, G., Schuman, J.S., Rehg, J.M.: Automated macular pathology diagnosis in retinal OCT images using multi-scale spatial pyramid and local binary patterns in texture and shape encoding. Med. Image Anal. 15(5), 748–759 (2011)
Lu, W., Tong, Y., Yu, Y., Xing, Y., Chen, C., Shen, Y.: Deep learning-based automated classification of multi-categorical abnormalities from optical coherence tomography images. Transl. Vis. Sci. Technol. 7(6), 41 (2018)
Motozawa, N., et al.: Optical coherence tomography-based deep-learning models for classifying normal and age-related macular degeneration and exudative and non-exudative age-related macular degeneration changes. Ophthalmol. Ther. 8(4), 527–539 (2019)
Sankar, S., et al.: Classification of SD-OCT volumes for DME detection: an anomaly detection approach. In: Medical Imaging 2016: Computer-Aided Diagnosis, vol. 9785, pp. 97852O. International Society for Optics and Photonics (2016)
Shih, F.Y., Patel, H.: Deep learning classification on optical coherence tomography retina images. Int. J. Pattern Recogn. Artif. Intell. 34(08), 2052002 (2020)
Srinivasan, P.P., et al.: Fully automated detection of diabetic macular edema and dry age-related macular degeneration from optical coherence tomography images. Biomed. Opt. Express 5(10), 3568–3577 (2014)
Sugmk, J., Kiattisin, S., Leelasantitham, A.: Automated classification between age-related macular degeneration and Diabetic macular edema in OCT image using image segmentation. In: The 7th 2014 Biomedical Engineering International Conference, pp. 1–4. IEEE (2014)
Sun, Y., Li, S., Sun, Z.: Fully automated macular pathology detection in retina optical coherence tomography images using sparse coding and dictionary learning. J. Biomed. Opt. 22(1), 016012 (2017)
Tian, Y., Krishnan, D., Isola, P.: Contrastive representation distillation. In: International Conference on Learning Representations (2019)
Venhuizen, F.G., et al.: Automated age-related macular degeneration classification in OCT using unsupervised feature learning. In: Hadjiiski, L.M., Tourassi, G.D. (eds.) SPIE Medical Imaging, Orlando, Florida, USA, pp. 94141I (2015)
Wang, J., et al.: Deep learning for quality assessment of retinal OCT images. Biomed. Opt. Express 10(12), 6057–6072 (2019)
Wang, L., Yoon, K.-J.: Knowledge distillation and student-teacher learning for visual intelligence: a review and new outlooks. IEEE Trans. Pattern Anal. Mach. Intell., 1 (2021). https://doi.org/10.1109/TPAMI.2021.3055564
Wang, Y., Yao, Q., Kwok, J.T., Ni, L.M.: Generalizing from a few examples: a survey on few-shot learning. ACM Comput. Surv. (CSUR) 53(3), 1–34 (2020)
Acknowledgments
This research is partially funded by NSF-IIS-1910492.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
Cite this paper
Mai, S., Li, Q., Zhao, Q., Gao, M. (2021). Few-Shot Transfer Learning for Hereditary Retinal Diseases Recognition. In: de Bruijne, M., et al. Medical Image Computing and Computer Assisted Intervention – MICCAI 2021. MICCAI 2021. Lecture Notes in Computer Science(), vol 12908. Springer, Cham. https://doi.org/10.1007/978-3-030-87237-3_10
Download citation
DOI: https://doi.org/10.1007/978-3-030-87237-3_10
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-87236-6
Online ISBN: 978-3-030-87237-3
eBook Packages: Computer ScienceComputer Science (R0)