Blockchain-Enabled Federated Learning to Enhance Security and Privacy in Internet of Medical Things (IoMT)

Document Type : Original Article

Authors

1 Department of computer engineering , Quchan University of Technology, Quchan, Iran

2 Department of computer engineering, Quchan University of Technology, quchan, Iran

Abstract

Federated learning is a distributed data analysis approach used in many IoT applications, including IoMT, due to its ability to provide acceptable accuracy and privacy. However, a critical issue with Federated learning is the poisoning attack, which has severe consequences on the accuracy of the global model caused by the server's lack of access to raw data. To deal with this problem effectively, a distributed federated learning approach involving blockchain technology is proposed. Using the consensus mechanism based on reputation-based verifier selection, verifiers are selected based on their honest participation in identifying compromised clients. This approach ensures that these clients are correctly identified and their attack is ineffective. The proposed detection mechanism can efficiently resist the data poisoning attack, which significantly improves the accuracy of the global model. Based on evaluation, the accuracy of the global model is compared with and without the proposed detection mechanism that varies with the percentage of poisonous clients and different values for the fraction of poisonous data. In addition to the stable accuracy range of nearly 93%, the accuracy of our proposed detection mechanism is not affected by the increase of α in different values of β.

Keywords

Main Subjects


  • Li, A. K. Sahu, A. Talwalkar, and V. Smith, “Federated learning: Challenges methods and future directions,” IEEE Signal Process. Mag., vol. 37, no. 3, pp. 50-60, May 2020. https://doi.org/10.1109/MSP.2020.2975749.
  • C. Nguyen, Q. V. Pham, P. N. Pathirana, M. Ding, A. Seneviratne, Z. Lin, O. Dobre, and W. J. Hwang, “Federated Learning for Smart Healthcare: A Survey,” ACM Comput. Surv., vol. 55, no. 3, pp. 1-37, 2022. https://doi.org/10.1145/3501296
  • Suciu, R. Marginean, Y. Kaya, H. Daume III, and T. Dumitras, “Wen does machine learning fail? Generalized transferability for evasion and poisoning attacks,” Proc. USENIX Conference on Security Symposium (USENIX Security’18), 2018 , pp. 1299–1316.
  • Kairouz, H. B. McMahan, B. Avent, A. Bellet, M. Bennis, A. N. Bhagoji, and S. Zhao, “Advances and open problems in federated learning,” Foundations and Trends® in Machine Learning, vol. 14, no.1–2, pp. 1-210, 2019. http://dx.doi.org/10.1561/2200000083
  • Baracaldo, B. Chen, H. Ludwig, and J. A. Safavi, “Mitigating poisoning attacks on machine learning models: A. data provenance based approach,” Proc. ACM 10th ACM Workshop Artif. Intell. Secur., 2017, pp. 103–110. https://doi.org/10.1145/3128572.3140450
  • Yin, Y. Chen, R. Kannan, and P. Bartlett, “Byzantine-robust distributed learning: Towards optimal statistical rates,” International Conference on Machine Learning, 2018, pp. 5650–5659. https://proceedings.mlr.press/v80/yin18a.html.
  • Zhang, C. Ge, F. Hu and B. Chen, “RobustFL: Robust Federated Learning Against Poisoning Attacks in Industrial IoT Systems,” IEEE Transactions on Industrial Informatics, vol. 18, no. 9, pp. 6388-6397, Sept. 2022, https://doi.org/10.1109/TII.2021.3132954
  • Andreina, G. A. Marson, H. Mollering, and G. Karame, “BaFFLe: Backdoor detection via feedback-based federated learning,” 2021 IEEE 41st International Conference on Distributed Computing Systems (ICDCS), DC, USA, 2021, pp. 852-863, https://doi.org/10.1109/ICDCS51616.2021.00086.
  • Y Zh.ao, J. Chen, J. Zhang, D. Wu, M. Blumenstein, and S. Yu, “Detecting and mitigating poisoning attacks in federated learning using generative adversarial networks,” Comput. Pract. Exp. Vol. 34, no. 7, e5906, 2022. https://doi.org/10.1002/cpe.5906
  • A. Monrat, O. Schelén and K. Andersson, “A Survey of Blockchain from the Perspectives of Applications, Challenges, and Opportunities,” IEEE Access, vol. 7, pp. 117134-117151, 2019, https://doi.org/10.1109/ACCESS.2019.2936094.
  • N. Dai, Z. Zheng and Y. Zhang, “Blockchain for Internet of Things: A Survey,” IEEE Internet of Things Journal, vol. 6, no. 5, pp. 8076-8094, Oct. 2019, https://doi.org/10.1109/JIOT.2019.2920987.
  • Wang, X. Zha, W. Ni, R. P. Liu, Y. J. Guo, X. Niu, and K. Zheng, “Survey on blockchain for Internet of Things,” Computer Communications, vol. 136, pp. 10-29, 2019, https://doi.org/10.1016/j.comcom.2019.01.006.
  • A. Uddin, A. Stranieri, I. Gondal, and V. Balasubramanian, “A survey on the adoption of blockchain in IoT: challenges and solutions,” Blockchain: Research and Applications, Vol. 2, no. 2, 100006, 2021. https://doi.org/10.1016/j.bcra.2021.100006.
  • Bagdasaryan, A. Veit, Y. Hua, D. Estrin, and V. Shmatikov, “How to backdoor federated learning,” Proceedings of the 23th International Conference on Artificial Intelligence and Statistics. PMLR, Palermo, Sicily, Italy, 2020, pp. 2938-2948. https://proceedings.mlr.press/v108/bagdasaryan20a.html.
  • Tolpegin, S. Truex, M. E. Gursoy, G. Mehmet Emre, and L. Liu, “Data poisoning attacks against federated learning systems,”Computer Security – ESORICS 2020. ESORICS 2020, Springer, Cham, 2020, pp. 480-50. https://doi.org/10.1007/978-3-030-58951-6_24
  • Qu, L. Gao, T. H., Luan, Y. Xiang, S. Yu, B. Li, G. Zheng, “Decentralized Privacy Using Blockchain-Enabled Federated Learning in Fog Computing,” IEEE Internet of Things Journal, vol. 7, no. 6, pp. 5171-5183. https://doi.org/10.1109/JIOT.2020.2977383.
  • T. de Oliveira, L. H. Reis, D. S. Medeiros, R. C. Carrano, S. D. Olabarriaga, and D. M. Mattos, “Blockchain reputation-based consensus: A scalable and resilient mechanism for distributed mistrusting applications,” Computer Networks, vol. 179, p. 107367, 2020. https://doi.org/10.1016/j.comnet.2020.107367.
  • http://www.kaggle.com/kumar012/hypothyroid.
  • Ma, J. Ma, Y. Miao, X. Liu, K. K. R. Choo, R. H. Deng, “Pocket Diagnosis: Secure Federated Learning Against Poisoning Attack in the Cloud,” IEEE Transactions on Services Computing, vol. 15, no. 6, pp. 3429-3442, 2021. https://doi.org/10.1109/TSC.2021.3090771.
  • Liu, H. Lin, X. Wang, J. Hu, G. Kaddoum, M. J. Piran, and A. Alamri, “D2MIF: A Malicious Model Detection Mechanism for Federated Learning Empowered Artificial Intelligence of Things,” IEEE Internet of Things Journal, vol. 10, no. 3, pp. 2141-2151, 2023. https://doi.org/10.1109/JIOT.2021.3081606.
  • N. Bhagoji, S. Chakraborty, P. Mittal, and S. Calo, “Analyzing federated learning through an adversarial lens,” International Conference on Machine Learning, 2018, pp. 634-643. https://proceedings.mlr.press/v97/bhagoji19a.html.
  • Jagielski, A. Oprea, B. Biggio, C. Liu, C. Nita-Rotaru, and B. Li, “Manipulating machine learning: Poisoning attacks and countermeasures for regression learning,” Proc. IEEE Symposium on Security and Privacy (S&P’18), IEEE, 2018, pp. 19–35. https://doi.org/10.1109/SP.2018.00057.
  • Barbieri, S. Savazzi, M. Brambilla, and M. Nicoli, “Decentralized federated learning for extended sensing in 6G connected vehicles,” Vehicular Communications, vol. 33, p. 100396, 2022. https://doi.org/10.1016/j.vehcom.2021.100396.
  • Połap, G. Srivastava, and K.Yu, “Agent architecture of an intelligent medical system based on federated learning and blockchain technology,” Journal of Information Security and Applications (JISA) vol. 58, p. 102748, 2021. https://doi.org/10.1016/j.jisa.2021.102748
  • Kumar, A. A. Khan, J. Kumar, N. A. Golilarz, S. Zhang, Y. Ting, C. Zheng, W. Wang, “Blockchain-federated-learning and deep learning models for covid-19 detection using CT imaging, IEEE Sensors Journal, vol. 21, no. 14, pp. 16301–16314, 2021. https://doi.org/10.1109/JSEN.2021.3076767.
  • A. Rahman, M. S. Hossain, M. S. Islam, N. A. Alrajeh, and G. Muhammad, “Secure and provenance enhanced internet of health things framework: A blockchain managed federated learning approach,” IEEE Access, vol. 8, pp. 205071–205087, 2020. https://doi.org/10.1109/ACCESS.2020.3037474
  • Savazzi, M. Nicoli and V. Rampa, “Federated Learning with Cooperating Devices: A Consensus Approach for Massive IoT Networks,” IEEE Internet of Things Journal, vol. 7, no. 5, pp. 4641-4654, May 2020, https://doi.org/10.1109/JIOT.2020.2964162.
  • Jin, X. Dai, J. Xiao, B. Li, H. Li and Y. Zhang, “Cross-Cluster Federated Learning and Blockchain for Internet of Medical Things,” IEEE Internet of Things Journal, vol. 8, no. 21, pp. 15776-15784, 1 Nov.1, 2021, https://doi.org/10.1109/JIOT.2021.3081578.