devbox
盐析白兔
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

预训练—冻结权重

作者：盐析白兔 | 2024-04-06 02:31:50

赞

踩

冻结权重

1、迁移学习finetune冻结部分权重

方式：将不更新的参数的requires_grad设置为Fasle，同时不将该参数传入optimizer.

(1)不更新的参数的requires_grad设置为Fasle

  1. # 冻结fc1层的参数
  2. for name, param in model.named_parameters():
  3.     if "fc1" in name:
  4.         param.requires_grad = False

（2）不将该参数传入optimizer

  1. # 定义一个fliter，只传入requires_grad=True的模型参数
  2. optimizer = optim.SGD(filter(lambda p : p.requires_grad, model.parameters()), lr=1e-2) 

总的代码：

  1. # 最优写法
  2. loss_fn = nn.CrossEntropyLoss()
  3.  
  4. # # 训练前的模型参数
  5. print("model.fc1.weight", model.fc1.weight)
  6. print("model.fc2.weight", model.fc2.weight)
  7. print("model.fc1.weight.requires_grad:", model.fc1.weight.requires_grad)
  8. print("model.fc2.weight.requires_grad:", model.fc2.weight.requires_grad)
  9.  
  10. # 冻结fc1层的参数
  11. for name, param in model.named_parameters():
  12.     if "fc1" in name:
  13.         param.requires_grad = False
  14.  
  15. optimizer = optim.SGD(filter(lambda p : p.requires_grad, model.parameters()), lr=1e-2)  # 定义一个fliter，只传入requires_grad=True的模型参数
  16.  
  17. for epoch in range(10):
  18.     x = torch.randn((3, 8))
  19.     label = torch.randint(0,3,[3]).long()
  20.     output = model(x)
  21.  
  22.     loss = loss_fn(output, label)
  23.     optimizer.zero_grad()
  24.     loss.backward()
  25.     optimizer.step()
  26.  
  27. print("model.fc1.weight", model.fc1.weight)
  28. print("model.fc2.weight", model.fc2.weight)
  29. print("model.fc1.weight.requires_grad:", model.fc1.weight.requires_grad)
  30. print("model.fc2.weight.requires_grad:", model.fc2.weight.requires_grad)

最优写法能够节省显存和提升速度：

节省显存：不将不更新的参数传入optimizer

提升速度：将不更新的参数的requires_grad设置为False，节省了计算这部分参数梯度的时间

二、 迁移训练，卷积层设置较小的学习率，全连接层设置较大的学习率，需要用到优化器

  1. # ============================ step 4/5 优化器 ============================
  2. # --------------------法2 : conv卷积层较小学习率，全连接层较大学习率----------------
  3. # flag = 0
  4. flag = 1  # 真
  5. # Python程序语言指定任何非0和非空（null）值为true，0 或者 null为false。"判断条件"成立时/条件为“true"（非零），则执行后面的语句
  6. if flag:
  7.     fc_params_id = list(map(id, resnet18_ft.fc.parameters()))     # 返回的是fc层的parameters的内存地址
  8.     base_params = filter(lambda p: id(p) not in fc_params_id, resnet18_ft.parameters()) # 将resnet18中的参数过滤掉fc层的参数后得到base_params==卷积层参数
  9.     optimizer = optim.SGD([
  10.         {'params': base_params, 'lr': LR*0.1},   # 第一个LR*0.1
  11.         {'params': resnet18_ft.fc.parameters(), 'lr': LR}], momentum=0.9) # 10^-3,两个参数的momentum都是0.9
  13. else:
  14.     optimizer = optim.SGD(resnet18_ft.parameters(), lr=LR, momentum=0.9)               # 选择优化器
  16. scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size=lr_deca y_step, gamma=0.1)     # 设置学习率下降策略

另外一个小技巧就是在nn.Module里冻结参数，这样前面的参数就是False，而后面的不变。

  1. class Net(nn.Module):
  2.     def __init__(self):
  3.         super(Net, self).__init__()
  4.         self.conv1 = nn.Conv2d(1, 6, 5)
  5.  
  6.         for p in self.parameters():
  7.             p.requires_grad=False
  8.  
  9.         self.fc1 = nn.Linear(16 * 5 * 5, 120)
  10.

