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

TL;DR

This paper presents a federated learning approach with MMD-based early stopping to improve GNSS interference classification, effectively handling unbalanced and out-of-distribution data across devices.

Contribution

It introduces a novel FL method using few-shot learning and dynamic early stopping based on maximum mean discrepancy for adaptive interference classification.

Findings

Outperforms state-of-the-art in GNSS interference classification

Effectively adapts to novel interference classes

Handles multipath scenarios with high accuracy

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.