联邦学习FedAvg和Krum

想要变酷的史努比

于 2023-06-02 08:51:41 发布

阅读量603

点赞数 2

文章标签： excel 数学建模 python

本文链接：https://blog.csdn.net/m0_46656885/article/details/130999739

版权

本文深入探讨了联邦学习中的两种聚合策略：FedAvg和Krum。FedAvg是联邦学习的基本聚合算法，通过平均各个客户端的模型更新来提升全局模型的性能。而Krum作为防御 Byzantine 故障的策略，通过选择距离全局平均值最近的模型更新来抵抗恶意攻击。了解这两种算法的工作原理对于理解联邦学习的安全性和效率至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
from torchvision import transforms, datasets
import numpy as np
import random
import copy
from tqdm import tqdm
from tensorboardX import SummaryWriter

class AlexNet(nn.Module):
    def __init__(self, num_classes=10):
        super(AlexNet, self).__init__()
        self.features = nn.Sequential(
            nn.Conv2d(3, 64, kernel_size=11, stride=4, padding=2),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
            nn.Conv2d(64, 192, kernel_size=5, padding=2),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
            nn.Conv2d(192, 384, kernel_size=3, padding=1),
            nn.ReLU(inplace=True),
            nn.Conv2d(384, 256, kernel_size=3, padding=1),
            nn.ReLU(inplace=True),
            nn.Conv2d(256, 256, kernel_size=3, padding=1),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
        )
        self.avgpool = nn.AdaptiveAvgPool2d((6, 6))
        self.classifier = nn.Sequential(
            nn.Dropout(),
            nn.Linear(256 * 6 * 6, 4096),
            nn.ReLU(inplace=True),
            nn.Dropout(),
            nn.Linear(4096, 4096),
            nn.ReLU(inplace=True),
            nn.Linear(4096, num_classes)

最低0.47元/天解锁文章