Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification

View a PDF of the paper titled Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification, by Nishant S. Gaikwad and Lucas Heublein and Nisha L. Raichur and Tobias Feigl and Christopher Mutschler and Felix Ott

View PDF

Abstract: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.