不可靠通信下的联邦抗干扰模型优化方案

李中捷 ,  郭海榕 ,  邱凡

中南民族大学学报(自然科学版) ›› 2025, Vol. 44 ›› Issue (06) : 826 -832.

PDF (1637KB)
中南民族大学学报(自然科学版) ›› 2025, Vol. 44 ›› Issue (06) : 826 -832. DOI: 10.20056/j.cnki.ZNMDZK.20250611
物理与电子信息科学

不可靠通信下的联邦抗干扰模型优化方案

作者信息 +

Optimization scheme of federated anti-interference model with unreliable communication

Author information +
文章历史 +
PDF (1675K)

摘要

针对蜂窝车联网(Cellular-Vehicle to Everything,C-V2X)通信场景下无线信道受到干扰导致通信过程中可能存在信息丢失的情况,通过对联邦分布式随机梯度下降(Federated Learning-Distributed Stochastic Gradient Descent,FL-DSGD)进行抗干扰模型更新机制的优化以减少上述通信链路不可靠情况的影响.该方案首先建立车辆与基站的通信链路及传输模型参数;然后在通信链路不可靠导致模型参数在传输过程中部分缺失的情况下,根据链路可靠性混合权重矩阵,利用车辆上存储的本地模型以及基站存储的全局模型参与当前轮次联邦学习的模型更新,以填充丢失的模型参数.仿真结果表明:在通信链路不可靠的情况下,FL-DSGD方案达到90%的训练准确率以及85%的测试准确率所需的通信轮次约为分布式基线方案所需通信轮次的50%.

Abstract

Aiming at the possible information loss in the communication process caused by interference to the wireless channel in the Cellular Vehicle to Everything (C-V2X) communication scenario, the impact of the above unreliable communication link is reduced by optimizing the anti-interference model update mechanism of the Federated Learning Distributed Stochastic Gradient Descent (FL-DSGD). Firstly, the communication link between the vehicle and the base station and the transmission model parameters are established; Then, when the communication link is unreliable, leading to partial loss of model parameters in the transmission process, according to the link reliability mixed weight matrix, the local model stored on the vehicle and the global model stored in the base station are used to participate in the model update of the current round of federated learning to fill in the missing model parameters. Simulation results show that when the communication link is unreliable, the communication rounds required for FL-DSGD scheme to achieve 90% training accuracy and 85% test accuracy are about 50% of the communication rounds required for the distributed baseline scheme.

Graphical abstract

关键词

联邦学习 / 车联网 / 随机梯度下降

Key words

federated learning / vehicle to everything / stochastic gradient descent

引用本文

引用格式 ▾
李中捷,郭海榕,邱凡. 不可靠通信下的联邦抗干扰模型优化方案[J]. 中南民族大学学报(自然科学版), 2025, 44(06): 826-832 DOI:10.20056/j.cnki.ZNMDZK.20250611

登录浏览全文

4963

注册一个新账户 忘记密码

随着道路上联网车辆的不断增多,车载通信已成为一个重要的研究领域1.为了满足车联网(Vehicle to Everything,V2X)通信的严格要求,C-V2X需要共享频谱资源,从而实现车对基础设施(Vehicle to Infrastructure,V2I)和车对车(Vehicle to Vehicle,V2V)的同步通信.因此,如何在有限的频谱资源内减少干扰并同时保证V2I链路的信道容量和V2V通信的可靠性,成为V2X通信亟待解决的一个重要问题.
传统的优化方法已经被用来应对V2X通信中的干扰问题2-5.在文献[2]中,提出了一种无线电资源管理(Radio Resource Management,RRM)算法,保证了基于端到端(Device to Device,D2D)的V2X系统的延迟和可靠性要求.基于类似的V2X框架,文献[3]提出了一个优化问题,在只考虑缓慢变化的大规模衰减信道的情况下为D2D车辆系统设计一个高效的频谱和功率分配算.此外,在文献[4-5]中研究了排队延迟对吞吐量和可靠性的影响.
由于车辆的高速移动性导致快速变化的信道以及多样化的服务需求,传统的优化算法难以利用数学方法进行精确建模并且会产生大量的计算开销.因此,机器学习被认为是一种有前途的技术,可以解决传统优化方法中的挑战.
联邦学习作为一种分布式机器学习范式,可以使多个车辆共同训练一个全局模型,而不需要在车辆间或与服务器交换数据6-10.文献[6]通过聚合所有智能体的本地模型,为所有车辆学习一个单一的全局模型来降低通信开销.文献[7]通过在本地目标中引入额外的近端项,以限制本地模型的数量从而提高了联邦学习的稳定性,文献[8-10]为本地模型学习额外的个性化模型来提高本地模型的效率和性能.
然而,上述文献都假设通信链路处于完美的情况,而在实际通信系统中,由于恶劣的无线信道会引入噪声、衰落和干扰,导致丢包和传输错误,以至于模型无法达到最优效果.
本文在C-V2X通信场景下,针对模型传输过程中数据缺失的情况,提出了一种联邦抗干扰模型更新方案,以减少通信链路不可靠的影响.该方案根据链路可靠性混合权重矩阵,对车辆和基站接收到的模型中缺失的部分,利用车辆和基站存储的模型填充缺失的模型,以进行更新模型.最后仿真实验评估了本算法的收敛性,分析不同程度的链路可靠性对系统性能的影响.

