用类脑记忆机制提升图像识别的数据效率与可解释性
Vision Hopfield Memory Networks for Image Recognition
- 引入层级记忆模块,结合局部与全局联想记忆动态
- 在小数据集上表现优异,ImageNet上性能媲美主流模型
- 适合追求可解释性与低数据依赖的视觉任务研究者
近期的视觉主干网络如Transformer和Mamba等在图像识别上取得了显著进展。然而,这些架构仍远未遵循人脑的计算原理,通常需要大量训练数据且可解释性有限。我们提出视觉霍普菲尔德记忆网络(V-HMN),一种类脑的视觉主干网络,通过分层记忆机制与迭代精炼更新实现统一建模。具体而言,V-HMN包含局部霍普菲尔德模块(在图像块层面提供关联记忆动态)、全局霍普菲尔德模块(作为情景记忆进行上下文调制),以及受预测编码启发的迭代误差修正规则。通过层次化组织这些基于记忆的模块,V-HMN能够统一捕捉局部与全局动态。记忆检索揭示了输入与存储模式之间的关系,以原型形式提供可解释性,同时复用存储模式提升了数据效率。在公开图像分类基准上的广泛实验表明,V-HMN在小规模和中等规模数据集上表现强劲,在ImageNet上虽经极少调优仍保持竞争力,同时展现出更高的数据效率与原型驱动的可解释性。这些发现凸显了以记忆为核心的V-HMN作为标准视觉主干替代方案的潜力,推动类脑计算与现代机器学习的融合。
原文摘要 · Abstract (English)
Recent vision backbones, such as Transformer families and state-space models like Mamba, have achieved remarkable progress on image recognition. Despite their empirical success, these architectures remain far from the computational principles of the human brain, often demanding enormous amounts of training data while offering limited interpretability. We propose the Vision Hopfield Memory Network (V-HMN), a brain-inspired vision backbone that integrates hierarchical memory mechanisms across layers with iterative refinement updates. Specifically, V-HMN incorporates local Hopfield modules that provide associative memory dynamics at the image patch level, global Hopfield modules that function as episodic memory for contextual modulation, and a predictive-coding-inspired refinement rule for iterative error correction. By organizing these memory-based modules hierarchically, V-HMN captures both local and global dynamics in a unified framework. Memory retrieval exposes the relationship between inputs and stored patterns, providing a prototype-based form of interpretability through explicit memory retrieval, while the reuse of stored patterns improves data efficiency. This brain-inspired design therefore enhances data efficiency and provides a prototype-based form of interpretability compared to existing self-attention- or state-space-based approaches. We conducted extensive experiments on public image classification benchmarks. V-HMN achieves strong performance on small- and medium-scale benchmarks, and remains competitive with widely adopted backbone architectures on ImageNet despite minimal architectural tuning, while offering improved data efficiency and a prototype-based form of interpretability. These findings highlight the potential of V-HMN as a memory-centric alternative to standard vision backbones, thereby bridging brain-inspired computation with modern machine learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。