Abstract
Optical coherence tomography (OCT) is a practical basis that is widely used for computer-aided retinal diagnosis, and OCT images from different devices show obvious intensity distribution differences. Recently, deep learning based models have achieved promising results for the classification tasks on single-device OCT images. However, when the models trained on source domain images are transferred for the classification of the target domain images from another OCT device, we can observe significant performance degradation. Re-annotating target domain images and fine-tuning models are inefficient. In this paper, we propose a multi-stage domain adaptation method (MSDA), which can learn generalized and effective domain invariant information for cross-device OCT classification from labeled source domain images and unlabeled target domain images. Specifically, task-independent feature alignment (TiFA) module firstly maps OCT images with different distributions to the same latent space, where the original image information is effectively preserved. Then, the downstream task-specific feature alignment (TsFA) module further distills out category-associated features from the output of TiFA. The experimental results demonstrate that the proposed MSDA improves the subretinal fluid classification performance of cross-device OCT images.
This study was supported in part by National Natural Science Foundation of China (62172223, 61671242), in part by Key R & D Program of Jiangsu Science and Technology Department (BE2018131) and the Fundamental Research Funds for the Central Universities (30921013105).
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Ben-David, S., Blitzer, J., Crammer, K., Kulesza, A., Pereira, F., Vaughan, J.W.: A theory of learning from different domains. Mach. Learn. 79(1), 151–175 (2010)
Borgwardt, K.M., Gretton, A., Rasch, M.J., Kriegel, H.P., Schölkopf, B., Smola, A.J.: Integrating structured biological data by kernel maximum mean discrepancy. Bioinformatics 22(14), e49–e57 (2006)
Duker, J.S., Waheed, N.K., Goldman, D.: Handbook of Retinal OCT: Optical Coherence Tomography E-Book. Elsevier, Amsterdam (2013)
Ghifary, M., Kleijn, W.B., Zhang, M.: Domain adaptive neural networks for object recognition. In: Pham, D.-N., Park, S.-B. (eds.) PRICAI 2014. LNCS (LNAI), vol. 8862, pp. 898–904. Springer, Cham (2014). https://doi.org/10.1007/978-3-319-13560-1_76
Goodfellow, I.J., et al.: Generative adversarial networks. arXiv preprint arXiv:1406.2661 (2014)
Hou, Y., Zheng, L.: Visualizing adapted knowledge in domain transfer. arXiv preprint arXiv:2104.10602 (2021)
Jaffe, G.J., et al.: Macular morphology and visual acuity in the comparison of age-related macular degeneration treatments trials. Ophthalmology 120(9), 1860–1870 (2013)
Johnson, M.W.: Etiology and treatment of macular edema. Am. J. Ophthalmol. 147(1), 11–21 (2009)
Li, J., Chen, E., Ding, Z., Zhu, L., Lu, K., Shen, H.T.: Maximum density divergence for domain adaptation. IEEE Trans. Pattern Anal. Mach. Intell. 43, 3918–3930 (2020)
Liu, M.Y., Breuel, T., Kautz, J.: Unsupervised image-to-image translation networks. arXiv preprint arXiv:1703.00848 (2017)
Liu, M.Y., Tuzel, O.: Coupled generative adversarial networks. arXiv preprint arXiv:1606.07536 (2016)
Long, M., Zhu, H., Wang, J., Jordan, M.I.: Unsupervised domain adaptation with residual transfer networks. arXiv preprint arXiv:1602.04433 (2016)
Long, M., Zhu, H., Wang, J., Jordan, M.I.: Deep transfer learning with joint adaptation networks. In: International Conference on Machine Learning, pp. 2208–2217. PMLR (2017)
Oulbacha, R., Kadoury, S.: MRI to CT synthesis of the lumbar spine from a pseudo-3D cycle GAN. In: 2020 IEEE 17th International Symposium on Biomedical Imaging (ISBI), pp. 1784–1787. IEEE (2020)
Romo-Bucheli, D., 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)
Saito, K., Watanabe, K., Ushiku, Y., Harada, T.: Maximum classifier discrepancy for unsupervised domain adaptation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3723–3732 (2018)
Schmidt-Erfurth, U., Klimscha, S., Waldstein, S., Bogunović, H.: A view of the current and future role of optical coherence tomography in the management of age-related macular degeneration. Eye 31(1), 26–44 (2017)
Sun, B., Feng, J., Saenko, K.: Return of frustratingly easy domain adaptation. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 30 (2016)
Tzeng, E., Hoffman, J., Saenko, K., Darrell, T.: Adversarial discriminative domain adaptation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7167–7176 (2017)
Xie, X., Chen, J., Li, Y., Shen, L., Ma, K., Zheng, Y.: Self-supervised CycleGAN for object-preserving image-to-image domain adaptation. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.-M. (eds.) ECCV 2020. LNCS, vol. 12365, pp. 498–513. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58565-5_30
Yu, F., Zhang, M., Dong, H., Hu, S., Dong, B., Zhang, L.: DAST: unsupervised domain adaptation in semantic segmentation based on discriminator attention and self-training. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, pp. 10754–10762 (2021)
Zhao, S., Li, B., Yue, X., Gu, Y., Xu, P., Hu, R., Chai, H., Keutzer, K.: Multi-source domain adaptation for semantic segmentation. arXiv preprint arXiv:1910.12181 (2019)
Zhu, J.Y., Park, T., Isola, P., Efros, A.A.: Unpaired image-to-image translation using cycle-consistent adversarial networks. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2223–2232 (2017)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 Springer Nature Switzerland AG
About this paper
Cite this paper
Li, T., Huang, K., Zhang, Y., Li, M., Zhang, W., Chen, Q. (2022). Multi-stage Domain Adaptation for Subretinal Fluid Classification in Cross-device OCT Images. In: Wallraven, C., Liu, Q., Nagahara, H. (eds) Pattern Recognition. ACPR 2021. Lecture Notes in Computer Science, vol 13188. Springer, Cham. https://doi.org/10.1007/978-3-031-02375-0_35
Download citation
DOI: https://doi.org/10.1007/978-3-031-02375-0_35
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-02374-3
Online ISBN: 978-3-031-02375-0
eBook Packages: Computer ScienceComputer Science (R0)