Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification
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...
Gespeichert in:
Hauptverfasser: | , , , , , |
---|---|
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | 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, 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
orchestrating 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. |
---|---|
DOI: | 10.48550/arxiv.2410.15681 |