Abstract
Segmentation of optical coherence tomography (OCT) images of retinal tissue has become an important task for the diagnosis and management of eye diseases. Deep convolutional neural networks have shown great success in retinal image segmentation. However, a well-trained deep learning model on OCT images from one device often fail when it is deployed on images from a different device since these images have different data distributions. Unsupervised domain adaptation (UDA) can solve the above problem by aligning the data distribution between labeled data (source domain) and unlabeled data (target domain). In this paper, we propose an UDA adversarial learning framework with self-selected active learning. The framework consists of two parts: domain adaptation module (DAM) and self-selected active learning module (SALM). The DAM learns domain-invariant features (i.e., common features) gradually to narrow the distribution discrepancy between two domains. The SALM introduces the target data into source domain through discrepancy method and similarity method, which promotes the DAM to learn unique features of target domain. Extensive experiments show the effectiveness of our method. Compared with the state-of-the-art UDA methods, our method has achieved better performance on two medical cross-domain datasets.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Mahapatra, D., Bozorgtabar, B., et al.: Pathological retinal region segmentation from oct images using geometric relation based augmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 9611–9620 (2020)
Tzeng, E., Hoffman, J., et al.: Deep domain confusion: maximizing for domain invariance. CoRR abs/1412.3474 (2014)
Long, M., Cao, Y., et al.: Learning transferable features with deep adaptation networks. In: Proceedings of the 32nd International Conference on Machine Learning. JMLR, Lille, France, pp. 97–105 (2015)
Sun, B., Saenko, K.: Deep CORAL: correlation alignment for deep domain adaptation. In: Hua, G., Jégou, H. (eds.) ECCV 2016. LNCS, vol. 9915, pp. 443–450. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-49409-8_35
Shen, J., Qu, Y., et al.: Wasserstein distance guided representation learning for domain adaptation. In: Proceedings of the Thirty-Second Conference on Artificial Intelligence. AAAI, New Orleans, Louisiana, USA, pp. 4058–4065 (2018)
Goodfellow, I.J., Pouget, A.J., et al.: Generative adversarial nets. In: Proceedings of Annual Conference on Neural Information Processing Systems. ACM, Montreal, Quebec, Canada, pp. 2672–2680 (2014)
Philipp, S., David, R., et al.: Using CycleGANs for effectively reducing image variability across OCT devices and improving retinal fluid segmentation. In: Proceedings of the IEEE International Symposium on Biomedical Imaging, pp. 605–609 (2019)
David, R., Philipp, S., et al.: Reducing image variability across OCT devices with unsupervised unpaired learning for improved segmentation of retina. Biomed. Opt. Express 11(1), 346–363 (2020)
Wang, J., Chen, Y.W., et al.: Domain adaptation model for retinopathy detection from cross-domain OCT images. In: Proceedings of the Third Conference on Medical Imaging with Deep Learning, in Proceedings of Machine Learning Research, vol. 121, pp. 795–810 (2020)
Wang, S., Yu, L., Li, K., Yang, X., Fu, C.-W., Heng, P.-A.: Boundary and entropy-driven adversarial learning for fundus image segmentation. In: Shen, D., et al. (eds.) MICCAI 2019. LNCS, vol. 11764, pp. 102–110. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32239-7_12
Vu, T.H., Jain, H., et al.: ADVENT: adversarial entropy minimization for domain adaptation in semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (2019)
Chen C., Dou Q., et al.: Synergistic image and feature adaptation: towards cross-modality domain adaptation for medical image segmentation. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 33, no. 01, pp. 865–872 (2019)
Wang, S., Yu, L., et al.: Patch-Based output space adversarial learning for joint optic disc and cup segmentation. IEEE Trans. Med. Imag. 38(11), 2485–2495 (2019)
Wang, H., Shen, T., Zhang, W., Duan, L.-Y., Mei, T.: Classes matter: a fine-grained adversarial approach to cross-domain semantic segmentation. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.-M. (eds.) ECCV 2020. LNCS, vol. 12359, pp. 642–659. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58568-6_38
Dou, Q., Chen, O.Y., et al.: Unsupervised cross-modality domain adaptation of ConvNets for biomedical image segmentations with adversarial loss. In: International Joint Conference on Artificial Intelligence (2018)
Hrvoje, B., Freerk, V., et al.: RETOUCH-the retinal OCT fluid detection and segmentation benchmark and challenge. IEEE Trans. Med. Imag. 38(8), 1858–1874 (2019)
Acknowledgments
This work is supported by the National Natural Science Foundation of China under Grant No. 61701192, No. 61872419, No. 61873324, the Natural Science Foundation of Shandong Province, China, under Grant No. ZR2020QF107, No. ZR2020MF137, No. ZR2019MF040, No. ZR2019MH106, No. ZR2018BF023, the China Postdoctoral Science Foundation under Grants No. 2017M612178. University Innovation Team Project of Jinan (2019GXRC015), Key Science & Technology Innovation Project of Shandong Province (2019JZZY010324, 2019JZZY010448), and the Higher Educational Science and Technology Program of Jinan City under Grant with No. 2020GXRC057. The National Key Research and Development Program of China (No. 2016YFC13055004).
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
Li, X., Niu, S., Gao, X., Liu, T., Dong, J. (2021). Unsupervised Domain Adaptation with Self-selected Active Learning for Cross-domain OCT Image Segmentation. In: Mantoro, T., Lee, M., Ayu, M.A., Wong, K.W., Hidayanto, A.N. (eds) Neural Information Processing. ICONIP 2021. Lecture Notes in Computer Science(), vol 13109. Springer, Cham. https://doi.org/10.1007/978-3-030-92270-2_50
Download citation
DOI: https://doi.org/10.1007/978-3-030-92270-2_50
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-92269-6
Online ISBN: 978-3-030-92270-2
eBook Packages: Computer ScienceComputer Science (R0)