1 FL-DSGD通信链路优化方案设计

本文考虑到车辆与基站在行驶过程建立V2I链路进行模型参数传输的过程中,双方只接收到部分模型参数消息的情况下进行共识更新,并根据链路可靠性矩阵Pr优化混合权重矩阵W.

1.1 不可靠通信链路问题描述

在FL-DSGD方案中,车辆n维护并训练自己的本地参数wnRdd为训练参数的维度.车辆根据从本地数据集𝒟n采样的样本ξn计算本地梯度[ggd]n=Fn(wn,ξn),式中,Fn(wn,ξn)为车辆的本地损失函数,之后通过V2I链路与基站服务器J传输模型参数.每次训练迭代的两个步骤分别为SGD更新和联邦共识更新:

wn(t+12)=wn(t)-μ[ggd]n(t)
wJ(t+1)=i=1Nwn,JwJ(t+12)

式中:wn(t)表示车辆nt时刻的参数,μ为学习率,wn,J是混合权重矩阵的第(n,J)项,WRN×N,如果第n个车辆向基站服务器J传输模型参数失败,则wn,J=0.在车辆通过本地训练进行SGD更新时,训练的收敛速度很大程度上取决于混合矩阵W,为了保证收敛性,将W设置为对称双随机,即WT=WW1=1,其中1表示一个全为1的N维向量.在实际的通信系统传输过程中模型参数暴露在不可靠的通信条件下导致容易出现丢包和传输错误.且由于没有数据重传这样的可靠性增强机制,因此无法保证交付.本文令矩阵pr=[pn,Jr]RN×N表示通信网络中的链路可靠性程度,式中pn,Jr表示从第n个车辆成功传输模型至基站服务器的概率.为了模拟随机通信网络,在每次车辆和基站服务器通信前产生一个随机掩码mnJ,以确定基站获取车辆数据的部分,第n个车辆向基站服务器传输模型参数wn的数据包被随机接收,出现在基站服务器上接收到的数据znJ可能只包括wn的一部分.这一过程可表示为:

znJ=mnJwn

式中:表示逐元素相乘法,如果基站服务器J成功接收到第n个车辆传输的第h个参数,则mnJ(h)=1,否则mnJ(h)=0.

1.2 FL-DSGD算法设计

联邦学习通过车辆一定的通信能力达成了协作学习的效果,每个车辆用自己的数据来训练和优化模型参数,并通过基站服务器实现模型信息的交互以达成共识11-12.

在实际的通信场景下,车辆上传wn(t+12)到基站服务器,基站服务器接收到车辆发送的信息表示为:

znJ(t+12)=mnJ(t)wn(t+12) .

本文设计用车辆的本地参数以及基站服务器的全局模型参数填补对应传输过程中缺失的数据.即在车辆进行本地梯度下降过程中,第n个车辆使用wn中的参数来填补zJn中缺失的参数,填补后的信息z^Jn可以表示为:

z^Jn(t+12)=zJn(t+12)+(1-mJn(t))wn(t+12) .

基站服务器进行联邦共识更新可以表示为:

wJ(t+1)=wJwJ(t+12)+n=1Nwn,Jz^nJ(t+12)=wJ(t+12)+n=1Nwn,JmnJ(t)(wn(t+12)-wJ(t+12)) .

由于通信网络的随机性,共识更新步骤具有随机性.本文设置了两个重要的矩阵W¯W2¯,分别表示更新参数的一阶矩阵和二阶矩阵.在期望中,W¯是一个双随机矩阵,不可靠通信网络下的共识更新等价于以W¯为权重矩阵的可靠共识更新.平均混合权重W¯不仅取决于W,还取决于链路可靠性矩阵Pr.其中矩阵pr=[pn,Jr]RN×N表示通信网络中的链路可靠性程度,pn,Jr表示从第n个车辆成功传输模型至基站服务器的概率.

