Federated Learning Framework Coping with Hierarchical Heterogeneity in
  Cooperative ITS
  In this paper, we introduce a federated learning framework coping with
Hierarchical Heterogeneity (H2-Fed), which can notably enhance the conventional
pre-trained deep learning model. The framework exploits data from connected
public traffic agents in vehicular networks without affecting user data
privacy. By coordinating existing traffic infrastructure, including roadside
units and road traffic clouds, the model parameters are efficiently
disseminated by vehicular communications and hierarchically aggregated.
Considering the individual heterogeneity of data distribution, computational
and communication capabilities across traffic agents and roadside units, we
employ a novel method that addresses the heterogeneity of different aggregation
layers of the framework architecture, i.e., aggregation in layers of roadside
units and cloud. The experiment results indicate that our method can well
balance the learning accuracy and stability according to the knowledge of
heterogeneity in current communication networks. Compared to other baseline
approaches, the evaluation on a Non-IID MNIST dataset shows that our framework
is more general and capable especially in application scenarios with low
communication quality. Even when 90% of the agents are timely disconnected, the
pre-trained deep learning model can still be forced to converge stably, and its
accuracy can be enhanced from 68% to over 90% after convergence.
