We propose FedRec, a federated learning workflow taking advantage of unlabelled data in a semi-supervised environment to assist in the training of a supervised aggregated model. In our proposed method, an encoder architecture extracting features from unlabelled data is aggregated with the feature extractor of a classification model via weight averaging. The fully connected layers of the supervised models are also averaged in a federated fashion. We show the effectiveness of our approach by comparing it with the state-of-the-art federated algorithm, an isolated and a centralised baseline, on novel cloud detection datasets. Our code is available at https://github.com/CasellaJr/FedRec.

Federated Learning in a Semi-Supervised Environment for Earth Observation Data

Casella, Bruno
Co-first
;
Aldinucci, Marco;Battiato, Sebastiano;
2024-01-01

Abstract

We propose FedRec, a federated learning workflow taking advantage of unlabelled data in a semi-supervised environment to assist in the training of a supervised aggregated model. In our proposed method, an encoder architecture extracting features from unlabelled data is aggregated with the feature extractor of a classification model via weight averaging. The fully connected layers of the supervised models are also averaged in a federated fashion. We show the effectiveness of our approach by comparing it with the state-of-the-art federated algorithm, an isolated and a centralised baseline, on novel cloud detection datasets. Our code is available at https://github.com/CasellaJr/FedRec.
2024
European Symposium on Artificial Neural Networks
Bruges, Belgium
9-11/10/22
ESANN 2024 Proceedings - 32th European Symposium on Artificial Neural Networks, Computational Intelligence and Machine Learning
Michel Verlesian
93
98
Casella, Bruno; Chisari, Alessio Barbaro; Aldinucci, Marco; Battiato, Sebastiano; Giuffrida, Mario Valerio
File in questo prodotto:
File Dimensione Formato  
ES2024-214.pdf

Accesso aperto

Tipo di file: PDF EDITORIALE
Dimensione 1.53 MB
Formato Adobe PDF
1.53 MB Adobe PDF Visualizza/Apri

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/2318/2012810
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus ND
  • ???jsp.display-item.citation.isi??? ND
social impact