Domain-Adversarial Generative and Dual-Feature Representation Discriminative Network for Hyperspectral Image Domain Generalization
Minghui Chu, Xiaodong Yu, Hongbin Dong, Shuying Zang
Abstract
Traditional training models often experience significant performance drops on test sets when the training and testing data distributions differ. To address this domain shift problem, we propose D3Net, a domain generalization (DG) classification network for hyperspectral images (HSIs) based on generative adversarial networks (GANs). Specifically, D3Net consists mainly of a generator and a discriminator. The generator extracts domain-invariant information from the source domain to generate data with core classification features, while the discriminator employs a dual-confidence model to enhance the capture of domain-invariant features. Through adversarial iterations, the model is able to adapt to the domain shift effects of unknown data. Unlike existing DG methods that rely on random perturbations for data augmentation, D3Net utilizes learnable convolutional neural networks (CNNs) rather than randomization to enhance the model’s learning capability. We conducted cross-scene classification experiments on datasets from Houston, Pavia, and Indiana, and the results demonstrate the effectiveness of our approach. The code for D3Net is available at: <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/gmsjzyq123/D3Net</uri>.