中文

构建灵活感知的视觉记忆

计算机视觉与模式识别 2025-08-14 v3 人工智能 机器学习

摘要

训练神经网络是一个单一的工程,类似于将知识雕刻成石头:一旦过程完成,编辑网络中的知识就困难了,因为所有信息都分布在网络权重中。我们在此探索一种简单而有力的替代方案,通过将深度神经网络的表征能力与数据库的灵活性相结合。将图像分类任务分解为图像相似性(来自预训练嵌入)和通过快速最近邻检索从知识数据库中搜索,我们利用众所周知的组件构建一种简单且灵活的视觉记忆,其具有以下关键功能:(1)在不同尺度上灵活添加数据的能力:从单个样本到整个类别乃至十亿规模的数据;(2)通过不可知和记忆修剪删除数据的能力;(3)可干预控制其行为的可解释决策机制。综上,这些功能全面展示了显式视觉记忆的优势。我们希望这可能有助于关于深度视觉模型中知识应如何表示的讨论——超越将其雕刻在“石头”权重中。

关键词

引用

@article{arxiv.2408.08172,
  title  = {Towards flexible perception with visual memory},
  author = {Robert Geirhos and Priyank Jaini and Austin Stone and Sourabh Medapati and Xi Yi and George Toderici and Abhijit Ogale and Jonathon Shlens},
  journal= {arXiv preprint arXiv:2408.08172},
  year   = {2025}
}

备注

ICML 2025 camera ready version