Generative Adversarial Networks and Conditional Random Fields for Hyperspectral Image Classification

被引:116
|
作者
Zhong, Zilong [1 ]
Li, Jonathan [2 ,3 ]
Clausi, David A. [1 ]
Wong, Alexander [1 ]
机构
[1] Univ Waterloo, Dept Syst Design Engn, Waterloo, ON N2L 3G1, Canada
[2] Univ Waterloo, Dept Geog & Environm Management, Waterloo, ON N2L 3G1, Canada
[3] Xiamen Univ, Fujian Key Lab Sensing & Comp Smart City, Xiamen 361005, Peoples R China
基金
加拿大自然科学与工程研究理事会;
关键词
Gallium nitride; Deep learning; Training; Generators; Generative adversarial networks; Data models; Hyperspectral imaging; Conditional random fields (CRFs); generative adversarial networks (GANs); hyperspectral image (HSI) classification; semisupervised deep learning; REPRESENTATION; FRAMEWORK;
D O I
10.1109/TCYB.2019.2915094
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this paper, we address the hyperspectral image (HSI) classification task with a generative adversarial network and conditional random field (GAN-CRF)-based framework, which integrates a semisupervised deep learning and a probabilistic graphical model, and make three contributions. First, we design four types of convolutional and transposed convolutional layers that consider the characteristics of HSIs to help with extracting discriminative features from limited numbers of labeled HSI samples. Second, we construct semisupervised generative adversarial networks (GANs) to alleviate the shortage of training samples by adding labels to them and implicitly reconstructing real HSI data distribution through adversarial training. Third, we build dense conditional random fields (CRFs) on top of the random variables that are initialized to the softmax predictions of the trained GANs and are conditioned on HSIs to refine classification maps. This semisupervised framework leverages the merits of discriminative and generative models through a game-theoretical approach. Moreover, even though we used very small numbers of labeled training HSI samples from the two most challenging and extensively studied datasets, the experimental results demonstrated that spectral-spatial GAN-CRF (SS-GAN-CRF) models achieved top-ranking accuracy for semisupervised HSI classification.
引用
收藏
页码:3318 / 3329
页数:12
相关论文
共 50 条
  • [41] Generative adversarial networks for hyperspectral image spatial super resolution
    Yilin, Jiang
    Ran, Shao
    Sanqiang, Tang
    [J]. Journal of China Universities of Posts and Telecommunications, 2020, 27 (04): : 8 - 16
  • [42] Generative Adversarial Networks-Based Semi-Supervised Learning for Hyperspectral Image Classification
    He, Zhi
    Liu, Han
    Wang, Yiwen
    Hu, Jie
    [J]. REMOTE SENSING, 2017, 9 (10)
  • [43] SAR image synthesis based on conditional generative adversarial networks
    Wang, Jianyu
    Li, Jingwen
    Sun, Bing
    Zuo, Zhixiong
    [J]. JOURNAL OF ENGINEERING-JOE, 2019, 2019 (21): : 8093 - 8097
  • [44] Transferring Microscopy Image Modalities with Conditional Generative Adversarial Networks
    Han, Liang
    Yin, Zhaozheng
    [J]. 2017 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2017, : 851 - 859
  • [45] Hyperspectral Image Classification Based on Visible-Infrared Sensors and Residual Generative Adversarial Networks
    Su, Hui-Wei
    Tan, Ri-hui
    Chen, Chih-Cheng
    Hu, Zhongzheng
    Shankaranarayanan, Avinash
    [J]. SENSORS AND MATERIALS, 2021, 33 (11) : 4045 - 4056
  • [46] A Survey of Image Translation Based on Conditional Generative Adversarial Networks
    Tu, Hangyao
    Wang, Wanliang
    Chen, Jiacheng
    Li, Guoqing
    Wu, Fei
    [J]. Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics, 2024, 36 (01): : 14 - 32
  • [47] Image Classification Based on Conditional Random Fields
    Yang, Yan
    Wen-bo, Huang
    Yun-ji, Wang
    [J]. MECHATRONICS ENGINEERING, COMPUTING AND INFORMATION TECHNOLOGY, 2014, 556-562 : 4901 - +
  • [48] Self-Attention-Based Conditional Variational Auto-Encoder Generative Adversarial Networks for Hyperspectral Classification
    Chen, Zhitao
    Tong, Lei
    Qian, Bin
    Yu, Jing
    Xiao, Chuangbai
    [J]. REMOTE SENSING, 2021, 13 (16)
  • [49] Image generation and classification via generative adversarial networks
    Mirabedini, Shirin
    Dastgerdi, Shadi Hejareh
    Kangavari, Mohammadreza
    AhmadiPanah, Mandi
    [J]. BIOSCIENCE RESEARCH, 2020, 17 (02): : 1356 - 1363
  • [50] Dual hybrid convolutional generative adversarial network for hyperspectral image classification
    Shi, Cuiping
    Zhang, Tianyu
    Liao, Diling
    Jin, Zhan
    Wang, Liguo
    [J]. INTERNATIONAL JOURNAL OF REMOTE SENSING, 2022, 43 (14) : 5452 - 5479