arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.08028cs.LGcs.DC

FedAuxHMTL: 面向网络边缘流量分类的联邦辅助硬参数共享多任务学习

FedAuxHMTL: Federated Auxiliary Hard-Parameter Sharing Multi-Task Learning for Network Edge Traffic Classification

North Carolina State University(北卡罗来纳州立大学);Cisco Research(思科研究院);Cisco Systems, Inc.(思科系统公司);The George Washington University(乔治·华盛顿大学);University of Electro-Communications(东京电机大学);Nagoya University(名古屋大学)

Faisal Ahmed, Myungjin Lee, Suresh Subramaniam, Motoharu Matsuura, Hiroshi Hasegawa, Shih-Chun Lin

AI总结:

针对联邦学习在网络边缘流量分类中面临的数据异质性与标注稀缺问题,本文提出FedAuxHMTL框架,通过硬参数共享多任务学习与边-基站参数交换提升主任务性能,实验验证其在多指标上优于同类方法。

摘要:

联邦学习(Federated Learning, FL)近年来因有望为各类应用场景中的诸多挑战提供有效解决方案而备受关注,例如网络边缘流量分类中的数据隐私问题。尽管FL的优势已得到公认,但在训练基于机器学习的流量分类单任务模型时,FL会遭遇与统计数据异质性和标注数据稀缺相关的障碍,进而导致学习性能受限。针对这些挑战,采用带有辅助任务的硬参数共享多任务学习模型被证明是一种合适的方法。该模型能够降低通信与计算成本,应对FL场景中固有的统计复杂性,并通过利用从关联辅助任务中获取的知识来缓解标注数据稀缺的问题。本文提出了一种新的联邦辅助硬参数共享多任务学习框架,即FedAuxHMTL。该框架引入了边缘服务器与基站之间的模型参数交换机制,使分布在不同区域的基站能够参与FedAuxHMTL流程,从而提升主任务——网络边缘流量分类的学习性能。本文开展了实证实验,以验证并证明FedAuxHMTL在准确率、全局总损失、通信成本、计算时间和能耗方面相较于同类方法的有效性。

英文摘要:

Federated Learning (FL) has garnered significant interest recently due to its potential as an effective solution for tackling many challenges in diverse application scenarios, for example, data privacy in network edge traffic classification. Despite its recognized advantages, FL encounters obstacles linked to statistical data heterogeneity and labeled data scarcity during the training of single-task models for machine learning-based traffic classification, leading to hindered learning performance. In response to these challenges, adopting a hard-parameter sharing multi-task learning model with auxiliary tasks proves to be a suitable approach. Such a model has the capability to reduce communication and computation costs, navigate statistical complexities inherent in FL contexts, and overcome labeled data scarcity by leveraging knowledge derived from interconnected auxiliary tasks. This paper introduces a new framework for federated auxiliary hard-parameter sharing multi-task learning, namely, FedAuxHMTL. The introduced framework incorporates model parameter exchanges between edge server and base stations, enabling base stations from distributed areas to participate in the FedAuxHMTL process and enhance the learning performance of the main task-network edge traffic classification. Empirical experiments are conducted to validate and demonstrate the FedAuxHMTL's effectiveness in terms of accuracy, total global loss, communication costs, computing time, and energy consumption compared to its counterparts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