如何在分类任务中可视化热图？

Question

我现在正在使用类似 ViT 的模型执行分类任务。我正在尝试创建这样的热图。

我查了一下网站并发现了这个，但没有给出解决方案。所以我试图再次提起它。帮助我吧伙计们！非常感谢。

Answer 1

我目前正在做一个这样的项目。虽然我无法为您提供实际的代码，但我可以为您提供实际执行此操作的博客文章和参考。

• https://jacobgil.github.io/deeplearning/vision-transformer-explainability
• https://www.kaggle.com/code/piantic/vision-transformer-vit-visualize-attention-map

这两篇文章都提供了代码示例，第一篇文章更深入地探讨了其背后的理论。我希望这有帮助。