An Efficient Federated Distillation Learning System for Multitask Time Series Classification

Huanlai Xing,Zhiwen Xiao,Rong Qu,Zonghai Zhu,Bowen Zhao
DOI: https://doi.org/10.1109/tim.2022.3201203
IF: 5.6
2022-09-10
IEEE Transactions on Instrumentation and Measurement
Abstract:This article proposes an efficient federated distillation learning system (EFDLS) for multitask time series classification (TSC). EFDLS consists of a central server and multiple mobile users, where different users may run different TSC tasks. EFDLS has two novel components: a feature-based student–teacher (FBST) framework and a distance-based weights matching (DBWM) scheme. For each user, the FBST framework transfers knowledge from its teacher's hidden layers to its student's hidden layers via knowledge distillation, where the teacher and student have identical network structures. For each connected user, its student model's hidden layers' weights are uploaded to the EFDLS server periodically. The DBWM scheme is deployed on the server, with the least square distance (LSD) used to measure the similarity between the weights of two given models. This scheme finds a partner for each connected user such that the user's and its partner's weights are the closest among all the weights uploaded. The server exchanges and sends back the user's and its partner's weights to these two users which then load the received weights to their teachers' hidden layers. Experimental results show that compared with a number of state-of-the-art federated learning (FL) algorithms, our proposed EFDLS wins 20 out of 44 standard UCR2018 datasets and achieves the highest mean accuracy (70.14%) on these datasets. In particular, compared with a single-task baseline, EFDLS obtains 32/4/8 regarding "win"/"tie"/"lose" and results in an improvement of approximately 4% in terms of mean accuracy.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?