一般化
高光谱成像
计算机科学
领域(数学分析)
人工智能
领域知识
数学
数学分析
作者
Haoyu Wang,Xiaomin Liu,Zhenzhuang Qiao,Hanqing Tao
标识
DOI:10.1109/tgrs.2024.3494796
摘要
Cross-domain hyperspectral image (HSI) classification can improve the model’s classification performance in the target domain by utilizing the rich knowledge from the source domain. However, existing cross-domain HSI classification methods mostly belong to transductive learning, which is difficult to apply to domain generalization tasks where the target domain is unseen during model learning. Inspired by human causal reasoning and knowledge induction mechanisms, this article develops an inductive learning-based framework for hyperspectral domain generalization: Causal meta-generalization. By simulating domain generalization scenarios, the framework helps the model induct domain-invariant causal meta-knowledge, thereby ensuring its strong generalization ability to unseen target domains. Specifically, we first propose a bottleneck variational auto-encoder (B-VAE) based on a forward–reverse information bottleneck, decoupling the domain distribution and class distribution of HSIs. By perturbing the domain distribution to generate virtual domains, we simulate potential domain distribution changes in the real world, providing a data basis for the induction of causal meta-knowledge. Second, in the process of simulating domain generalization scenarios, we establish a dual-layer optimization mechanism (DLOM) based on invariant-generalization risk minimization. In the inner layer optimization, by minimizing the model’s invariant causal effect loss (ICEL) in the virtual source domains, we guide the model to learn domain-invariant causal meta-knowledge. In the outer optimization, by minimizing the model’s generalization risk in the unseen virtual target domain, we enhance the applicability of causal meta-knowledge in domain generalization tasks. This proposed method has potential applications in remote sensing signal-processing tasks, such as the recognition of crop pests and diseases and the identification of minerals. The code can be accessed at https://github.com/wzr78998/CMG.
科研通智能强力驱动
Strongly Powered by AbleSci AI