唯一需要注意的是，pytorch 固定部分参数训练时需要在优化器中施加过滤。

optimizer = optim.Adam(filter(lambda p: p.requires_grad, net.parameters()), lr=0.1)

所有代码如下：finetune_resnet18.py

  1. # -*- coding: utf-8 -*-
  2. """
  3. # @file name  : finetune_resnet18.py
  4. # @brief      : 模型finetune方法
  5. """
  6. import os
  7. import numpy as np
  8. import torch
  9. import torch.nn as nn
  10. from torch.utils.data import DataLoader
  11. import torchvision.transforms as transforms
  12. import torch.optim as optim
  13. from matplotlib import pyplot as plt
  14. from tools.my_dataset import AntsDataset
  15. from tools.common_tools import set_seed
  16. import torchvision.models as models
  17. import torchvision
  18. BASEDIR = os.path.dirname(os.path.abspath(__file__))
  19. device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
  20. print("use device :{}".format(device))
  21.  
  22. set_seed(1)  # 设置随机种子
  23. label_name = {"ants": 0, "bees": 1}
  24.  
  25. # 参数设置
  26. MAX_EPOCH = 25
  27. BATCH_SIZE = 16
  28. LR = 0.001
  29. log_interval = 10
  30. val_interval = 1
  31. classes = 2
  32. start_epoch = -1
  33. lr_decay_step = 7
  34.  
  35.  
  36. # ============================ step 1/5 数据 ============================
  37. data_dir = os.path.join(BASEDIR, "..", "..", "data/hymenoptera_data")
  38. train_dir = os.path.join(data_dir, "train")
  39. valid_dir = os.path.join(data_dir, "val")
  40.  
  41. norm_mean = [0.485, 0.456, 0.406]
  42. norm_std = [0.229, 0.224, 0.225]
  43.  
  44. train_transform = transforms.Compose([
  45.     transforms.RandomResizedCrop(224),
  46.     transforms.RandomHorizontalFlip(),
  47.     transforms.ToTensor(),
  48.     transforms.Normalize(norm_mean, norm_std),
  49. ])
  50.  
  51. valid_transform = transforms.Compose([
  52.     transforms.Resize(256),
  53.     transforms.CenterCrop(224),
  54.     transforms.ToTensor(),
  55.     transforms.Normalize(norm_mean, norm_std),
  56. ])
  57.  
  58. # 构建MyDataset实例
  59. train_data = AntsDataset(data_dir=train_dir, transform=train_transform)
  60. valid_data = AntsDataset(data_dir=valid_dir, transform=valid_transform)
  61.  
  62. # 构建DataLoder
  63. train_loader = DataLoader(dataset=train_data, batch_size=BATCH_SIZE, shuffle=True)
  64. valid_loader = DataLoader(dataset=valid_data, batch_size=BATCH_SIZE)
  65.  
  66. # ============================ step 2/5 模型 ============================
  67.  
  68. # 1/3 构建模型
  69. resnet18_ft = models.resnet18()
  70.  
  71. # 2/3 加载预训练模型的参数
  72. # flag = 0
  73. flag = 1
  74. if flag:
  75.     path_pretrained_model = os.path.join(BASEDIR, "..", "..", "data/resnet18-5c106cde.pth")
  76.     state_dict_load = torch.load(path_pretrained_model)
  77.     resnet18_ft.load_state_dict(state_dict_load)
  78.  
  79. # ----------------------法1 : 冻结卷积层------------------------
  80. flag_m1 = 0
  81. # flag_m1 = 1
  82. if flag_m1:
  83.     for param in resnet18_ft.parameters():
  84.         param.requires_grad = False #参数不求取梯度，即参数不再更新
  85.     print("conv1.weights[0, 0, ...]:\n {}".format(resnet18_ft.conv1.weight[0, 0, ...]))
  86.  
  87.  
  88. # 3/3 替换fc层，适应新任务：设置新的Linear层
  89. num_ftrs = resnet18_ft.fc.in_features #获取原始resnet18模型fc层features的个数
  90. # 构建Linear层的两个参数：输入神经元个数，输出神经元个数=
  91. resnet18_ft.fc = nn.Linear(num_ftrs, classes)
  92.  
  93.  
  94. resnet18_ft.to(device) #将模型放到cpu or gpu
  95. # ============================ step 3/5 损失函数 ============================
  96. criterion = nn.CrossEntropyLoss()                                                   # 选择损失函数
  97.  
  98. # ============================ step 4/5 优化器 ============================
  99. # --------------------法2 : conv卷积层较小学习率，全连接层较大学习率----------------
  100. # flag = 0
  101. flag = 1  # 真
  102. # Python程序语言指定任何非0和非空（null）值为true，0 或者 null为false。"判断条件"成立时/条件为“true"（非零），则执行后面的语句
  103. if flag:
  104.     fc_params_id = list(map(id, resnet18_ft.fc.parameters()))     # 返回的是parameters的 内存地址
  105.     base_params = filter(lambda p: id(p) not in fc_params_id, resnet18_ft.parameters()) # 将resnet18中的参数过滤掉fc层的参数后得到base_params
  106.     optimizer = optim.SGD([
  107.         {'params': base_params, 'lr': LR*0},   # 0
  108.         {'params': resnet18_ft.fc.parameters(), 'lr': LR}], momentum=0.9)
  110. else:
  111.     optimizer = optim.SGD(resnet18_ft.parameters(), lr=LR, momentum=0.9)               # 选择优化器
  113. scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size=lr_deca y_step, gamma=0.1)     # 设置学习率下降策略
  116. # ============================ step 5/5 训练 ============================
  117. train_curve = list()
  118. valid_curve = list()
  120. for epoch in range(start_epoch + 1, MAX_EPOCH):
  122.     loss_mean = 0.
  123.     correct = 0.
  124.     total = 0.
  126.     resnet18_ft.train()
  127.     for i, data in enumerate(train_loader):
  129.         # forward
  130.         inputs, labels = data
  131.         inputs, labels = inputs.to(device), labels.to(device) #将数据、标签放到cpu or gpu
  132.         outputs = resnet18_ft(inputs)
  134.         # backward
  135.         optimizer.zero_grad()
  136.         loss = criterion(outputs, labels)
  137.         loss.backward()
  139.         # update weights
  140.         optimizer.step()
  142.         # 统计分类情况
  143.         _, predicted = torch.max(outputs.data, 1)
  144.         total += labels.size(0)
  145.         correct += (predicted == labels).squeeze().cpu().sum().numpy()
  147.         # 打印训练信息
  148.         loss_mean += loss.item()
  149.         train_curve.append(loss.item())
  150.         if (i+1) % log_interval == 0:
  151.             loss_mean = loss_mean / log_interval
  152.             print("Training:Epoch[{:0>3}/{:0>3}] Iteration[{:0>3}/{:0>3}] Loss: {:.4f} Acc:{:.2%}".format(
  153.                 epoch, MAX_EPOCH, i+1, len(train_loader), loss_mean, correct / total))
  154.             loss_mean = 0.
  156.             # if flag_m1:
  157.             print("epoch:{} conv1.weights[0, 0, ...] :\n {}".format(epoch, resnet18_ft.conv1.weight[0, 0, ...]))
  159.     scheduler.step()  # 更新学习率
  161.     # validate the model
  162.     if (epoch+1) % val_interval == 0:
  164.         correct_val = 0.
  165.         total_val = 0.
  166.         loss_val = 0.
  167.         resnet18_ft.eval()
  168.         with torch.no_grad():
  169.             for j, data in enumerate(valid_loader):
  170.                 inputs, labels = data
  171.                 inputs, labels = inputs.to(device), labels.to(device)
  173.                 outputs = resnet18_ft(inputs)
  174.                 loss = criterion(outputs, labels)
  176.                 _, predicted = torch.max(outputs.data, 1)
  177.                 total_val += labels.size(0)
  178.                 correct_val += (predicted == labels).squeeze().cpu().sum().numpy()
  180.                 loss_val += loss.item()
  182.             loss_val_mean = loss_val/len(valid_loader)
  183.             valid_curve.append(loss_val_mean)
  184.             print("Valid:\t Epoch[{:0>3}/{:0>3}] Iteration[{:0>3}/{:0>3}] Loss: {:.4f} Acc:{:.2%}".format(
  185.                 epoch, MAX_EPOCH, j+1, len(valid_loader), loss_val_mean, correct_val / total_val))
  186.         resnet18_ft.train()
  188. train_x = range(len(train_curve))
  189. train_y = train_curve
  191. train_iters = len(train_loader)
  192. valid_x = np.arange(1, len(valid_curve)+1) * train_iters*val_interval # 由于valid中记录的是epochloss，需要对记录点进行转换到iterations
  193. valid_y = valid_curve
  195. plt.plot(train_x, train_y, label='Train')
  196. plt.plot(valid_x, valid_y, label='Valid')
  198. plt.legend(loc='upper right')
  199. plt.ylabel('loss value')
  200. plt.xlabel('Iteration')
  201. plt.show()

