Hybrid Makes Better: Privacy-Preserving Medical Image Classification Based on Federated Learning

Zhi Wang, Haotian Chi, Yonggang Li, Yuxuan Zhang, Shunrong Jiang, Xiaojiang Du, Mohsen Guizani

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

1 Scopus citations

Abstract

The power of machine learning makes it available for medical image classification. However, machine learning requires large medical datasets to improve accuracy, which will involve patients' private information and lead to their privacy leakage. Federated learning is a trending technique to both protect patients' privacy and improve the accuracy of medical image classification. Unfortunately, current research shows that federated learning faces the risk of privacy leakage. In this paper, we propose a privacy-preserving scheme FHDM. Specifically, we construct a local hybrid differential privacy algorithm (LHDP) against patients' privacy leakage. This hybrid algorithm utilizes both Gaussian and Laplace differential privacy without enlarging the privacy budget. We prove that the algorithm applies to the model parameter. Moreover, we design loss optimization and global optimization strategies on the algorithm to achieve higher accuracy in medical image classification. Finally, we validate FHDM in terms of privacy-preserving and model accuracy on real datasets. Experiments show that FHDM effectively protects privacy and improves the average accuracy by 10.0% compared to adopting the LDP-based scheme with the same medical image dataset and privacy budget.

Original languageEnglish
Title of host publicationGLOBECOM 2024 - 2024 IEEE Global Communications Conference
Pages2816-2821
Number of pages6
ISBN (Electronic)9798350351255
DOIs
StatePublished - 2024
Event2024 IEEE Global Communications Conference, GLOBECOM 2024 - Cape Town, South Africa
Duration: 8 Dec 202412 Dec 2024

Publication series

NameProceedings - IEEE Global Communications Conference, GLOBECOM
ISSN (Print)2334-0983
ISSN (Electronic)2576-6813

Conference

Conference2024 IEEE Global Communications Conference, GLOBECOM 2024
Country/TerritorySouth Africa
CityCape Town
Period8/12/2412/12/24

Keywords

  • Differential privacy
  • Federated learning
  • Image classification
  • Optimization

Fingerprint

Dive into the research topics of 'Hybrid Makes Better: Privacy-Preserving Medical Image Classification Based on Federated Learning'. Together they form a unique fingerprint.

Cite this