当前位置: 首页 > news >正文

Day 51 在预训练 ResNet18 中注入

预训练 ResNet18 模型注入方法

在预训练的 ResNet18 模型中注入自定义模块或修改现有结构时,需确保不破坏原有特征提取能力。以下是常见注入方法:

修改特定层结构通过替换或扩展特定层(如卷积层、全连接层)实现功能注入。例如,在最后一层前插入自定义模块:

import torch.nn as nn from torchvision.models import resnet18 model = resnet18(pretrained=True) model.fc = nn.Sequential( nn.Linear(512, 256), nn.ReLU(), nn.Linear(256, 10) # 修改输出维度 )

插入注意力机制在中间层注入注意力模块(如SEBlock):

class SEBlock(nn.Module): def __init__(self, channels, reduction=16): super().__init__() self.squeeze = nn.AdaptiveAvgPool2d(1) self.excitation = nn.Sequential( nn.Linear(channels, channels // reduction), nn.ReLU(), nn.Linear(channels // reduction, channels), nn.Sigmoid() ) model.layer2.add_module("se_block", SEBlock(128))

特征融合注入通过跳跃连接将不同层特征融合:

class FusionModule(nn.Module): def __init__(self, in_channels): super().__init__() self.conv = nn.Conv2d(in_channels*2, in_channels, 1) def forward(self, x1, x2): x = torch.cat([x1, x2], dim=1) return self.conv(x) fusion = FusionModule(64) model.layer1.register_forward_hook(lambda m, inp, out: fusion(out, inp[0]))

注入后的微调策略

分层学习率设置对注入模块使用更高学习率,预训练层采用较低学习率:

optimizer = torch.optim.SGD([ {'params': model.parameters(), 'lr': 0.001}, {'params': fusion.parameters(), 'lr': 0.01} ], momentum=0.9)

渐进式解冻先冻结所有层仅训练注入模块,逐步解冻深层:

for param in model.parameters(): param.requires_grad = False fusion.train() # 后续按epoch逐步解冻layer4、layer3等

验证注入效果

特征可视化使用Grad-CAM等工具比较注入前后特征响应:

from torchcam.methods import GradCAM cam_extractor = GradCAM(model, 'layer4') activations = cam_extractor(input_tensor)

性能指标对比记录注入前后的验证集准确率、推理速度等关键指标,确保注入不会显著降低模型效率。典型评估流程:

with torch.no_grad(): for data in val_loader: outputs = model(data) # 计算accuracy, F1-score等

rogbbs.asus.com.cn/topics/101/feeds/63030
rogbbs.asus.com.cn/topics/101/feeds/63031
rogbbs.asus.com.cn/topics/101/feeds/63032
rogbbs.asus.com.cn/topics/101/feeds/63033
rogbbs.asus.com.cn/topics/101/feeds/63034
rogbbs.asus.com.cn/topics/101/feeds/63035
rogbbs.asus.com.cn/topics/101/feeds/63036
rogbbs.asus.com.cn/topics/101/feeds/63037
rogbbs.asus.com.cn/topics/101/feeds/63038
rogbbs.asus.com.cn/topics/101/feeds/63039
rogbbs.asus.com.cn/topics/101/feeds/63040
rogbbs.asus.com.cn/topics/101/feeds/63041
rogbbs.asus.com.cn/topics/101/feeds/63044
rogbbs.asus.com.cn/topics/101/feeds/63045
rogbbs.asus.com.cn/topics/101/feeds/63047
rogbbs.asus.com.cn/topics/101/feeds/63048
rogbbs.asus.com.cn/topics/101/feeds/63049
rogbbs.asus.com.cn/topics/101/feeds/63050
rogbbs.asus.com.cn/topics/101/feeds/63046
rogbbs.asus.com.cn/topics/101/feeds/63052
rogbbs.asus.com.cn/topics/101/feeds/63053
rogbbs.asus.com.cn/topics/101/feeds/63043
rogbbs.asus.com.cn/topics/101/feeds/63055
rogbbs.asus.com.cn/topics/101/feeds/63056
rogbbs.asus.com.cn/topics/101/feeds/63057
rogbbs.asus.com.cn/topics/101/feeds/63058
rogbbs.asus.com.cn/topics/101/feeds/63059
rogbbs.asus.com.cn/topics/101/feeds/63062
rogbbs.asus.com.cn/topics/101/feeds/63060
rogbbs.asus.com.cn/topics/101/feeds/63051
rogbbs.asus.com.cn/topics/101/feeds/63063
rogbbs.asus.com.cn/topics/101/feeds/63064
rogbbs.asus.com.cn/topics/101/feeds/63067
rogbbs.asus.com.cn/topics/101/feeds/63066
rogbbs.asus.com.cn/topics/101/feeds/63070
rogbbs.asus.com.cn/topics/101/feeds/63071
rogbbs.asus.com.cn/topics/101/feeds/63072
rogbbs.asus.com.cn/topics/101/feeds/63073
rogbbs.asus.com.cn/topics/101/feeds/63069
rogbbs.asus.com.cn/topics/101/feeds/63074
rogbbs.asus.com.cn/topics/101/feeds/63075
rogbbs.asus.com.cn/topics/101/feeds/63076
rogbbs.asus.com.cn/topics/101/feeds/63078
rogbbs.asus.com.cn/topics/101/feeds/63079
rogbbs.asus.com.cn/topics/101/feeds/63080
rogbbs.asus.com.cn/topics/101/feeds/63082
rogbbs.asus.com.cn/topics/101/feeds/63084
rogbbs.asus.com.cn/topics/101/feeds/63085
rogbbs.asus.com.cn/topics/101/feeds/63086
rogbbs.asus.com.cn/topics/101/feeds/63088
rogbbs.asus.com.cn/topics/101/feeds/63087
rogbbs.asus.com.cn/topics/101/feeds/63089
rogbbs.asus.com.cn/topics/101/feeds/63090
rogbbs.asus.com.cn/topics/101/feeds/63091
rogbbs.asus.com.cn/topics/101/feeds/63092
rogbbs.asus.com.cn/topics/101/feeds/63093
rogbbs.asus.com.cn/topics/101/feeds/63094
rogbbs.asus.com.cn/topics/101/feeds/63095
rogbbs.asus.com.cn/topics/101/feeds/63096
rogbbs.asus.com.cn/topics/101/feeds/63097
rogbbs.asus.com.cn/topics/101/feeds/63098
rogbbs.asus.com.cn/topics/101/feeds/63099
rogbbs.asus.com.cn/topics/101/feeds/63100
rogbbs.asus.com.cn/topics/101/feeds/63101
rogbbs.asus.com.cn/topics/101/feeds/63102
rogbbs.asus.com.cn/topics/101/feeds/63103
rogbbs.asus.com.cn/topics/101/feeds/63104
rogbbs.asus.com.cn/topics/101/feeds/63107
rogbbs.asus.com.cn/topics/101/feeds/63105
rogbbs.asus.com.cn/topics/101/feeds/63106
rogbbs.asus.com.cn/topics/101/feeds/63108
rogbbs.asus.com.cn/topics/101/feeds/63109
rogbbs.asus.com.cn/topics/101/feeds/63110
rogbbs.asus.com.cn/topics/101/feeds/63111
rogbbs.asus.com.cn/topics/101/feeds/63112
rogbbs.asus.com.cn/topics/101/feeds/63113
rogbbs.asus.com.cn/topics/101/feeds/63114
rogbbs.asus.com.cn/topics/101/feeds/63115
rogbbs.asus.com.cn/topics/101/feeds/63116
rogbbs.asus.com.cn/topics/101/feeds/63117
rogbbs.asus.com.cn/topics/101/feeds/63118
rogbbs.asus.com.cn/topics/101/feeds/63119
rogbbs.asus.com.cn/topics/101/feeds/63121
rogbbs.asus.com.cn/topics/101/feeds/63122
rogbbs.asus.com.cn/topics/101/feeds/63123
rogbbs.asus.com.cn/topics/101/feeds/63124
rogbbs.asus.com.cn/topics/101/feeds/63125
rogbbs.asus.com.cn/topics/101/feeds/63126
rogbbs.asus.com.cn/topics/101/feeds/63127
rogbbs.asus.com.cn/topics/101/feeds/63128
rogbbs.asus.com.cn/topics/101/feeds/63129
rogbbs.asus.com.cn/topics/101/feeds/63130
rogbbs.asus.com.cn/topics/101/feeds/63131
rogbbs.asus.com.cn/topics/101/feeds/63132
rogbbs.asus.com.cn/topics/101/feeds/63133
rogbbs.asus.com.cn/topics/101/feeds/63134
rogbbs.asus.com.cn/topics/101/feeds/63135
rogbbs.asus.com.cn/topics/101/feeds/63136
rogbbs.asus.com.cn/topics/101/feeds/63137
rogbbs.asus.com.cn/topics/101/feeds/63138
rogbbs.asus.com.cn/topics/101/feeds/63139
rogbbs.asus.com.cn/topics/101/feeds/63140
rogbbs.asus.com.cn/topics/101/feeds/63142
rogbbs.asus.com.cn/topics/101/feeds/63143
rogbbs.asus.com.cn/topics/101/feeds/63144
rogbbs.asus.com.cn/topics/101/feeds/63146
rogbbs.asus.com.cn/topics/101/feeds/63145
rogbbs.asus.com.cn/topics/101/feeds/63120
rogbbs.asus.com.cn/topics/101/feeds/63148
rogbbs.asus.com.cn/topics/101/feeds/63150
rogbbs.asus.com.cn/topics/101/feeds/63149
rogbbs.asus.com.cn/topics/101/feeds/63151
rogbbs.asus.com.cn/topics/101/feeds/63152
rogbbs.asus.com.cn/topics/101/feeds/63154
rogbbs.asus.com.cn/topics/101/feeds/63153
rogbbs.asus.com.cn/topics/101/feeds/63141
rogbbs.asus.com.cn/topics/101/feeds/63155
rogbbs.asus.com.cn/topics/101/feeds/63156
rogbbs.asus.com.cn/topics/101/feeds/63157
rogbbs.asus.com.cn/topics/101/feeds/63158
rogbbs.asus.com.cn/topics/101/feeds/63159
rogbbs.asus.com.cn/topics/101/feeds/63160
rogbbs.asus.com.cn/topics/101/feeds/63161
rogbbs.asus.com.cn/topics/101/feeds/63162
rogbbs.asus.com.cn/topics/101/feeds/63163
rogbbs.asus.com.cn/topics/101/feeds/63164
rogbbs.asus.com.cn/topics/101/feeds/63165
rogbbs.asus.com.cn/topics/101/feeds/63167
rogbbs.asus.com.cn/topics/101/feeds/63168
rogbbs.asus.com.cn/topics/101/feeds/63190
rogbbs.asus.com.cn/topics/101/feeds/63166
rogbbs.asus.com.cn/topics/101/feeds/63184
rogbbs.asus.com.cn/topics/101/feeds/63177
rogbbs.asus.com.cn/topics/101/feeds/63178
rogbbs.asus.com.cn/topics/101/feeds/63174
rogbbs.asus.com.cn/topics/101/feeds/63169
rogbbs.asus.com.cn/topics/101/feeds/63173
rogbbs.asus.com.cn/topics/101/feeds/63187
rogbbs.asus.com.cn/topics/101/feeds/63179
rogbbs.asus.com.cn/topics/101/feeds/63180
rogbbs.asus.com.cn/topics/101/feeds/63185
rogbbs.asus.com.cn/topics/101/feeds/63175
rogbbs.asus.com.cn/topics/101/feeds/63172
rogbbs.asus.com.cn/topics/101/feeds/63181
rogbbs.asus.com.cn/topics/101/feeds/63186
rogbbs.asus.com.cn/topics/101/feeds/63182
rogbbs.asus.com.cn/topics/101/feeds/63183
rogbbs.asus.com.cn/topics/101/feeds/63188
rogbbs.asus.com.cn/topics/101/feeds/63176
rogbbs.asus.com.cn/topics/101/feeds/63170
rogbbs.asus.com.cn/topics/101/feeds/63189
rogbbs.asus.com.cn/topics/101/feeds/63191
rogbbs.asus.com.cn/topics/101/feeds/63192
rogbbs.asus.com.cn/topics/101/feeds/63193
rogbbs.asus.com.cn/topics/101/feeds/63194
rogbbs.asus.com.cn/topics/101/feeds/63195
rogbbs.asus.com.cn/topics/101/feeds/63196
rogbbs.asus.com.cn/topics/101/feeds/63197
rogbbs.asus.com.cn/topics/101/feeds/63199
rogbbs.asus.com.cn/topics/101/feeds/63200
rogbbs.asus.com.cn/topics/101/feeds/63201
rogbbs.asus.com.cn/topics/101/feeds/63203
rogbbs.asus.com.cn/topics/101/feeds/63204
rogbbs.asus.com.cn/topics/101/feeds/63205
rogbbs.asus.com.cn/topics/101/feeds/63206
rogbbs.asus.com.cn/topics/101/feeds/63208
rogbbs.asus.com.cn/topics/101/feeds/63207
rogbbs.asus.com.cn/topics/101/feeds/63209
rogbbs.asus.com.cn/topics/101/feeds/63210
rogbbs.asus.com.cn/topics/101/feeds/63211
rogbbs.asus.com.cn/topics/101/feeds/63212
rogbbs.asus.com.cn/topics/101/feeds/63213
rogbbs.asus.com.cn/topics/101/feeds/63215
rogbbs.asus.com.cn/topics/101/feeds/63214
rogbbs.asus.com.cn/topics/101/feeds/63216
rogbbs.asus.com.cn/topics/101/feeds/63217
rogbbs.asus.com.cn/topics/101/feeds/63218
rogbbs.asus.com.cn/topics/101/feeds/63220
rogbbs.asus.com.cn/topics/101/feeds/63219
rogbbs.asus.com.cn/topics/101/feeds/63221
rogbbs.asus.com.cn/topics/101/feeds/63224
rogbbs.asus.com.cn/topics/101/feeds/63222
rogbbs.asus.com.cn/topics/101/feeds/63223
rogbbs.asus.com.cn/topics/101/feeds/63225
rogbbs.asus.com.cn/topics/101/feeds/63226
rogbbs.asus.com.cn/topics/101/feeds/63227
rogbbs.asus.com.cn/topics/101/feeds/63228
rogbbs.asus.com.cn/topics/101/feeds/63229
rogbbs.asus.com.cn/topics/101/feeds/63231
rogbbs.asus.com.cn/topics/101/feeds/63230
rogbbs.asus.com.cn/topics/101/feeds/63233
rogbbs.asus.com.cn/topics/101/feeds/63234
rogbbs.asus.com.cn/topics/101/feeds/63232
rogbbs.asus.com.cn/topics/101/feeds/63235
rogbbs.asus.com.cn/topics/101/feeds/63236
rogbbs.asus.com.cn/topics/101/feeds/63237
rogbbs.asus.com.cn/topics/101/feeds/63238
rogbbs.asus.com.cn/topics/101/feeds/63239
rogbbs.asus.com.cn/topics/101/feeds/63240
rogbbs.asus.com.cn/topics/101/feeds/63241
rogbbs.asus.com.cn/topics/101/feeds/63242
rogbbs.asus.com.cn/topics/101/feeds/63247
rogbbs.asus.com.cn/topics/101/feeds/63248
rogbbs.asus.com.cn/topics/101/feeds/63250
rogbbs.asus.com.cn/topics/101/feeds/63245
rogbbs.asus.com.cn/topics/101/feeds/63244
rogbbs.asus.com.cn/topics/101/feeds/63253
rogbbs.asus.com.cn/topics/101/feeds/63254
rogbbs.asus.com.cn/topics/101/feeds/63246
rogbbs.asus.com.cn/topics/101/feeds/63256
rogbbs.asus.com.cn/topics/101/feeds/63255
rogbbs.asus.com.cn/topics/101/feeds/63257
rogbbs.asus.com.cn/topics/101/feeds/63258
rogbbs.asus.com.cn/topics/101/feeds/63259
rogbbs.asus.com.cn/topics/101/feeds/63260
rogbbs.asus.com.cn/topics/101/feeds/63261
rogbbs.asus.com.cn/topics/101/feeds/63262
rogbbs.asus.com.cn/topics/101/feeds/63263
rogbbs.asus.com.cn/topics/101/feeds/63264
rogbbs.asus.com.cn/topics/101/feeds/63265
rogbbs.asus.com.cn/topics/101/feeds/63266

http://www.jsqmd.com/news/173232/

相关文章:

  • 拉链用一年就坏?一件好羽绒服的“隐形品质”,都藏在这三个细节里!
  • WebGIS开发智慧校园(1)GIS开发的基本概念
  • C#上位机大数据量处理:异步采集+多线程解析
  • 洞察趋势:低代码如何驱动企业数字化转型效率提升?
  • 西电考研 | 26官方复试专业课参考书目汇总,强烈推荐!
  • DeepSeek总结的SQL 数独:约束编程
  • Redis Cluster:从数据分布到故障转移的全方位解读
  • 【毕业设计】基于Springboot的牧场管理系统的设计与实现(源码+文档+远程调试,全bao定制等)
  • Java计算机毕设之基于Springboot的牧场管理系统的设计与实现养殖管理记录牲畜档案(完整前后端代码+说明文档+LW,调试定制等)
  • 【JetCompose】入门教程实战基础案例02之列表项显隐效果
  • 【JetCompose】入门教程实战基础案例03之凡尔赛朋友圈
  • 高频Robot Framework软件测试面试题
  • 全面讲解并发编程模式,重点分析其在 .NET 中的实现
  • NET Runtime 8.0.13实战深度解析
  • Overcoming the uncertainty challenges in detecting building changes from remote sensing images
  • 【课程设计/毕业设计】基于springboot的畜牧管理系统的设计与实现 基于Springboot的牧场管理系统的设计与实现【附源码、数据库、万字文档】
  • 2025-12-31
  • 12月第二篇笔记
  • Java毕设选题推荐:基于Springboot的牧场管理系统的设计与实现基于springboot的畜牧管理系统的设计与实现 【附源码、mysql、文档、调试+代码讲解+全bao等】
  • 全网最详细的软件测试面试题总结+基础知识(完整版)
  • 链表的逻辑 - f
  • 6个值得收藏的AI论文工具,智能降重与改写让论文更流畅且规避查重问题
  • [KaibaMath]1035 调和平均数的一个基本性质及证明
  • 年薪50W的软件测试面试题,到底长啥样?
  • 腾讯智能工作台ima.copilot
  • [KaibaMath]1034 关于AS⁻B=BS⁻A的证明
  • 大模型呼叫中心选型指南:从七大厂商透视到三步决策法
  • AI论文降重平台排名:9款工具实测,开题报告撰写功能同样出色
  • 【深入理解 Android 中的 build.gradle 文件】
  • 计算机Java毕设实战-基于Springboot的牧场管理系统的设计与实现springboot的牧场信息化系统【完整源码+LW+部署说明+演示视频,全bao一条龙等】