Digital Twin Enabled Multi-task Federated Learning in Heterogeneous Vehicular Networks

Abstract

In the heterogeneous vehicular networks (HetVNets), the base stations (BSs) can exploit the massive amounts of valuable data collected by vehicles to complete federated learning tasks. However, most of the existing studies consider the scenario of one task requester (TR) and ignore the fact that multiple TRs may concurrently generate their model training requests in the HetVNets. In this paper, we consider the scenario of multi-TR and multi-BS and propose a digital twin enabled scheme for multitask federated learning to address the two-way selection problem between the TRs and the BSs. We first analyze the diversified requirements of the TRs in the HetVNets. Then, we develop a novel model that jointly considers the available training data, the declared price, and the training experience to evaluate the differentiated training capabilities of the BSs. After that, based on the requirements of the TRs and the training capabilities of the BSs, the two-way selection problem between the TRs and the BSs is formulated as a matching game in the digital twin networks, where a matching algorithm is designed to obtain their optimal strategies. The simulation results demonstrate that the proposed scheme can obtain the highest model accuracy and bring the highest utility to the TRs compared with the conventional schemes.