感谢作者：【PyTorch框架】 迁移学习 & 模型微调_迁移学习模型微调_HUI 别摸鱼了的博客-CSDN博客

当有大部分层需要冻结，只有少部分层需要权重更新时，一层一层的写比较麻烦。那么就可以在网络定义时直接设置requires_grad=False，具体如下：

  1. class Net(nn.Module):
  2.     def __init__(self):
  3.         super(Net, self).__init__()
  4.         self.conv1 = nn.Conv2d()
  5.         self.conv2 = nn.Conv2d()
  6.         self.fc1 = nn.Squential(
  7.                                  nn.Linear(),
  8.                                  nn.Linear(),
  9.                                  ReLU(inplace=True),
  10.                                 )
  11.  
  12.         for param in self.parameters():
  13.             param.requires_grad = False
  14.         #这样for循环之前的参数都被冻结，其后的正常更新。
  15.         self.classifier = nn.Linear()

同样，在定义过滤器时要进行过滤，如下：

目的是，告诉优化器，哪些需要更新，那些不需要，这一步至关重要！

optimizer.SGD(filter(lambda p: p.requires_grad, model.parameters()), lr=1e-5)

3、另外一个作者的finetune实现

resnet18是 imagenet 数据集的预训练模型

  1. import os
  2. import torch
  3. import torchvision
  4. from torch import nn
  5. from d2l import torch as d2l
  6.  
  7.  
  8.  
  9. # 解压下载的数据集，我们获得了两个文件夹hotdog/train和hotdog/test。 这两个文件夹都有hotdog（有热狗）和not-hotdog（无热狗）两个子文件夹， 子文件夹内都包含相应类的图像。
  10.  
  11. # d2l.DATA_HUB['hotdog'] = (d2l.DATA_URL + 'hotdog.zip', 'fba480ffa8aa7e0febbb511d181409f899b9baa5')
  12. # data_dir = d2l.download_extract('hotdog')
  13.  
  14. data_dir = './hotdog'
  15.  
  16. # 我们创建两个实例来分别读取训练和测试数据集中的所有图像文件。
  17. train_imgs = torchvision.datasets.ImageFolder(os.path.join(data_dir, 'train'))
  18. test_imgs = torchvision.datasets.ImageFolder(os.path.join(data_dir, 'test'))
  19.  
  20. # 下面显示了前8个正类样本图片和最后8张负类样本图片。 正如你所看到的，[图像的大小和纵横比各有不同]。
  21. hotdogs = [train_imgs[i][0] for i in range(8)]
  22. not_hotdogs = [train_imgs[-i - 1][0] for i in range(8)]
  23. d2l.show_images(hotdogs + not_hotdogs, 2, 8, scale=1.4)
  24. d2l.plt.show()
  25.  
  26. # 在训练期间，我们首先从图像中裁切随机大小和随机长宽比的区域，然后将该区域缩放为 224×224 输入图像。
  27. # 在测试过程中，我们将图像的高度和宽度都缩放到256像素，然后裁剪中央 224×224 区域作为输入。 此外，对于RGB（红、绿和蓝）颜色通道，我们分别标准化每个通道。
  28. # 具体而言，该通道的每个值减去该通道的平均值，然后将结果除以该通道的标准差。
  29. # 使用RGB通道的均值和标准差，以标准化每个通道
  30. normalize = torchvision.transforms.Normalize(
  31.     [0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
  32.  
  33. train_augs = torchvision.transforms.Compose([
  34.     torchvision.transforms.RandomResizedCrop(224),
  35.     torchvision.transforms.RandomHorizontalFlip(),
  36.     torchvision.transforms.ToTensor(),
  37.     normalize])
  38.  
  39. test_augs = torchvision.transforms.Compose([
  40.     torchvision.transforms.Resize(256),
  41.     torchvision.transforms.CenterCrop(224),
  42.     torchvision.transforms.ToTensor(),
  43.     normalize])
  44.  
  45. # [定义和初始化模型] 我们使用在ImageNet数据集上预训练的ResNet-18作为源模型。 在这里，我们指定pretrained=True以自动下载预训练的模型参数。
  46. pretrained_net = torchvision.models.resnet18(pretrained=True)
  47.  
  48. # [定义需要finetune的模型]
  49. finetune_net = torchvision.models.resnet18(pretrained=True)
  50. # 因为我们要做的是一个二分类问题，所以只需要改变最后一层的输出分类为2，然后只初始化最后一层的weight
  51. finetune_net.fc = nn.Linear(finetune_net.fc.in_features, 2)
  52. nn.init.xavier_uniform_(finetune_net.fc.weight)
  53.  
  54.  
  55. # [微调模型]
  56. # 首先，我们定义了一个训练函数train_fine_tuning，该函数使用微调，因此可以多次调用
  57. # 如果param_group=True，输出层中的模型参数将使用十倍的学习率
  58. def train_fine_tuning(net, learning_rate, batch_size=128, num_epochs=5, param_group=True):
  59.     train_iter = torch.utils.data.DataLoader(torchvision.datasets.ImageFolder(os.path.join(data_dir, 'train'), transform=train_augs), batch_size=batch_size, shuffle=True)
  60.     test_iter  = torch.utils.data.DataLoader(torchvision.datasets.ImageFolder(os.path.join(data_dir, 'test'), transform=test_augs), batch_size=batch_size)
  61.     devices = d2l.try_all_gpus()
  62.     loss = nn.CrossEntropyLoss(reduction="none")
  63.     if param_group:
  64.         # 取出除了最后一层(fc)之外的全部层的参数：params_1x
  65.         params_1x = [param for name, param in net.named_parameters() if name not in ["fc.weight", "fc.bias"]]
  66.         # params_1x使用正常的学习率，但是fc的学习率为正常学习率的十倍
  67.         trainer = torch.optim.SGD([{'params': params_1x},
  68.                                    {'params': net.fc.parameters(),
  69.                                     'lr': learning_rate * 10}],
  70.                                      lr=learning_rate, weight_decay=0.001)
  71.     else:
  72.         trainer = torch.optim.SGD(net.parameters(), lr=learning_rate, weight_decay=0.001)
  73.     # 画出
  74.     d2l.train_ch13(net, train_iter, test_iter, loss, trainer, num_epochs, devices)
  75.     d2l.plt.show()
  76.  
  77.  
  78. # 我们[使用较小的学习率]，通过微调预训练获得的模型参数。
  79. train_fine_tuning(finetune_net, 5e-5)
  80.  
  81.  
  82.

有finetune的效果：

没有finetune的效果：

# [为了进行比较，]我们定义了一个相同的模型，但是将其(所有模型参数初始化为随机值)。 由于整个模型需要从头开始训练，因此我们需要使用更大的学习率。

 

 

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号