A Hybrid Multi-Task Learning Network for Hyperspectral Image Classification with Few Labels
Hao Liu,Mingyang Zhang,Ziqi Di,Maoguo Gong,Tianqi Gao,A. K. Qin
DOI: https://doi.org/10.1109/tgrs.2024.3356008
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Recently, the field of hyperspectral image (HSI) classification has witnessed advancements with the emergence of deep learning models. Promising approaches, such as self-supervised strategies and domain adaptation, have effectively tackled the overfitting challenges posed by limited labeled samples in HSI classification. To extract comprehensive semantic information from different types of auxiliary tasks, which view the problem from multiple perspectives, and efficiently integrate multiple tasks into a single network, this paper proposes a hybrid multi-task learning framework (HyMuT) by sharing representations across multiple tasks. Based on the similarity between the data and target classification task, we construct three auxiliary tasks that are similar, related and weakly correlated to the target task, while three corresponding multi-task learning methods are integrated. The framework establishes a backbone network with a hard parameter sharing mechanism, which handles the main task and a similar spatial mask classification task. Subsequently, a hierarchical transfer multi-task learning approach is introduced to transfer the knowledge of a spatial-spectral joint mask reconstruction task from the autoencoder to the backbone network. Furthermore, a new source domain HSI dataset is introduced as an auxiliary task weakly correlated. To solve the source domain classification task and assist the hard parameter sharing mechanism, a dual adversarial classifier based on adversarial learning is employed. This classifier effectively extracts domain and task invariance. Extensive experiments are conducted on four benchmark HSI datasets to evaluate the performance. The results demonstrate that HyMuT outperforms state-of-the-art methods. This code will be available from the website: https://github.com/HaoLiu-XDU/HyMuT.