Abstract
Endoscopy is a widely used imaging modality to diagnose and treat diseases in gastrointestinal tract. However, varied modalities and use of different imaging protocols at various clinical centers impose significant challenges when generalising deep learning models. Moreover, the assembly of large datasets from different clinical centers can introduce a huge label biases in multi-center studies that renders any learnt model unusable. Additionally, when using new modality or presence of images with rare pattern abnormalities such as dysplasia; a bulk amount of similar image data and their corresponding labels may not be available for training these models. In this work, we propose to use a few-shot learning approach that requires less training data and can be used to predict class labels of test samples from an unseen dataset. We propose a novel additive angular margin metric in the framework of the prototypical network in few-shot learning setting. We compare our approach to the several established methods on a large cohort of multi-center, multi-organ, multi-disease, and multi-modal gastroendoscopy data. The proposed algorithm outperforms existing state-of-the-art methods.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Ali, S., et al.: Endoscopy disease detection challenge 2020. CoRR abs/2003.03376 (2020). https://arxiv.org/abs/2003.03376
Biswas, M., et al.: State-of-the-art review on deep learning in medical imaging. Front. Biosci. 24, 392–426 (2019)
Deng, J., Guo, J., Xue, N., Zafeiriou, S.: ArcFace: additive angular margin loss for deep face recognition. In: CVPR (2019)
Dong, N., Xing, E.: Few-shot semantic segmentation with prototype learning. In: BMVC (2018)
Garcia, V., Bruna, J.: Few-shot learning with graph neural networks. In: ICLR (2018)
Liu, B., Yu, X., Yu, A., Zhang, P., Wan, G., Wang, R.: Deep few-shot learning for hyperspectral image classification. IEEE Trans. Geosci. Remote Sens. 57(4), 2290–2304 (2018)
Liu, M.Y., et al.: Few-shot unsupervised image-to-image translation. In: ICCV (2019)
Liu, W., Wen, Y., Yu, Z., Li, M., Raj, B., Song, L.: SphereFace: deep hypersphere embedding for face recognition. In: CVPR (2017)
Medela, A., et al.: Few shot learning in histopathological images: reducing the need of labeled data on biological datasets, pp. 1860–1864 (2019)
Min, J.K., Kwak, M.S., Cha, J.M.: Overview of deep learning in gastrointestinal endoscopy. Gut Liver 13, 388–393 (2019)
Pogorelov, K., et al.: KVASIR: a multi-class image dataset for computer aided gastrointestinal disease detection. In: Proceedings of the 8th ACM on Multimedia Systems Conference, MMSys 2017, pp. 164–169 (2017)
Puch, S., Sánchez, I., Rowe, M.: Few-shot learning with deep triplet networks for brain imaging modality recognition. In: Wang, Q., et al. (eds.) DART/MIL3ID -2019. LNCS, vol. 11795, pp. 181–189. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-33391-1_21
Qian, Q., Shang, L., Sun, B., Hu, J., Li, H., Jin, R.: SoftTriple loss: deep metric learning without triplet sampling. In: ICCV (2019)
Snell, J., Swersky, K., Zemel, R.: Prototypical networks for few-shot learning. In: NIPS, pp. 4077–4087 (2017)
Vinyals, O., Blundell, C., Lillicrap, T., Wierstra, D., et al.: Matching networks for one shot learning. In: NIPS, pp. 3630–3638 (2016)
Wang, Y., Yao, Q.: Few-shot learning: a survey. arXiv:1904.05046 (2019)
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, 1–34 (2019)
Zhang, C., Lin, G., Liu, F., Yao, R., Shen, C.: CANet: class-agnostic segmentation networks with iterative refinement and attentive few-shot learning. In: CVPR (2019)
Acknowledgement
S. Ali is supported by the National Institute for Health Research (NIHR) Oxford Biomedical Research Centre (BRC). The views expressed are those of the author(s) and not necessarily those of the NHS, the NIHR or the Department of Health. J. Rittscher is supported by Ludwig Institute for Cancer Research and EPSRC Seebibyte Programme Grant (EP/M0133774/1).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
1 Electronic supplementary material
Below is the link to the electronic supplementary material.
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
Ali, S., Bhattarai, B., Kim, TK., Rittscher, J. (2020). Additive Angular Margin for Few Shot Learning to Classify Clinical Endoscopy Images. In: Liu, M., Yan, P., Lian, C., Cao, X. (eds) Machine Learning in Medical Imaging. MLMI 2020. Lecture Notes in Computer Science(), vol 12436. Springer, Cham. https://doi.org/10.1007/978-3-030-59861-7_50
Download citation
DOI: https://doi.org/10.1007/978-3-030-59861-7_50
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-59860-0
Online ISBN: 978-3-030-59861-7
eBook Packages: Computer ScienceComputer Science (R0)