arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2503.23402cs.CVcs.AI

超越合成回放:将扩散特征转化为小样本类别增量学习知识

Beyond Synthetic Replays: Turning Diffusion Features into Few-Shot Class-Incremental Learning Knowledge

UNIST(蔚山科学技术院);NAVER AI Lab(NAVER人工智能实验室);DeepBrain AI

Junsu Kim, Yunhoe Ku, Dongyoon Han, Seungryul Baek

AI总结:

本文提出 Diffusion-FSCIL,将 Stable Diffusion 的多尺度特征作为统一骨干,通过反演、条件合成、噪声注入和生成回放支持小样本类别增量学习,并在三个基准上达到最先进性能。

摘要:

小样本类别增量学习(FSCIL)极具挑战性,因为训练数据极其有限,同时又要求模型在获取新知识时不发生灾难性遗忘。近期工作探索了使用生成模型,尤其是 Stable Diffusion(SD),来应对这些挑战。然而,现有方法主要将 SD 用作回放生成器,而我们证明 SD 丰富的多尺度表征可以作为统一骨干网络。受此观察启发,我们提出 Diffusion-FSCIL,它从 SD 中提取四种协同特征类型:通过反演捕获真实图像特征,通过类别条件合成提供语义多样性,通过受控噪声注入增强泛化,并借助生成特征实现无需图像存储的回放。与需要合成缓冲区和独立分类骨干的传统方法不同,我们的统一框架完全在潜空间中运行,仅需轻量级网络(约 6M 参数)。在 CUB-200、miniImageNet 和 CIFAR-100 上的大量实验展示了最先进性能,全面消融实验确认了每种特征类型的必要性。此外,我们证实精简变体在显著提升效率的同时保持了具有竞争力的准确率,确立了生成模型作为 FSCIL 实用且有效骨干的可行性。

英文摘要:

Few-shot class-incremental learning (FSCIL) is challenging due to extremely limited training data while requiring models to acquire new knowledge without catastrophic forgetting. Recent works have explored generative models, particularly Stable Diffusion (SD), to address these challenges. However, existing approaches use SD mainly as a replay generator, whereas we demonstrate that SD's rich multi-scale representations can serve as a unified backbone. Motivated by this observation, we introduce Diffusion-FSCIL, which extracts four synergistic feature types from SD by capturing real image characteristics through inversion, providing semantic diversity via class-conditioned synthesis, enhancing generalization through controlled noise injection, and enabling replay without image storage through generative features. Unlike conventional approaches requiring synthetic buffers and separate classification backbones, our unified framework operates entirely in the latent space with only lightweight networks ($\approx$6M parameters). Extensive experiments on CUB-200, miniImageNet, and CIFAR-100 demonstrate state-of-the-art performance, with comprehensive ablations confirming the necessity of each feature type. Furthermore, we confirm that our streamlined variant maintains competitive accuracy while substantially improving efficiency, establishing the viability of generative models as practical and effective backbones for FSCIL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