期刊
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING
卷 59, 期 2, 页码 1424-1436出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TGRS.2020.3003341
关键词
Task analysis; Generative adversarial networks; Generators; Gallium nitride; Training; Deep learning; Image reconstruction; Convolutional neural network (CNN); generative adversarial networks (GANs); hyperspectral image (HSI) classification; multitask learning
类别
资金
- Natural Science Foundation of China [61825601, 61532009, 61906096]
- Natural Science Foundation of Jiangsu Province, China [BK20180786, 18KJB520032]
In this article, a multitask generative adversarial network (MTGAN) is proposed to address the issue of deep learning models heavily depending on the quantity of available training samples in hyperspectral image classification. By utilizing rich information from unlabeled samples and employing an adversarial learning method, the MTGAN model is able to indirectly improve the discrimination and generalization ability of the classification task. Additionally, skip-layer connections are used to fully explore useful information from shallow layers, resulting in higher performance compared to other state-of-the-art deep learning models.
Deep learning has shown its huge potential in the field of hyperspectral image (HSI) classification. However, most of the deep learning models heavily depend on the quantity of available training samples. In this article, we propose a multitask generative adversarial network (MTGAN) to alleviate this issue by taking advantage of the rich information from unlabeled samples. Specifically, we design a generator network to simultaneously undertake two tasks: the reconstruction task and the classification task. The former task aims at reconstructing an input hyperspectral cube, including the labeled and unlabeled ones, whereas the latter task attempts to recognize the category of the cube. Meanwhile, we construct a discriminator network to discriminate the input sample coming from the real distribution or the reconstructed one. Through an adversarial learning method, the generator network will produce real-like cubes, thus indirectly improving the discrimination and generalization ability of the classification task. More importantly, in order to fully explore the useful information from shallow layers, we adopt skip-layer connections in both reconstruction and classification tasks. The proposed MTGAN model is implemented on three standard HSIs, and the experimental results show that it is able to achieve higher performance than other state-of-the-art deep learning models.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据