用Kolmogorov-Arnold网络提升卫星网络流量预测的联邦学习性能
Fed-KAN: Federated Learning with Kolmogorov-Arnold Networks for Traffic Prediction
- 在联邦学习中引入KAN网络,利用其函数逼近优势
- 相比传统模型,测试损失降低77.39%
- 适合动态卫星网络环境下的隐私保护预测任务
非地面网络(NTNs)正成为现代通信基础设施的关键部分,尤其在低地球轨道(LEO)卫星系统兴起的背景下。传统集中式学习方法因高延迟、间歇性连接和带宽限制面临挑战。联邦学习(FL)作为替代方案,可在保护数据隐私的同时实现分布式训练。然而,现有联邦学习模型如基于多层感知机的联邦学习(Fed-MLP)在计算复杂度和对动态NTN环境的适应性方面表现不佳。本文提出一种新型联邦学习框架——基于科尔莫戈罗夫-阿诺德网络的联邦学习(Fed-KAN),充分利用KAN在函数逼近方面的优势。我们在真实卫星运营商的交通数据集上评估了Fed-KAN与Fed-MLP的表现,结果表明:相比Fed-MLP,Fed-KAN平均测试损失降低77.39%,显著提升了性能和泛化能力。最后,论文还探讨了Fed-KAN在O-RAN中的潜在应用及其在NTN架构中拆分功能的使用场景。
原文摘要 · Abstract (English)
Non-Terrestrial Networks (NTNs) are becoming a critical component of modern communication infrastructures, especially with the advent of Low Earth Orbit (LEO) satellite systems. Traditional centralized learning approaches face major challenges in such networks due to high latency, intermittent connectivity and limited bandwidth. Federated Learning (FL) is a promising alternative as it enables decentralized training while maintaining data privacy. However, existing FL models, such as Federated Learning with Multi-Layer Perceptrons (Fed-MLP), can struggle with high computational complexity and poor adaptability to dynamic NTN environments. This paper provides a detailed analysis for Federated Learning with Kolmogorov-Arnold Networks (Fed-KAN), its implementation and performance improvements over traditional FL models in NTN environments for traffic forecasting. The proposed Fed-KAN is a novel approach that utilises the functional approximation capabilities of KANs in a FL framework. We evaluate Fed-KAN compared to Fed-MLP on a traffic dataset of real satellite operator and show a significant reduction in training and test loss. Our results show that Fed-KAN can achieve a 77.39% reduction in average test loss compared to Fed-MLP, highlighting its improved performance and better generalization ability. At the end of the paper, we also discuss some potential applications of Fed-KAN within O-RAN and Fed-KAN usage for split functionalities in NTN architecture.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。