Improving Federated Learning Personalization via MAML 论文笔记 【未完】
Improving Federated Learning Personalization via MAML Abstract FL算法与MAML具有很多相似性,可以用元学习算法来对其进行解释 微调可以使得gloabl 模型具有更强的准确率,同时更容易做定制化处理 通过标准的中心化数据库训练出来的模型相比Fedavg训练的更难进行定制化处理 Introduction 指出了FL与MAML算法的联系,并用MAML算法对FL算法进行解释 对FedAvg进行改进,采用两阶段的训练和fine-tune进行优化 发现FedAvg其实本质是一种metalearning算法,用于优化个性化定制的效果,而不是全局模型的优化。 Interpreting FedAvg as a Meta Learning Algorithm 下图展现了在FL中应用MAML算法(左侧),Reptile算法(中间)和FL的训练算法FedAvg(右侧)。设L为损失函数,在每一轮的迭代中,MAML会通过随机采样一个batch的任务T来进行训练。对于每个任务T,会有一个内循环,然后在外循环中聚集每个任务所获得的的梯度更新。对于FL算法会随机采样数个client T。对于每个T和其权重,会在local数据上进行数轮的迭代优化,然后将更新的梯度聚集形成一个新的global model。如果我们简化设置,并认为所有的client拥有相同的数据,那么所有的权重就会一样,这个时候reptile和fedavg其实就是同一种算法。 假设在FedAvg中的权重相同为wi。考虑有T个clients,并设置每个相关模型参数为θ\theta。对于每个cilent i,其损失函数为Li(θ)L_{i}(\theta),记gjig_{j}^{i}为第jthj^{t h}local训练过程所计算得到的梯度。 FedSGD的梯度更新函数为: gFedSGD=−βT∑i=1T∂Li(θ)∂θ=1T∑i=1Tg1i g_{F e d S G D}=\frac{-\beta}{T} \sum_{i=1}^{T} \frac{\partial L_{i}(\theta)}{\partial \theta}=\frac{1}{T} \sum_{i=1}^{T} g_{1}^{i} 将设我们将FOMAML用相同的术语来表示。假设Client 学习率为β\beta, 每个client的个性化模型经过K步后所获得的梯度更新为θKi=UKi(θ)=θ−β∑j=1Kgji=θ−β∑j=1K∂Li(θj)∂θ\theta_{K}^{i}=U_{K}^{i}(\theta)=\theta-\beta \sum_{j=1}^{K} g_{j}^{i}=\theta-\beta \sum_{j=1}^{K} \frac{\partial L_{i}\left(\theta_{j}\right)}{\partial \theta} 求微分可得到: ∂UKi(θ)∂θ=I−β∂∑j=1Kgji∂θ=I−β∑j=1K∂2Li(θj)∂θ2 \frac{\partial U_{K}^{i}(\theta)}{\partial \theta}=I-\beta \frac{\partial \sum_{j=1}^{K} g_{j}^{i}}{\partial \theta}=I-\beta \sum_{j=1}^{K} \frac{\partial^{2} L_{i}\left(\theta_{j}\right)}{\partial \theta^{2}} 在进行K次梯度更新后,对整个模型进行更新: gMAML=∂LMAML∂θ=1T∑i=1T∂Li(UKi(θ))∂θ=1T∑i=1TLi′(UKi(θ))(I−β∑j=1K∂2Li(θj)∂θ2) g_{M A M L}=\frac{\partial L_{M A M L}}{\partial \theta}=\frac{1}{T} \sum_{i=1}^{T} \frac{\partial L_{i}\left(U_{K}^{i}(\theta)\right)}{\partial \theta}=\frac{1}{T} \sum_{i=1}^{T} L_{i}^{\prime}\left(U_{K}^{i}(\theta)\right)\left(I-\beta \sum_{j=1}^{K} \frac{\partial^{2} L_{i}\left(\theta_{j}\right)}{\partial \theta^{2}}\right) 为了避免二次求导带来的计算量问题,FOMAML应运而生,通过K次的梯度更新后,直接采用第K+1次的梯度更新作为local update。 ...