Federated Self-supervised Learning in Computer Vision
2023 (English)Independent thesis Advanced level (degree of Master (Two Years)), 20 credits / 30 HE credits
Student thesis
Abstract [en]
With an ever-increasing amount of available image data, self-supervised learning (SSL) circumvents the necessity for annotations in traditional supervised learning methods. SSL methods such as SimSiam have shown excellent results on popular benchmark datasets, even outperforming supervised methods. Federated learning, on the other hand, is transforming traditional centralized training of machine learning models to distributed training on edge devices (e.g., phones, self-driving cars). Data stays on devices, and only model updates are shared with a central server. This reduces data migration costs and preserves data privacy. Even though edge devices collect massive amounts of image data, it cannot be labeled due to missing interfaces, expertise, or the reluctance of users to label their own private data. SSL can address this label deficiency. While SSL has achieved impressive results in settings where data is stored centrally, its performance in federated settings remains relatively unexplored. The present study investigates federated self-supervised learning (FSSL) in the domain of computer vision. First, four FSSL methods based on SimSiam are implemented and evaluated on CIFAR-10. This work proposes FedAlign, which outperforms FedSimSiam in linear evaluation and achieves highest accuracies in IID settings. Results further show that Hetero-SSL performs best in non- IID settings. Additionally, FSSL is applied to an autonomous driving dataset with the aim of detecting cars. The findings show that the weights obtained through FSSL can closely match supervised ImageNet weights, without the need for expensive annotations. The results highlight the potential of FSSL methods in real- life applications.
Place, publisher, year, edition, pages
2023. , p. 50
Series
IT ; 23073
National Category
Engineering and Technology
Identifiers
URN: urn:nbn:se:uu:diva-512503OAI: oai:DiVA.org:uu-512503DiVA, id: diva2:1800393
Educational program
Master's Programme in Data Science
Supervisors
Examiners
2023-09-262023-09-262023-09-26Bibliographically approved