Abstract
The fast proliferation of the coronavirus around the globe has put several countries’ healthcare systems in danger of collapsing. As a result, locating and separating COVID-19-positive patients is a critical task. Deep learning approaches were used in several computer-aided automated systems that utilized chest computed tomography or chest X-ray images to create diagnostic tools. However, current convolutional neural network (CNN) based deep learning algorithms cannot capture the global context because of inherent image-specific inductive bias. These techniques also require large and labeled datasets to train the algorithm, but not many labeled COVID-19 datasets exist publicly. This paper proposes a Federated Learning framework with a Vision Transformer for COVID-19 detection on chest X-ray images to improve training efficiency and accuracy. The transformer architecture can exploit the unlabeled datasets using pre-training, whereas federated learning enables participating clients to jointly train models without disclosing source data outside the originating site. We experimentally establish that our proposed Vision Transformer based Federated Learning architecture outperforms CNN based centralized models. We also provide the characteristics of X-ray images of the COVID-19-affected patients. Our findings show that the proposed model can assist medical professionals in effective COVID-19 screening.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Afshar, P., Heidarian, S., Naderkhani, F., Oikonomou, A., Plataniotis, K.N., Mohammadi, A.: Covid-caps: a capsule network-based framework for identification of covid-19 cases from x-ray images. Pattern Recogn. Lett. 138, 638–643 (2020)
Arya, N., Saha, S.: Multi-modal advanced deep learning architectures for breast cancer survival prediction. Knowl.-Based Syst. 221, 106965 (2021)
Chowdhury, M.E., et al.: Can AI help in screening viral and covid-19 pneumonia? IEEE Access 8, 132665–132676 (2020)
Cohen, J.P., Morrison, P., Dao, L., Roth, K., Duong, T.Q., Ghassemi, M.: Covid-19 image data collection: prospective predictions are the future. arXiv2006.11988 (2020). github.com/ieee8023/covid-chestxray-dataset
Cordonnier, J.B., Loukas, A., Jaggi, M.: On the relationship between self-attention and convolutional layers. arXiv preprint arXiv:1911.03584 (2019)
Devlin, J., Chang, M.W., Lee, K., Toutanova, K.: BERT: pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805 (2018)
Dosovitskiy, A., et al.: An image is worth 16x16 words: transformers for image recognition at scale. arXiv preprint arXiv:2010.11929 (2020)
Dutta, P., Patra, A.P., Saha, S.: DeePROG: deep attention-based model for diseased gene prognosis by fusing multi-omics data. IEEE/ACM Trans. Comput. Biol. Bioinf. 19, 2770–2781 (2021)
Dutta, P., Saha, S., Chopra, S., Miglani, V.: Ensembling of gene clusters utilizing deep learning and protein-protein interaction information. IEEE/ACM Trans. Comput. Biol. Bioinf. 17(6), 2005–2016 (2019)
Feki, I., Ammar, S., Kessentini, Y., Muhammad, K.: Federated learning for covid-19 screening from chest x-ray images. Appl. Soft Comput. 106, 107330 (2021)
figshare: figshare.com, April 2022. http://www.figshare.com/articles/COVID-19_Chest_X-Ray_Image_Repository/12580328/
Gupta, A., Gupta, S., Katarya, R., et al.: InstaCovNet-19: a deep learning classification model for the detection of covid-19 patients using chest x-ray. Appl. Soft Comput. 99, 106859 (2021)
Hu, H., Gu, J., Zhang, Z., Dai, J., Wei, Y.: Relation networks for object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3588–3597 (2018)
Huang, L., Shea, A.L., Qian, H., Masurkar, A., Deng, H., Liu, D.: Patient clustering improves efficiency of federated machine learning to predict mortality and hospital stay time using distributed electronic medical records. J. Biomed. Inform. 99, 103291 (2019)
Irvin, J., et al.: Chexpert: a large chest radiograph dataset with uncertainty labels and expert comparison. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 33, pp. 590–597 (2019)
Islam, M.Z., Islam, M.M., Asraf, A.: A combined deep CNN-LSTM network for the detection of novel coronavirus (covid-19) using x-ray images. Inform. Med. Unlocked 20, 100412 (2020)
Japkowicz, N., Shah, M.: Evaluating Learning Algorithms: A Classification Perspective. Cambridge University Press (2011)
Kermany, D., Zhang, K., Goldbaum, M., et al.: Labeled optical coherence tomography (OCT) and chest x-ray images for classification. Mendeley Data 2(2), 651 (2018)
Lee, J., Sun, J., Wang, F., Wang, S., Jun, C.H., Jiang, X.: Privacy-preserving patient similarity learning in a federated environment: development and analysis. JMIR Med. Inform. 6(2), e7744 (2018)
Liu, B., Yan, B., Zhou, Y., Yang, Y., Zhang, Y.: Experiments of federated learning for covid-19 chest x-ray images. arXiv preprint arXiv:2007.05592 (2020)
McMahan, B., Moore, E., Ramage, D., Hampson, S., y Arcas, B.A.: Communication-efficient learning of deep networks from decentralized data. In: Artificial Intelligence and Statistics, pp. 1273–1282. PMLR (2017)
Parmar, N., et al.: Image transformer. In: International Conference on Machine Learning, pp. 4055–4064. PMLR (2018)
Rahman, T., et al.: Exploring the effect of image enhancement techniques on covid-19 detection using chest x-ray images. Comput. Biol. Med. 132, 104319 (2021)
Rousan, L.A., Elobeid, E., Karrar, M., Khader, Y.: Chest x-ray findings and temporal lung changes in patients with covid-19 pneumonia. BMC Pulm. Med. 20(1), 1–9 (2020)
Sahoo, P., Saha, S., Mondal, S., Chowdhury, S., Gowda, S.: Computer-aided covid-19 screening from chest CT-scan using a fuzzy ensemble-based technique. In: 2022 International Joint Conference on Neural Networks (IJCNN), pp. 1–8. IEEE (2022)
Singh, A., Sen, T., Saha, S., Hasanuzzaman, M.: Federated multi-task learning for complaint identification from social media data. In: Proceedings of the 32nd ACM Conference on Hypertext and Social Media, pp. 201–210 (2021)
Sun, C., Myers, A., Vondrick, C., Murphy, K., Schmid, C.: VideoBERT: a joint model for video and language representation learning. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 7464–7473 (2019)
Toraman, S., Alakus, T.B., Turkoglu, I.: Convolutional capsnet: a novel artificial neural network approach to detect covid-19 disease from x-ray images using capsule networks. Chaos Solitons Fractals 140, 110122 (2020)
Turkoglu, M.: Covidetectionet: Covid-19 diagnosis system based on x-ray images using features selected from pre-learned deep features ensemble. Appl. Intell. 51(3), 1213–1226 (2021)
Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, vol. 30 (2017)
Wang, L., Lin, Z.Q., Wong, A.: Covid-net: a tailored deep convolutional neural network design for detection of covid-19 cases from chest x-ray images. Sci. Rep. 10(1), 1–12 (2020)
Wang, Z., et al.: Automatically discriminating and localizing covid-19 from community-acquired pneumonia on chest x-rays. Pattern Recogn. 110, 107613 (2021)
ML workgroup: github-ml-workgroup (2022). github.com/ml-workgroup/covid-19-image-repository/tree/master/png/
Acknowledgement
Dr. Sriparna Saha gratefully acknowledges the Young Faculty Research Fellowship (YFRF) Award, supported by Visvesvaraya Ph.D. Scheme for Electronics and IT, Ministry of Electronics and Information Technology (MeitY), Government of India, being implemented by Digital India Corporation (formerly Media Lab Asia) for carrying out this research.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Sahoo, P., Saha, S., Mondal, S., Chowdhury, S., Gowda, S. (2023). Vision Transformer-Based Federated Learning for COVID-19 Detection Using Chest X-Ray. In: Tanveer, M., Agarwal, S., Ozawa, S., Ekbal, A., Jatowt, A. (eds) Neural Information Processing. ICONIP 2022. Communications in Computer and Information Science, vol 1794. Springer, Singapore. https://doi.org/10.1007/978-981-99-1648-1_7
Download citation
DOI: https://doi.org/10.1007/978-981-99-1648-1_7
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-1647-4
Online ISBN: 978-981-99-1648-1
eBook Packages: Computer ScienceComputer Science (R0)