【可解释】|Layer CAM

主页:
https://mmcheng.net/layercam/

发现问题

Grad CAM浅层特征图的显著图噪声过多。
在这里插入图片描述

分析:

浅层权重方差变化大,不能使用梯度均值为特征图每个像素赋予同等重要度

在这里插入图片描述

提出解决方法

LayerCAM:基于元素的,每一张特征图中每一个元素都有一个对应的权重。
在这里插入图片描述
在这里插入图片描述

实验验证

1 弱监督目标定位实验

ILSVRC 50000张验证集

定位精度由loc1和loc5指标衡量。如果估计边界框和地面真值边界框之间的联合交集(IoU)大于或等于0.5,同时前1个预测类是正确的,则loc1度量表示估计结果属于正确类别。loc5指标用于预测的前5个类别。
在这里插入图片描述
在这里插入图片描述
不同层特征图融合
在这里插入图片描述
在这里插入图片描述
图像遮挡实验
在这里插入图片描述

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
1D-Grad-CAM是一种基于梯度的可视化方法,用于理解深度学习模型在输入序列中的关注点。下面是使用Pytorch实现1D-Grad-CAM的步骤: 1. 首先,加载训练好的模型和输入序列。可以使用torchvision.models中的预训练模型,例如resnet18。 2. 然后,定义一个Grad-CAM类,该类包含一个前向传递函数和一个反向传递函数。前向传递函数计算模型输出和特定层的特征图,反向传递函数计算特征图相对于输出的梯度。 3. 接下来,使用Grad-CAM类计算输入序列的梯度。这可以通过将输入序列传递给前向传递函数,然后将输出和特定层的特征图传递给反向传递函数来完成。 4. 最后,将梯度与特征图相乘,并将结果求和。这将生成一个热力图,用于可视化模型在输入序列中的关注点。 下面是一个使用Pytorch实现1D-Grad-CAM的示例代码: ```python import torch import torch.nn as nn import torch.nn.functional as F from torchvision import models class GradCAM: def __init__(self, model, target_layer): self.model = model self.target_layer = target_layer self.feature_maps = None self.gradient = None def save_feature_maps(self, module, input, output): self.feature_maps = output.detach() def save_gradient(self, grad): self.gradient = grad.detach() def forward(self, x): for name, module in self.model.named_modules(): if name == self.target_layer: module.register_forward_hook(self.save_feature_maps) module.register_backward_hook(self.save_gradient) break output = self.model(x) output = F.softmax(output, dim=1) return output def backward(self): self.gradient = torch.mean(self.gradient, dim=[2, 3], keepdim=True) feature_maps_weights = torch.mean(self.gradient * self.feature_maps, dim=1, keepdim=True) cam = F.relu(torch.sum(feature_maps_weights * self.feature_maps, dim=1, keepdim=True)) cam = F.interpolate(cam, size=x.shape[-1], mode='linear', align_corners=False) cam = cam.squeeze() cam = cam - torch.min(cam) cam = cam / torch.max(cam) return cam # 加载模型和输入序列 model = models.resnet18(pretrained=True) x = torch.randn(1, 3, 224, 224) # 创建Grad-CAM对象并计算热力图 grad_cam = GradCAM(model, 'layer4') output = grad_cam.forward(x) output[:, 0].backward() cam = grad_cam.backward() # 可视化热力图 import matplotlib.pyplot as plt plt.imshow(cam.detach().numpy()) plt.show() ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值