Page 79 - 《软件学报》2026年第4期
P. 79

1520                                                       软件学报  2026  年第  37  卷第  4  期


                                                   w . 随后, 依据客户端上传的局部类别原型加权聚合更新全局类别原
                                                    t+1
                 行加权聚合, 更新得到新一轮全局模型参数
                                                    g
                        +
                           −
                 型表示  {P ,P }. 完成更新后, 服务端将模型参数和类别原型广播至各客户端, 用于启动下一轮本地训练.
                  3.4   收敛性分析
                    本节对于    FedRPDA  在非凸条件下的收敛性给出了简要的理论分析, 并作出与现有框架相似的假设条件.
                    假设  1.                              L n  (公式                             w n 和w , 满足:
                                                                                                  ′
                          L -光滑性. 每个客户端的局部目标函数                  (18)) 是   L -光滑的, 即对任意模型参数        n
                                                    (  ′  )         ′
                                                ||∇L n w −∇L n (w n )|| ⩽ L||w −w n ||               (19)
                                                     n              n
                 等价于二次上界条件:

                                                                       L
                                           (  )        ⟨            ⟩          2
                                             ′
                                                                ′
                                                                          ′
                                         L n w ⩽ L n (w n )+ ∇L n (w n ),w −w n + ||w −w n ||        (20)
                                                                n
                                                                          n
                                             n
                                                                       2
                    假设                                       ∇L n (w n ;ξ) 是对真实梯度的无偏估计:
                        2. 无偏梯度. 基于小批量数据       ξ  计算的随机梯度
                                                     [       ]
                                                   E ξ ∇L n (w n ;ξ) = ∇L n (w n )                   (21)
                    假设                                      2          w n  有:
                                                            n
                        3. 有界方差. 随机梯度方差有界, 即存在常数           σ , 使得对任意
                                                  [                 ]
                                                                    2
                                                E ξ ||∇L n (w n ;ξ)−∇L n (w n )|| ⩽ σ 2 n            (22)
                    假设                        2      2                    w n  有:
                        4. 有界非相似性. 存在常数      β ⩾ 1 与  κ ⩾ 0, 使得对任意模型参数
                                             N ∑              N ∑
                                                        2   2            2  2
                                               ρ n ||∇L n (w n )|| ⩽ β ||  ρ n ∇L n (w n )|| +κ      (23)
                                             n=1              n=1
                           γ n
                 其中,   ρ n = ∑   表示归一化聚合权重.
                           N  γ
                          n ′ =1 n ′
                    假设                               h(·) 满足  L 2 -Lipschitz 连续:
                        5. Lipschitz 连续. 节点特征提取网络
                                               (  )
                                                ′                ′
                                             ||h w ;v −h(w n ;v)|| ⩽ L 2 ||w −w n ||, ∀v ∈ V n       (24)
                                                n
                                                                 n
                    定理  1. FedRPDA  的非凸收敛性. 基于上述假设, 在非凸条件假设下, FedRPDA            算法经过   T  轮通信后可以达到
                 如下收敛保证:

                            1  ∑     (  ) 2  ]  4β 2 ( (  )  )    (      )  2κ 2
                             T−1 [
                                                                                 2
                                                               2 2
                                E ||∇L w || ⩽    L w −L +2β L η K D +σ +      +2β Lησ +4β λL 2 G     (25)
                                                                                     2
                                                    0
                                                                    2
                                                                                         2
                                                             2
                                       t
                                                                        2
                                                        ∗
                            T          g     ηKT    g                       K
                              t=0
                    证明: 考虑在通信轮次       t 时客户端  n 的第  k 步本地更新:

                                                                 (    )
                                                         t
                                                                  t
                                                  w t n,k+1  = w −η∇L n w ;ξ k                       (26)
                                                         n,k
                                                                  n,k
                    根据假设    1–3  及假设  5, 可以得到任意客户端在单个通信轮次的局部目标函数偏差界限:

                                                   K−1 (  2   )              2
                                  [  (   )]   (  ) ∑     Lη    
   (  )
 2 LKη
                                                                               2
                                                t

                                 E L n w t  ⩽ L n w −  η−   −  λ 
 
∇L n w t  
 +  σ +ληL 2 KG       (27)
                                       n,K      g        2   2       n,k    2  n
                                                   k=0
                    通过选取适当的      η 和  λ 值, 且扰动项受控, 可以保证单轮期望下降.
                    对于服务端加权聚合项, 可以得到全局损失的递推上界:

                                                                                    
                                              N ∑         K−1       )
 2 LKη 2
                                    [ (   )]        (  )  η  ∑ 
  (                 
                                                                             2
                                                t 

                                   E L w t+1  ⩽  ρ    t g  
 
∇L n w t n,k  
 +  σ +ληL 2 KG      (28)
                                                n L n w −
                                                                             n
                                        g
                                             n=1         2  k=0          2
                    由假设   4                                     T −1 求和, 可以得到:
                            和梯度有界性     D ≜ max w,n ||∇L n (w n )||, 对   t = 0 到
                            1  ∑  
  (  )
 2 ]  4β 2 ( (  )  )     (     )  2κ 2
                             T−1 [
                                                                        2
                                                             2
                                                                                         2
                                                                                     2
                                                                                 2
                                                               2 2
                                                                     2
                                E 
∇L w 
 ⩽      L w −L +2β L η K D +σ +      +2β Lησ +4β λL 2 G     (29)
                                                    0
                                       t

                                                         ∗
                            T          g     ηKT    g                       K
                              t=0
                 其中,  L ≜ min w L(w).
                       ∗
                    这表明在非凸情形下, 随着通信轮次            T  的增加, FedRPDA  的平均梯度范数平方持续下降, 其收敛水平受异构
                 性、噪声以及多样化项共同约束. 详细推导过程可见附录                  A.
   74   75   76   77   78   79   80   81   82   83   84