根据矩阵Pr的可靠性,本文采用两种方法来选择混合矩阵W:(1)若链路可靠性矩阵Pr未知,则对每条链路采用平均混合权重,即W=WT/N;(2)如果链路可靠性矩阵Pr已知,则可以通过优化W以加快收敛速度.FL-DSGD方案的收敛性取决于矩阵W2¯-WT/N的最大特征值.因此,本文的优化目标为最小化W矩阵的最大特征值:

minWμmaxW2¯-WT/Ns.t.0wn,J1,WT=W,W1=1 .

在FL-DSGD方案下各车辆与基站服务器进行联邦学习交互信息过程如图1所示,车辆的模型参数交换包括以下步骤:

(1)车辆以及基站服务器初始化模型.

(2)深度神经网络的参数通常无法在一个数据包中传输,本方案通过将参数随机分组为多个数据包并独立传输.

(3)模型通过V2I链路传输到基站服务器进行联邦共识更新.由于通信链路的不可靠,传输信息在接收时可能被随机丢弃或污染.在这种情况下,部分模型数据被宣布丢失并被丢弃,否则数据包将被成功接收.对于缺失部分,基站服务器利用存储的全局模型以及混合权重矩阵进行联邦共识更新.

(4)全局模型反馈至本地车辆并进行梯度下降.若全局模型在传输过程中缺失,车辆利用存储的本地模型以及混合权重矩阵进行模型更新.

FL-DSGD优化通信链路方案的流程如表1所示:

2 收敛性分析

假设每个车辆本地损失函数fn(w)是光滑的且具有L-Lipschitzian梯度,即存在常数L>0,使得

