CAM 和 Hooks

类激活图(CAM) 由 Bolei Zhou 等人介绍。在 “为判别性本地化学习深层特征”中。它使用最后的输出 卷积层（就在平均池化层之前）与预测一起为我们提供了模型做出决定的原因的热图可视化。这是一个有用的解释工具。

更准确地说，在最后一个卷积层的每个位置，我们有与最后一个线性层一样多的过滤器。因此，我们可以计算这些激活与最终权重的点积，以获得特征图上每个位置的用于做出决定的特征的分数。

我们将需要一种方法来在模型训练时访问模型内​​部的激活。在 PyTorch 中，这可以通过hook来完成。Hooks 相当于 PyTorch 的 fastai 的回调。Learner然而，钩子允许您将代码注入前向和后向计算本身，而不是让您像 fastai 回调一样将代码注入训练循环。我们可以将一个钩子附加到模型的任何层，它会在我们计算输出（前向钩子）或 反向传播（后向钩子）期间执行。前向钩子是一个接受三样东西的函数——一个模块、它的输入和它的输出——它可以执行你想要的任何行为。（fastai 还提供了一个方便 HookCallback的，我们不会在这里介绍，但看看 fastai 文档；它使使用钩子更容易一些。）

为了说明，我们将使用我们在第 1 章中训练的相同猫狗模型 ：

path = untar_data(URLs.PETS)/'images'
def is_cat(x): return x[0].isupper()
dls = ImageDataLoaders.from_name_func(
    path, get_image_files(path), valid_pct=0.2, seed=21,
    label_func=is_cat, item_tfms=Resize(224))
learn = cnn_learner(dls, resnet34, metrics=error_rate)
learn.fine_tune(1)

首先，我们将获取一张猫图片和一批数据：

img = PILImage.create(image_cat())
x, = first(dls.test_dl([img]))

对于 CAM，我们要存储最后一个卷积层的激活值。我们把我们的钩子函数放在一个类中，这样它就有一个我们可以稍后访问的状态，并且只存储输出的副本：

class Hook():
    def hook_func(self, m, i, o): self.stored = o.detach().clone()

然后我们可以实例化 一个Hook并将其附加到我们想要的层，即 CNN 主体的最后一层：

hook_output = Hook()
hook = learn.model[0].register_forward_hook(hook_output.hook_func)

现在我们可以抓取一批并通过我们的模型提供它：

with torch.no_grad(): output = learn.model.eval()(x)

我们可以访问我们存储的激活：

act = hook_output.stored[0]

让我们也仔细检查我们的预测：

F.softmax(output, dim=-1)

tensor([[7.3566e-07, 1.0000e+00]], device='cuda:0')

我们知道0(for False) 是“狗”，因为类在 fastai 中自动排序，但我们仍然可以通过查看来仔细检查dls.vocab：

dls.vocab

(#2) [False,True]

所以，我们的模型非常有信心这是一张猫的照片。

为了将我们的权重矩阵（2 乘以激活数）与激活（批量大小按行按列）进行点积，我们使用自定义einsum：

act.shape

torch.Size([512, 7, 7])

cam_map = torch.einsum('ck,kij->cij', learn.model[1][-1].weight, act)
cam_map.shape

torch.Size([2, 7, 7])

对于我们批次中的每张图像，对于每个类别，我们都会得到一个 7×7 的特征图，告诉我们哪里的激活值更高，哪里的激活值更低。这将使我们看到图片的哪些区域影响了模型的决定。

例如，我们可以找出哪些区域使模型决定这只动物是一只猫（请注意，DataLoader我们需要输入decode，x因为它已经被索引时不维护类型TensorImage——这可能会在您阅读本文时修复）：

x_dec = TensorImage(dls.train.decode((x,))[0][0])
_,ax = plt.subplots()
x_dec.show(ctx=ax)
ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,224,224,0),
              interpolation='bilinear', cmap='magma');

在这种情况下，亮黄色区域对应高激活，紫色区域对应低激活。在这种情况下，我们可以看到头部和前爪是让模型确定这是一张猫图片的两个主要区域。

一旦你完成了你的钩子，你应该删除它，否则它可能会泄漏一些内存：

hook.remove()

这就是为什么让 Hook类成为上下文管理器通常是个好主意，在您输入时注册挂钩 它并在您退出时将其删除。上下文管理器是一种 Python 结构，__enter__当对象在with 子句中创建时以及__exit__在子句末尾调用with。例如，这就是 Python 处理with open(...) as f:结构的方式，您经常会在打开文件时看到这种结构，而无需close(f)在末尾显式显示。

如果我们定义Hook如下

class Hook():
    def __init__(self, m):
        self.hook = m.register_forward_hook(self.hook_func)
    def hook_func(self, m, i, o): self.stored = o.detach().clone()
    def __enter__(self, *args): return self
    def __exit__(self, *args): self.hook.remove()

我们可以这样安全地使用它：

with Hook(learn.model[0]) as hook:
    with torch.no_grad(): output = learn.model.eval()(x.cuda())
    act = hook.stored

fastaiHook为您提供了这个类，以及一些其他方便的类，使使用钩子更容易。

此方法很有用，但仅适用于最后一层。Gradient CAM是解决这个问题的一个变体。

我们刚刚看到的方法让我们只计算具有最后激活的热图，因为一旦我们有了我们的特征，我们就必须将它们相乘 通过最后的权重矩阵。这不适用于网络的内层。2016 年论文 “Grad-CAM：你为什么这么说？”中介绍的变体 由 Ramprasaath R. Selvaraju 等人撰写。使用所需类别的最终激活的梯度。如果你还记得一点关于反向传递的知识，最后一层输出相对于该层输入的梯度等于层权重，因为它是一个线性层。

class HookBwd():
    def __init__(self, m):
        self.hook = m.register_backward_hook(self.hook_func)
    def hook_func(self, m, gi, go): self.stored = go[0].detach().clone()
    def __enter__(self, *args): return self
    def __exit__(self, *args): self.hook.remove()

cls = 1
with HookBwd(learn.model[0]) as hookg:
    with Hook(learn.model[0]) as hook:
        output = learn.model.eval()(x.cuda())
        act = hook.stored
    output[0,cls].backward()
    grad = hookg.stored

w = grad[0].mean(dim=[1,2], keepdim=True)
cam_map = (w * act[0]).sum(0)

_,ax = plt.subplots()
x_dec.show(ctx=ax)
ax.imshow(cam_map.detach().cpu(), alpha=0.6, extent=(0,224,224,0),
              interpolation='bilinear', cmap='magma');

with HookBwd(learn.model[0][-2]) as hookg:
    with Hook(learn.model[0][-2]) as hook:
        output = learn.model.eval()(x.cuda())
        act = hook.stored
    output[0,cls].backward()
    grad = hookg.stored

w = grad[0].mean(dim=[1,2], keepdim=True)
cam_map = (w * act[0]).sum(0)

_,ax = plt.subplots()
x_dec.show(ctx=ax)
ax.imshow(cam_map.detach().cpu(), alpha=0.6, extent=(0,224,224,0),
              interpolation='bilinear', cmap='magma');