This paper presents a correlation-based weighted federated learning approach that combines multimodal-sensing models with knowledge distillation methods. In this case, local models residing on the client devices are regarded as student models and are trained individually, after which their parameters are weighted by the Pearson correlation coefficient before aggregating into a global model. The global model parameters are then distilled by a teacher model on the server. This solution is especially advantageous in cases where edge devices are deployed with weak and heterogeneous configurations, as it permits efficient computational resource management at the clients while still ensuring acceptable performance, aided by a strong teacher model hosted on the server. This solution has demonstrated efficiency when tested on NICT benchmark datasets, and the correlation-based weighted federated learning approach proves to be more stable than traditional FedAvg.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Correlation-Based Weighted Federated Learning with Multimodal Sensing and Knowledge Distillation: An Application on a Real-World Benchmark Dataset

  • Duy-Dong Le,
  • Duy-Thanh Huynh,
  • Pham The Bao

摘要

This paper presents a correlation-based weighted federated learning approach that combines multimodal-sensing models with knowledge distillation methods. In this case, local models residing on the client devices are regarded as student models and are trained individually, after which their parameters are weighted by the Pearson correlation coefficient before aggregating into a global model. The global model parameters are then distilled by a teacher model on the server. This solution is especially advantageous in cases where edge devices are deployed with weak and heterogeneous configurations, as it permits efficient computational resource management at the clients while still ensuring acceptable performance, aided by a strong teacher model hosted on the server. This solution has demonstrated efficiency when tested on NICT benchmark datasets, and the correlation-based weighted federated learning approach proves to be more stable than traditional FedAvg.