A federated learning approach for thermal comfort management - Université de technologie de Troyes Accéder directement au contenu
Article Dans Une Revue Advanced Engineering Informatics Année : 2022

A federated learning approach for thermal comfort management

Maysaa Khalil
Moez Esseghir

Résumé

Existing thermal comfort prediction approaches by machine learning models have been achieving great success based on large datasets in sustainable Industry 4.0 environment. However, the industrial Internet of Things (IoT) environment generates small-scale datasets where each dataset may contain lots of worker’s private data. The latter is challenging the current prediction approaches as small datasets running a large number of iterations can result in overfitting. Moreover, worker’s privacy has been a public concern throughout recent years. Therefore, there must be a trade-off between developing accurate thermal comfort prediction models and worker’s privacy-preserving. To tackle this challenge, we present a privacy-preserving machine learning technique, federated learning (FL), where an FL-based neural network algorithm (Fed-NN) is proposed for thermal comfort prediction. Fed-NN departs from current centralized machine learning approaches where a universal learning model is updated through a secured parameter aggregation process in place of sharing raw data among different industrial IoT environments. Besides, we designed a branch selection protocol to solve the problem of communication overhead in federating learning. Experimental studies on a real dataset reveal the robustness, accuracy, and stability of our algorithm in comparison to other machine learning algorithms while taking privacy into consideration.

Dates et versions

hal-04447370 , version 1 (08-02-2024)

Identifiants

Citer

Maysaa Khalil, Moez Esseghir, Leila Merghem-Boulahia. A federated learning approach for thermal comfort management. Advanced Engineering Informatics, 2022, 52, pp.101526. ⟨10.1016/j.aei.2022.101526⟩. ⟨hal-04447370⟩
25 Consultations
0 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More