Brain Inspired Adaptive Memory Dual-Net for Few-Shot Image Classification

作者: Kexin Di, Xiuxing Li, Yuyang Han, Ziyu Li, Qing Li, Xia Wu

发布时间: 2025-03-11

来源: arxiv

研究方向: 计算机视觉,少样本图像分类,生物启发式方法

主要内容

本文提出了一种名为SCAM-Net的少样本图像分类方法,该方法受人类大脑的通用优化系统巩固机制启发,通过构建海马体-新皮层双网络结构,模拟系统巩固过程,以解决少样本场景下特征识别困难的问题。

主要贡献

1. 提出了一种受生物启发的自适应记忆双网络,首次将优化泛化的原理引入到系统巩固双网络中。

2. 引入了SCAM-Net技术,该技术通过海马体-新皮层双模型巩固结构化信息,并通过与优化泛化原理对齐的先验知识自适应地调整结构化信息。

3. 在四个基准数据集上进行了广泛实验,实现了最先进的性能。

研究方法

1. 构建了一个海马体-新皮层双网络,用于巩固每个类别的结构化表示。

2. 在皮层中存储和自适应调节结构化表示,遵循“泛化优化”原则。

3. 使用掩码图像建模(MIM)进行自监督预训练。

4. 利用自监督预训练得到的嵌入空间,计算支持集和查询集之间的相似性矩阵。

5. 使用一致性损失来强制海马体模型的预测与皮层模型保持一致。

6. 在皮层模型中执行系统巩固,通过指数移动平均(EMA)缓慢更新海马体网络的权重。

7. 在皮层中维护一个用于自适应记忆调节的长期记忆,根据先验知识对CLS标记进行调节。

实验结果

在miniImagenet、tieredImagenet、CIFAR-FS和FC-100四个基准数据集上,SCAM-Net在1个样本和5个样本的少样本分类任务中取得了最先进的性能。

未来工作

未来研究可以探索以下方向:1) 将SCAM-Net应用于其他少样本学习任务;2) 将SCAM-Net与其他少样本学习方法结合;3) 探索更有效的自适应记忆调节机制。