Source-linked AI summary
An Efficient Federated Distillation Learning System for Multi-task Time Series Classification
Huanlai Xing, Zhiwen Xiao, Rong Qu, Zonghai Zhu, Bowen Zhao
TL;DR
Multi-task time-series classification needs secure knowledge sharing across users with different tasks. EFDLS combines local feature distillation with server-side matching of similar user weights; on selected UCR 2018 datasets, it outperformed six FL algorithms on reported top-1-accuracy measures.
Problem
Securely sharing similar expertise across different TSC tasks remains challenging, while time-series applications span diverse domains and tasks.
Source: §1 · p. 1, §2.2 · p. 3Method
EFDLS distills teacher hidden-layer knowledge into each user's student and uses server-side weight matching to connect users with similar expertise.
Source: §1 · p. 2Results
EFDLS outperformed six FL algorithms across 44 UCR 2018 datasets on mean accuracy, win/tie/lose, and AVG rank, based on top-1 accuracy.
Source: §1 · p. 2Takeaways & Limitations
The results indicate EFDLS's potential for multi-task TSC in real-world domains, with reported wins against benchmark FL algorithms across the selected datasets.
Source: §5 · p. 9Evaluation covers 44 of the UCR archive's 128 datasets because limited computing resources prevented considering the whole archive.
Source: §4.1.1 · p. 6
Abstract
from arXiv · showhide
This paper proposes an efficient federated distillation learning system (EFDLS) for multi-task 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, namely a feature-based student-teacher (FBST) framework and a distance-based weights matching (DBWM) scheme. Within each user, the FBST framework transfers knowledge from its teacher's hidden layers to its student's hidden layers via knowledge distillation, with the teacher and student having identical network structure. 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 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 the proposed EFDLS achieves excellent performance on a set of selected UCR2018 datasets regarding top-1 accuracy.
1 INTRODUCTION
The introduction frames multi-task TSC as a privacy-sensitive knowledge-sharing problem across users with different tasks and data. EFDLS combines local distillation with server-side partner matching and is reported to outperform six FL algorithms on 44 UCR 2018 datasets.
- Earlier TSC research introduced approaches including fully convolutional networks, TapNet, FCN-LSTM, RTFN, shapelet-neural networks, and dynamic temporal pooling.Source: §1 · p. 1
- Federated learning lets users benefit from shared models without sending their original data, and FedAvg, FTL, and FKD are identified as its three mainstream directions.Source: §1 · p. 1
- Multi-task TSC requires secure sharing of expertise across tasks, but varied dataset characteristics make this difficult and privacy and knowledge sharing both need attention.The introduction notes that datasets can differ in length and variance and be strongly Non-I.I.D.Source: §1 · p. 2
- EFDLS uses feature-based knowledge distillation within each user and matches each user with the most similar expertise by least square distance for cross-user sharing.Unlike traditional FKD, which uses average user weights, EFDLS selects a partner for each user.Source: §1 · p. 2
- EFDLS outperforms six state-of-the-art FL algorithms across 44 UCR 2018 datasets on mean accuracy, win/tie/lose, and AVG rank, all based on top-1 accuracy.Source: §1 · p. 2
2 RELATED WORK
The related work surveys traditional and deep-learning approaches to TSC, most of which emphasize single tasks. It identifies secure, efficient sharing of expertise across different TSC tasks as an open challenge.
2 RELATED WORK
- The section broadly notes that many traditional and deep-learning algorithms have been developed for TSC.Source: §2 · p. 2
2.1 Traditional Algorithms
- Traditional TSC methods include distance-based DTW-nearest-neighbor ensembles and feature-based approaches such as shapelet transformation, bag-of-features, and convolutional-kernel features.Source: §2.1 · pp. 2–3
2.2 Deep Learning Algorithms
- Deep TSC models are described as single-network or dual-network architectures, with examples including FCN, ResNet, FCN-LSTM, RTFN, and TapNet.Source: §2.2 · p. 3
- Most existing algorithms emphasize single-task TSC, while secure and efficient sharing of similar expertise across different tasks remains challenging.Source: §2.2 · p. 3
3 EFDLS
EFDLS combines local feature-based distillation with server-side matching of users’ model weights in a multi-task time-series classification system. Its users train locally and exchange hidden-layer weights through a central server to share expertise with users whose weights are most similar.
3.1 System Overview
- Users run one time-series classification task each, while the system supports different tasks or data sources across users through a central server and mobile clients.Users train locally and share model weights through the server.Source: §3.1 · pp. 3–4
3.3 Distance-based Weights Matching
- DBWM matches each connected user with the user whose model weights are most similar, using least square distance to measure weight similarity.The server launches matching after receiving all connected users’ uploaded weights.Source: §3.1 · p. 4, §3.3 · p. 4
3.2 Feature-based Student-Teacher Framework
- FBST uses teacher and student models with identical network structures, transferring hidden-layer knowledge from teacher to student to support richer feature representations.The student captures representations from its input data under teacher supervision.Source: §3.2 · p. 4
3.2.1 Feature Extractor
- The feature extractor uses three 1D convolutional blocks, average pooling, and a dense layer for local-feature extraction, followed by a dense-layer and Softmax classifier.Each convolutional block contains a convolution module, batch normalization, and ReLU.Source: §3.2.1 · p. 4
3.2.2 Knowledge Distillation
- Knowledge distillation defines student training loss using teacher and student hidden-layer outputs, alongside supervised cross-entropy loss.The paper sets ε = 0.9.Source: §3.2.2 · p. 4
3.3.1 Least Square Distance
- The server computes pairwise squared distances between uploaded student hidden-layer weights and uses each distance row’s minimum to identify a matching user.Matched weights are sent to users, who load them into their teachers’ hidden layers at the next federated learning epoch.Source: §3.3.1 · p. 5
3.4 Communication Overhead
- EFDLS waits until all connected users’ weights arrive before matching, and its total communication overhead is proportional to 2BW · FLEs · Nconn.Uploading and downloading each require BW per user per epoch because each user has the same model structure.Source: §3.4 · p. 5
4 PERFORMANCE EVALUATION
The section proceeds from experimental setup and performance metrics to an ablation study, then evaluates EFDLS.
- The evaluation section introduces its experimental setup and performance metrics, presents an ablation study, and then evaluates EFDLS.Source: §4 · p. 6
4.1 Experimental Setup
Experiments used 44 selected UCR 2018 datasets, with implementation settings and hardware specified for evaluation.
4.1.1 Data Description
- 44 datasets were selected from the UCR 2018 archive’s 128 because available computing resources could not support evaluating the full archive.The selection comprised 11 datasets from each of four length categories.Source: §4.1.1 · p. 6
4.1.2 Implementation Details
- Training used batch-normalization decay 0.9, L2 regularization, and Adam with an initial learning rate of 0.0001.Source: §4.1.2 · p. 6
- Table 2 reports results for different algorithms on 44 datasets with Nconn = 44 and Ntot = 44.Source: Table 2 · p. 7
- Experiments ran on an Nvidia GTX 1080Ti GPU with 11GB memory and an AMD R5 1400 CPU with 16G RAM under Ubuntu 18.04.Source: §4.1.2 · p. 7
4.2 Performance Metrics
The study evaluates algorithms using top-1-accuracy-based dataset comparisons, mean accuracy, and average rank.
- ‘Win’/‘tie’/‘lose’ count datasets where an algorithm performs better than, equal to, or worse than the others; ‘best’ sums wins and ties.Source: §4.2 · p. 7
- AVG rank measures the average difference between a model’s accuracy and the best accuracy among all models.Source: §4.2 · p. 7
4.3 Ablation Study
The ablation study examines EFDLS on 44 datasets while varying the number of connected users and the loss-balancing coefficient ϵ.
4.3 Ablation Study
- The ablation uses 44 users across the 44 datasets, with each user assigned data from one dataset; users may share a task but use different data sources.Source: §4.3 · pp. 7–8
4.3.1 Impact of Nconn
- A larger Nconn tends to produce higher MeanACC across 44 datasets, with more connected users making more time-series data available to the system.Source: §4.3.1 · p. 8, Figure 2 · p. 8
4.3.2 Impact of ϵ
- With Nconn = 44 and Ntot = 44, ϵ = 0.90 yields the highest MeanACC, 0.7014.The paper describes ϵ as balancing each connected user’s supervised and KD losses.Source: §4.3.2 · p. 8, Figure 3 · p. 8
4.4 Experimental Analysis
Across the benchmark comparison, EFDLS ranks best on the reported aggregate metrics. Comparisons with Baseline, FTLS, and FKD examine knowledge sharing and the system’s two components.
4.4 Experimental Analysis
- EFDLS wins 18 of 44 datasets versus 4 for Baseline, which uses standalone feature extractors without knowledge sharing.Source: §4.4 · p. 8, Figure 4 · p. 9
- EFDLS outperforms FTLS on ‘best’, MeanACC, and AVG rank; the paper attributes the contrast to FBST’s teacher-to-student knowledge transfer.Source: §4.4 · p. 8, Figure 4 · p. 9
- EFDLS outperforms FKD on ‘best’, MeanACC, and AVG rank; DBWM pairs users with similar expertise, unlike FKD’s use of average weights.The paper notes that averaging may lead to catastrophic forgetting when a user’s weights differ substantially from the average.Source: §4.4 · pp. 8–9, Figure 4 · p. 9
- EFDLS leads all seven comparison algorithms with MeanACC 0.7014, ‘best’ 20, and AVG rank 2.1478; FKD is second on these metrics.FedAvg and FedAvgM are reported as the two worst algorithms.Source: §4.4 · pp. 8–9, Figure 4 · p. 9
5 CONCLUSION
EFDLS combines feature-based student-teacher distillation with distance-based partner matching to share similar expertise across multi-task time series classification. It outperforms six benchmark FL algorithms on 44 datasets by MeanACC and AVG rank, and is a winner on 20 datasets by the ‘best’ measure.
- FBST transfers knowledge from a teacher’s hidden layers to its student’s, while DBWM pairs users with the most similar uploaded weights.Together, the components support secure sharing of similar expertise across different tasks.Source: §5 · p. 9
- EFDLS is a winner over six benchmark FL algorithms on 44 datasets by MeanACC and AVG rank, and on 20 datasets by the ‘best’ measure.Source: §5 · p. 9
- Against single-task Baseline, EFDLS records 32/4/8 wins/ties/losses.Source: §5 · p. 9
Related papers
This paper cites
- Federated Machine Learning: Concept and Applications2019
- Knowledge Distillation: A Survey2020
- Robust and Communication-Efficient Federated Learning from Non-IID Data2019
- A Survey on Federated Learning Systems: Vision, Hype and Reality for Data Privacy and Protection2019
- Multi-Task Learning with Deep Neural Networks: A Survey2020