用动态早停提升联邦学习对新型干扰的适应能力
Federated Learning with MMD-based Early Stopping for Adaptive GNSS Interference Classification
- 基于特征嵌入的MMD度量实现动态早停,缓解设备间数据分布不均
- 在4个真实道路数据集上,对新干扰类型和多径场景的识别准确率领先
- 适合需要隐私保护的卫星导航干扰实时分类场景
联邦学习(FL)允许多个设备在本地服务器上协作训练全局模型,仅共享模型更新(如梯度权重)。一个关键挑战是处理设备间新型且不平衡的数据特征分布。本文提出一种结合少样本学习与全局服务器模型权重聚合的联邦学习方法,引入基于特征嵌入最大均值差异(MMD)的动态早停机制,以平衡分布外类别。该方法应用于高速公路沿线全球导航卫星系统(GNSS)接收器快照的干扰分类任务。在来自两条真实道路及受控环境的四个GNSS数据集上的大量实验表明,本方法在适应新干扰类别和多径场景方面优于现有先进技术。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。