Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. Geometry-Aware Deep Recurrent Neural Networks for Hyperspectral Image Classification
 
research article

Geometry-Aware Deep Recurrent Neural Networks for Hyperspectral Image Classification

Hao, Siyuan
•
Wang, Wei
•
Salzmann, Mathieu  
March 1, 2021
IEEE Transactions On Geoscience And Remote Sensing

Variants of deep networks have been widely used for hyperspectral image (HSI)-classification tasks. Among them, in recent years, recurrent neural networks (RNNs) have attracted considerable attention in the remote sensing community. However, complex geometries cannot be learned easily by the traditional recurrent units [e.g., long short-term memory (LSTM) and gated recurrent unit (GRU)]. In this article, we propose a geometry-aware deep recurrent neural network (Geo-DRNN) for HSI classification. We build this network upon two modules: a U-shaped network (U-Net) and RNNs. We first input the original HSI patches to the U-Net, which can be trained with very few images and obtain a preliminary classification result. We then add RNNs on the top of the U-Net so as to mimic the human brain to refine continuously the output-classification map. However, instead of using the traditional dot product in each gate of the RNNs, we introduce a Net-Gated GRU that increases the nonlinear representation power. Finally, we use a pretrained ResNet as a regularizer to improve further the ability of the proposed network to describe complex geometries. To this end, we construct a geometry-aware ResNet loss, which leverages the pretrained ResNet's knowledge about the different structures in the real world. Our experimental results on real HSIs and road topology images demonstrate that our approach outperforms the state-of-the-art classification methods and can learn complex geometries.

  • Details
  • Metrics
Type
research article
DOI
10.1109/TGRS.2020.3005623
Web of Science ID

WOS:000622319000044

Author(s)
Hao, Siyuan
Wang, Wei
Salzmann, Mathieu  
Date Issued

2021-03-01

Published in
IEEE Transactions On Geoscience And Remote Sensing
Volume

59

Issue

3

Start page

2448

End page

2460

Subjects

Geochemistry & Geophysics

•

Engineering, Electrical & Electronic

•

Remote Sensing

•

Imaging Science & Photographic Technology

•

Engineering

•

deep learning

•

geometry-aware loss

•

gated recurrent unit (gru)

•

hyperspectral image (hsi) classification

•

net-gated recurrent neural networks (rnns)

•

remote sensing

•

u-shaped network (u-net)

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
CVLAB  
Available on Infoscience
March 26, 2021
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/176434
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés