Federated Learning-Empowered Disease Diagnosis Mechanism in the Internet of Medical Things: From the Privacy-Preservation Perspective

• Published in: IEEE transactions on industrial informatics Vol. 19; no. 7; pp. 7905 - 7913
• Main Authors: Wang, Xiaoding, Hu, Jia, Lin, Hui, Liu, Wenxin, Moon, Hyeonjoon, Piran, Md. Jalil
• Format: Journal Article
• Language: English
• Published: Piscataway IEEE 01.07.2023; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Artificial intelligence; Data models; Data privacy; Diagnosis; Disease diagnosis; Federated learning; federated learning (FL); Heart; Heart diseases; Hospitals; Inference; Internet of medical things; Internet of Medical Things (IoMT); Internet of Things; Medical diagnosis; Medical diagnostic imaging; Privacy; privacy protection
• ISSN: 1551-3203; 1941-0050
• DOI: 10.1109/TII.2022.3210597

The deep integration of the Internet of Things (IoT) and the medical industry has given birth to the Internet of Medical Things (IoMT). In IoMT, physicians treat a patient's disease by analyzing patient data collected through mobile devices with the assistance of an artificial intelligence (AI)-empowered systems. However, the traditional AI technologies may lead to the leakage of patient privacy data due to its own design flaws. As a privacy-preserving federated learning (FL) can generate a global disease diagnosis model through multiparty collaboration. However, FL is still unable to resist inference attacks. In this article, to address such problems, we propose a privacy-enhanced disease diagnosis mechanism using FL for IoMT. Specifically, we first reconstruct medical data through a variational autoencoder and add differential privacy noise to it to resist inference attacks. These data are then used to train local disease diagnosis models, thereby preserving patients' privacy. Furthermore, to encourage participation in FL, we propose an incentive mechanism to provide corresponding rewards to participants. Experiments are conducted on the arrhythmia database Massachusetts Institute of Technology and Beth Israel Hospital (MIT-BIH). The experimental results show that the proposed mechanism reduces the probability of reconstructing patient medical data while ensuring high-precision heart disease diagnosis.