arXiv:2505.11793cs.CVcs.AI2025-05被引 22

用胶囊网络+持续学习,提升高光谱异常检测跨场景能力

CL-CaGAN: Capsule differential adversarial continuous learning for cross-domain hyperspectral anomaly detection

  • 设计胶囊结构结合对抗学习,弥补先验信息不足
  • 通过聚类重放与自蒸馏,缓解灾难性遗忘问题
  • 适合需要持续学习的高光谱异常检测实际应用

高光谱图像(HSI)中的异常检测(AD)受到广泛关注,现有深度学习方法在特定场景下表现良好,但在开放场景中面临先验信息有限和灾难性遗忘的挑战。为此,提出一种基于持续学习的胶囊差分生成对抗网络(CL-CaGAN),以提升跨域检测性能。首先,构建改进的胶囊结构与对抗学习网络,用于估计背景分布,弥补先验不足;为缓解灾难性遗忘,引入基于聚类的样本重放策略和额外自蒸馏正则化,实现历史与新知识融合,并保留从旧场景到新场景的判别能力。此外,引入可微增强机制,提升训练数据生成性能,稳定训练过程并强化背景重建能力。在多个真实高光谱图像数据集上的实验表明,所提方法在跨域场景下具备更高检测性能和持续学习能力,有效缓解了灾难性遗忘。

原文摘要 · Abstract (English)

Anomaly detection (AD) has attracted remarkable attention in hyperspectral image (HSI) processing fields, and most existing deep learning (DL)-based algorithms indicate dramatic potential for detecting anomaly samples through specific training process under current scenario. However, the limited prior information and the catastrophic forgetting problem indicate crucial challenges for existing DL structure in open scenarios cross-domain detection. In order to improve the detection performance, a novel continual learning-based capsule differential generative adversarial network (CL-CaGAN) is proposed to elevate the cross-scenario learning performance for facilitating the real application of DL-based structure in hyperspectral AD (HAD) task. First, a modified capsule structure with adversarial learning network is constructed to estimate the background distribution for surmounting the deficiency of prior information. To mitigate the catastrophic forgetting phenomenon, clustering-based sample replay strategy and a designed extra self-distillation regularization are integrated for merging the history and future knowledge in continual AD task, while the discriminative learning ability from previous detection scenario to current scenario is retained by the elaborately designed structure with continual learning (CL) strategy. In addition, the differentiable enhancement is enforced to augment the generation performance of the training data. This further stabilizes the training process with better convergence and efficiently consolidates the reconstruction ability of background samples. To verify the effectiveness of our proposed CL-CaGAN, we conduct experiments on several real HSIs, and the results indicate that the proposed CL-CaGAN demonstrates higher detection performance and continuous learning capacity for mitigating the catastrophic forgetting under cross-domain scenarios.

高光谱检测持续学习胶囊网络生成对抗

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。