Image Classification using Federated Learning with Differential Privacy: A Comparison of Different Aggregation Algorithms
2024 (English)Independent thesis Advanced level (professional degree), 20 credits / 30 HE credits
Student thesis
Abstract [en]
The objective of this thesis was to investigate how the addition of a privacy-preserving mechanism to a federated learning model was affecting the performance of the model for an image classification task. Further, it was to get knowledge on how the outlook to use federated learning in the biotech industry is and what possible threats and attacks that could obstruct the utilization of federated learning among competitors. In the project four different aggregation algorithms for federated learning were examined. The methods were weighted fedAvg, unweighted FedAvg, weighted FedProx and unweighted FedProx. The experiment was using tensorflow federated to simulate the different methods. They were evaluated using accuracy, loss, recall, precision and F1 score. The result of this study shows that the performance of the deep neural network model is decreasing as differential privacy is introduced to the process. Out of the four aggregation algorithms used, weighted fedProx was the one that performed the best despite the added noise. It was also concluded that federated learning has potential to be used in the biotechnology industry among competitors, but that there are still security threats and attacks to avoid.
Place, publisher, year, edition, pages
2024. , p. 51
Keywords [en]
Federated Learning, Differential Privacy, weighted FedAvg, unweighted FedAvg, weighted FedProx, unweighted FedProx, Image Classification, Biotechnology
National Category
Computer Engineering
Identifiers
URN: urn:nbn:se:miun:diva-51690Local ID: DT-V23-A2-003OAI: oai:DiVA.org:miun-51690DiVA, id: diva2:1876999
Subject / course
Computer Engineering DT1
Educational program
Master of Science in Engineering - Computer Engineering TDTEA 300 higher education credits
Supervisors
Examiners
2024-06-252024-06-252024-06-25Bibliographically approved