模仿海马体结构提升目标检测的记性与泛化能力
Hippocampus-DETR: An Explicit Memory Object Detection Framework Based on Hippocampus Modeling
- 用海马体各区域建模记忆模块,实现特征分离与补全
- 在少样本分类等任务中准确率更高,数据效率显著提升
- 适合追求高效学习与可解释性的视觉模型研究者
本文针对当前目标检测模型缺乏显式记忆机制的问题,提出基于生物海马体建模的Hippocampus-DETR框架。该框架将海马体网络模块HipNet融入DETR架构,系统模拟海马体亚区(内嗅皮层、齿状回、CA3、CA1和下托)的解剖结构与功能组织,实现视觉编码特征的模式分离、模式补全、重要性过滤与信息整合。训练时采用分层优化策略,使各记忆子模块协同形成具备记忆检索与补全能力的完整记忆系统。实验表明,Hippocampus-DETR在主流检测任务中精度优于现有模型;更重要的是,在少样本图像分类、多模态特征构建与图像恢复等任务中表现出优异的泛化能力与数据效率。后续实验验证了各记忆子模块的功能必要性与内部可解释性。本研究不仅提供了一种新型目标检测框架,也为神经认知机制与深度学习融合提供了可行路径,具有提升模型学习效率与任务鲁棒性的显著价值。项目代码已开源:https://github.com/2186cloud/hipnet。
原文摘要 · Abstract (English)
This paper addresses the lack of explicit memory mechanisms in current object detection models and proposes Hippocampus-DETR, a novel detection framework based on biological hippocampal memory modeling. This framework integrates a hippocampal memory network module, HipNet, into the DETR architecture and systematically simulates the anatomical structure and functional organization of hippocampal subregions, including the entorhinal cortex, dentate gyrus, CA3, CA1, and subiculum. Through this design, Hippocampus-DETR realizes pattern separation, pattern completion, importance filtering, and information integration of visual encoding features. During training, different memory submodules are optimized using a layer-wise training strategy, ultimately forming a memory system with memory retrieval and completion capabilities. Experimental results demonstrate that Hippocampus-DETR achieves higher detection accuracy than current mainstream models. More importantly, models equipped with this framework also exhibit excellent generalization ability and data efficiency in tasks such as few-shot image classification, multimodal feature construction, and image restoration. Subsequent experiments further validate the functional necessity and internal interpretability of each memory submodule. This study not only provides a novel object detection framework, but also offers a feasible technical pathway for integrating neurocognitive mechanisms with deep learning models, highlighting its significant value in improving model learning efficiency and task robustness. The project is available at https://github.com/2186cloud/hipnet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。