Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866916544531922944 |
|---|---|
| author | Gaikwad, Nishant S. Heublein, Lucas Raichur, Nisha L. Feigl, Tobias Mutschler, Christopher Ott, Felix |
| author_facet | Gaikwad, Nishant S. Heublein, Lucas Raichur, Nisha L. Feigl, Tobias Mutschler, Christopher Ott, Felix |
| contents | Federated learning (FL) enables multiple devices to collaboratively train a global model while maintaining data on local servers. Each device trains the model on its local server and shares only the model updates (i.e., gradient weights) during the aggregation step. A significant challenge in FL is managing the feature distribution of novel and unbalanced data across devices. In this paper, we propose an FL approach using few-shot learning and aggregation of the model weights on a global server. We introduce a dynamic early stopping method to balance out-of-distribution classes based on representation learning, specifically utilizing the maximum mean discrepancy of feature embeddings between local and global models. An exemplary application of FL is to orchestrate machine learning models along highways for interference classification based on snapshots from global navigation satellite system (GNSS) receivers. Extensive experiments on four GNSS datasets from two real-world highways and controlled environments demonstrate that our FL method surpasses state-of-the-art techniques in adapting to both novel interference classes and multipath scenarios. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2410_15681 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification Gaikwad, Nishant S. Heublein, Lucas Raichur, Nisha L. Feigl, Tobias Mutschler, Christopher Ott, Felix Machine Learning Distributed, Parallel, and Cluster Computing 62P30, 68T30, 68T05, 68T37 G.3; I.2.4; I.2.6 Federated learning (FL) enables multiple devices to collaboratively train a global model while maintaining data on local servers. Each device trains the model on its local server and shares only the model updates (i.e., gradient weights) during the aggregation step. A significant challenge in FL is managing the feature distribution of novel and unbalanced data across devices. In this paper, we propose an FL approach using few-shot learning and aggregation of the model weights on a global server. We introduce a dynamic early stopping method to balance out-of-distribution classes based on representation learning, specifically utilizing the maximum mean discrepancy of feature embeddings between local and global models. An exemplary application of FL is to orchestrate machine learning models along highways for interference classification based on snapshots from global navigation satellite system (GNSS) receivers. Extensive experiments on four GNSS datasets from two real-world highways and controlled environments demonstrate that our FL method surpasses state-of-the-art techniques in adapting to both novel interference classes and multipath scenarios. |
| title | Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification |
| topic | Machine Learning Distributed, Parallel, and Cluster Computing 62P30, 68T30, 68T05, 68T37 G.3; I.2.4; I.2.6 |
| url | https://arxiv.org/abs/2410.15681 |