fn(w)-fn(w')Lw-w',

式中,对于w,w'Rd·为欧式范数.

假设存在常数σ>0ζ>0,且σ2ζ2分别限制每个车辆上SGD的方差以及数据分布的差异:

E{Fn(w,ξ)-fn(w)}σ2
1Nn=1Nfn(w)-f(w)ζ2

每个车辆获得的随机梯度是对本地目标的真实梯度的无偏估计13

E{[ggd]n}=fn(wn). .

为了分析FL-DSGD方案的收敛性,本文首先研究了不可靠通信网络下的共识算法.假设设备初始化为{wn(0)Rd,n=1,,N},每次迭代中只执行共识的更新步骤,即:

wJ(t+1)=wJ(t)+n=1Nwn,JmnJt(wJ(t)-wn(t)). .

按照共识更新规则,每一次梯度下降的期望受wt方差的限制.其中κ为建立通信链路的确定性.如果通信链路是确定的,即pn,Jr等于0或1,κ值则为0.

E{w¯t+1-w¯t2}κN2n=1Nwn(t)-w¯t2            κ=2maxnn=1Nwn,J2pn,Jr(1-pn,Jr) .

FL-DSGD方案的收敛程度取决于共识更新的收敛速度.不可靠通信链路的共识收敛率和期望分别为:

1Nn=1Nwn(t)-w¯t2E1Nn=1Nwn(t)-w¯t2ϖtn=1Nwn(0)-w¯02

式中:ϖ=max(W2¯-WT/N),如果通信网络是可靠的,共识更新规则具有指数级的收敛速度14-15.

FL-DSGD方案中假设所有本地模型都用wn(0)Rd进行初始化.其中学习率满足μLmin{1,ϖ-1-n},则在T次迭代之后:

1Tt=1Tf(w¯t)2(E[f(w¯t)]-E[f(w¯t)]μT               +μLNσ2+2γLκNσ2+6Lκμζ2N)1-D1-2D               +L2+2LκμN+2(3N+1)L3μκN                2μ2σ2ϖ1-ϖ+6μ2ζ2ϖ(1-ϖ)211-2D

式中:D=6μ2L2ϖ(1-ϖ)2ϖ式(4)的最大特征值.收敛边界在很大程度上取决于ϖ,如果学习率μ=NT,在总共T次迭代之后:

1Tt=1Tf(w¯t)2=𝒪1NT+𝒪NT

式中:𝒪为常数汇总,因此具有不可靠通信链路的FL-DSGD方案可以实现收敛.

3 仿真结果及分析

本文设置所有车辆在仿真场景内初始化距离相同,在60次训练迭代后学习率衰减10倍.为了验证FL-DSGD方案的效率,本文在模拟实验中采用了两种对比算法:(1)完美通信链路下的联邦学习方案(perfect-FL):假设所有车辆都与基站服务器建立连接并且通信信道处于可靠状态下,不会出现信息丢失情况,车辆在训练本地模型后将本地模型上传至基站,并由基站服务器聚合后反馈给每个车辆.(2)传统分布式学习方案(DSGD):每个车辆统一分配数据独自训练模型.

信道系统在Pytorch中实现,数据集为CIFAR-10,其中包含50000个训练样本和10000个测试样本.网络模型为ResNet-20,激活函数为Tanh,表2给出模型训练的超参数.

图2给出了DSGD方案以及FL-DSGD方案在通信链路可靠性程度k=0.7的情况下的收敛性对比.DSGD方案与FL-DSGD方案相比,准确率收敛所需的通信轮次更多.从图中可以看出,DSGD方案达到90%训练准确率以及85%测试准确率所需的通信轮次约为FL-DSGD方案所需通信轮次的两倍,这是因为在保证所有训练数据完全被车辆接收的情况下,需要另外花费更多通信轮次来维持稳定的通信链路.同时,FL-DSGD方案的训练准确率高于测试准确率,这是由于车辆模型需要在不同的数据分布上进行训练,而联邦学习的全局模型泛化能力不够强,使车辆无法很好地适应测试数据中的新分布,导致训练准确率高于测试准确率.上述实验结果表明FL-DSGD方案在准确率以及模型传输需要的通信开销有较好的平衡性,更符合车联网场景下建立V2V通信链路的稳定性与即时性.

图3比较了FL-DSGD、perfect-FL以及DSGD三种方案在通信链路可靠性程度为0.7的迭代训练和测试损失以及迭代训练和测试准确率.实验结果显示,在损失方面,perfect-FL方案的训练损失和测试损失分别为0.07和0.4,而FL-DSGD方案的训练损失和测试损失分别为0.1和0.44.FL-DSGD与perfect-FL方案的训练和测试损失相近,并且比DSGD方案降低了约0.1.在准确率方面,perfect-FL方案处于完美通信链路下传输模型,因此可以获取较优的效果,其次为FL-DSGD方案.在准确率方面,perfect-FL方案的收敛速度最快,且可以达到97%的准确率,而FL-DSGD方案的训练准确率比DSGD方案提高了2%.这是因为FL-DSGD方案利用来自所有不可靠通信链路的信息并使用部分接收到的消息进行更新,而DSGD方案仅根据车辆本地数据训练网络模型参数并进行更新.

图4表示在通信链路可靠性程度不同的情况下,FL-DSGD方案的训练准确率对比.本文分别使用具有链路可靠性不同的矩阵权重k=[0.3,0.5,0.7,0.9]来评估FL-DSGD方案在训练准确率上的表现.从实验结果可以看出,随着通信链路可靠性权重降低,FL-DSGD方案的训练准确率收敛过程都会变慢.此外,当k为0.7和0.9时,通信链路的可靠性较高,FL-DSGD的训练准确率差距只有0.01.但当k为0.3和0.5时,链路可靠性较低,可靠性权重之间的性能差距会增加.这是因为当存在较多传输成功概率很小的链路时,均匀权重仍然对这些链路分配相同的权重,从而阻碍了收敛过程,导致整体的收敛速度降低.

图5表示在不同车辆数量和不同链路可靠性程度下,perfect-FL方案与FL-DSGD方案训练准确率的比值.本文分别使用具有不同车辆数量[4,8,12,16]和不同链路可靠性矩阵的权重k=[0.3,0.5,0.7,0.9]来评估FL-DSGD方案在训练准确率的表现.从实验结果可以看出,随着链路可靠性程度的降低,perfect-FL方案与FL-DSGD方案训练准确率的比值随之下降.具体来看,当车辆数为4时,k=0.9和k=0.3的准确率比值相差约为0.02,但当车辆数为16时,k=0.9和k=0.3的准确率比值相差约为0.05.这是因为随着车辆数量的增加,模型参数传输过程中的计算和通信开销会增加,导致训练过程变得更加耗时和复杂,从而影响了车辆的准确率.

图6展示了在不同车辆数下FL-DSGD方案的训练准确率.在保证通信链路可靠性分布一致性的前提下,分别设置了[4,8,12,16]辆车,每个车辆从训练集中随机选择3000个样本进行训练.从图6可以看出,在不同数量的车辆参与联邦学习过程中,都能达到收敛的效果,但随着车辆数量的增加其收敛效果会有一定程度的下降,其中当车辆数分别为4和16时,FL-DSGD方案的训练准确率只相差0.02,表明了该方案对车辆数量增多时具有鲁棒性.虽然更多的车辆参与到联邦学习会导致基站服务器的信息处理压力增大,但车辆的增加也会导致通信连接更密集,使整体训练样本增加,因此训练的模型准确率都保持在95%以上.

4 结语

针对C-V2X通信场景下,车辆与基站服务器建立V2I链路通信进行联邦共识更新,由于通信链路不可靠导致模型参数在传输过程中丢失的问题.本文设计了一个联邦抗干扰模型更新方案——FL-DSGD.该方案考虑到实际V2X通信场景中通信链路的不可靠性,实现了低通信开销的同时对通信链路不可靠具有鲁棒性的效果.实验结果表明:在通信链路受到干扰导致不可靠的场景下,FL-DSGD方案不仅与利用完美通信链路进行联邦聚合的perfect-FL方案拥有相似的收敛速率,且在通信链路可靠性程度发生变化以及参与车辆数量变化的情况下仍有较好的准确率,证明所提方案鲁棒性较好.

参考文献

[1]

WANG JLIU JKATO N. Networking and communications in autonomous driving: A survey[J]. IEEE Communications Surveys & Tutorials201921(2): 1243-1274.

[2]

SUN WSTRÖM E GBRÄNNSTRÖM Fet al. Radio resource management for D2D-based V2V communication[J]. IEEE Transactions on Vehicular Technology201665(8): 6636-6650.

[3]

LIANG LLI G YXU W. Resource allocation for D2D-enabled vehicular communications[J]. IEEE Transactions on Communications201765(7): 3186-3197.

[4]

MEI JZHENG KZHAO Let al. A latency and reliability guaranteed resource allocation scheme for LTE V2V communication systems[J]. IEEE Transactions on Wireless Communications201817(6): 3850-3860.

[5]

LIU C FBENNIS M. Ultra-reliable and low-latency vehicular transmission: An extreme value theory approach[J]. IEEE Communications Letters201822(6): 1292-1295.

[6]

MCMAHAN H BMOORE ERAMAGE Det al. Communication-efficient learning of deep networks from decentralized data[C]//Artificial Intelligence and Statistics, Fort Lauderdale:PMLR,2017: 1273-1282.

[7]

LI TSAHU A KZAHEER Met al. Federated optimization in heterogeneous networks[J]. Proceedings of Machine learning and systems20202: 429-450.

[8]

FALLAH AMOKHTARI AOZDAGLAR A. Personalized federated learning with theoretical guarantees: A model-agnostic meta-learning approach[J]. Advances in Neural Information Processing Systems202033: 3557-3568.

[9]

DINH C TTRAN N HNGUYEN T D. Personalized federated learning with Moreau envelopes[J]. Advances in Neural Information Processing Systems202033:21394-21405.

[10]

LI THU SBEIRAMI Aet al. Ditto: Fair and robust federated learning through personalization[C]//International Conference on Machine Learning. Virtual:PMLR, 2021: 6357-6368.

[11]

YUAN KLING QYIN W. On the convergence of decentralized gradient descent[J]. SIAM Journal on Optimization201626(3): 1835-1854.

[12]

ASSRAN MLOIZOU NBALLAS Net al. Stochastic gradient push for distributed deep learning[C]//International Conference on Machine Learning. Long Beach:PMLR, 2019: 344-353.

[13]

LIAN XZHANG WZHANG Cet al. Asynchronous decentralized parallel stochastic gradient descent[C]//International Conference on Machine Learning. Stockholm:PMLR, 2018: 3043-3052.

[14]

TANG HGAN SZHANG Cet al. Communication compression for decentralized training[J]. Advances in Neural Information Processing Systems201831: 7663-7673.

[15]

KOLOSKOVA ASTICH S UJAGGI M. Decentralized stochastic optimization and gossip algorithms with compressed communication[C]//International Conference on Machine Learning. Long Beach:PMLR, 2019: 3478-3487.

基金资助

国家自然科学基金资助项目(61379028)

湖北省自然科学基金资助项目(2022CFB905)

中央高校基本科研业务费专项资金资助(CZY23027)

AI Summary AI Mindmap
PDF (1637KB)

242

访问

0

被引

详细

导航
相关文章

AI思维导图

/