Multi-branch fusion network for hyperspectral image classification

被引:38
|
作者
Gao, Hongmin [1 ]
Yang, Yao [1 ]
Lei, Sheng [2 ]
Li, Chenming [1 ]
Zhou, Hui [1 ]
Qu, Xiaoyu [1 ]
机构
[1] Hohai Univ, Coll Comp & Informat, Nanjing 211100, Jiangsu, Peoples R China
[2] Jiangxi Prov Inst Water Sci, Nanchang 330029, Jiangxi, Peoples R China
基金
中国国家自然科学基金; 美国国家科学基金会; 中国博士后科学基金;
关键词
Hyperspectral remote sensing image classification; Convolutional neural network; Multi-branch fusion network; Small sample size problem; Class imbalance; SPECTRAL-SPATIAL CLASSIFICATION; SVM;
D O I
10.1016/j.knosys.2019.01.020
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Hyperspectral remote sensing image (HSI) has the characteristics of large data volume and high spectral resolution. It contains abundant spectral information and has tremendous applicable value. Convolutional neural network (CNN) has been successfully applied to HSI classification. However, the limited labeled samples of the HSI make the existing CNN based HSI classification methods generally be plagued by small sample size problem and class imbalance, which cause great challenges for HSI classification. This work proposes a novel CNN architecture for HSI classification. The proposed CNN is a multi-branch fusion network, which is formed by merging multiple branches on an ordinary CNN. It can effectively extract features of HSIs. In addition, the 1 x 1 convolutional layer is introduced into the branches to reduce the number of parameters and then improve the classification efficiency. Furthermore, the L2 regularization is introduced into this work to improve the generalization performance of the proposed model under small sample set. Experimental results on three benchmark hyperspectral images demonstrate that the proposed CNN can provide excellent classification performance under small training set. (C) 2019 Elsevier B.V. All rights reserved.
引用
收藏
页码:11 / 25
页数:15
相关论文
共 50 条
  • [41] A lightweight multi-branch network for low-light image enhancement
    Yu, Youjiang
    Yuan, Cheng
    Zhang, Kaibing
    Wang, Xiaohua
    [J]. ELECTRONICS LETTERS, 2023, 59 (09)
  • [42] A Modularized Architecture of Multi-Branch Convolutional Neural Network for Image Captioning
    He, Shan
    Lu, Yuanyao
    [J]. ELECTRONICS, 2019, 8 (12)
  • [43] Double-branch feature fusion transformer for hyperspectral image classification
    Lanxue Dang
    Libo Weng
    Yane Hou
    Xianyu Zuo
    Yang Liu
    [J]. Scientific Reports, 13
  • [44] Double-branch feature fusion transformer for hyperspectral image classification
    Dang, Lanxue
    Weng, Libo
    Hou, Yane
    Zuo, Xianyu
    Liu, Yang
    [J]. SCIENTIFIC REPORTS, 2023, 13 (01)
  • [45] A Lightweight Multi-Branch Context Network for Unsupervised Underwater Image Restoration
    Wang, Rong
    Zhang, Yonghui
    Zhang, Yulu
    [J]. WATER, 2024, 16 (05)
  • [46] Channel rearrangement multi-branch network for image super-resolution
    Wei, Deyun
    Wang, Zhaowu
    [J]. DIGITAL SIGNAL PROCESSING, 2022, 120
  • [47] Multi-Branch and Progressive Network for Low-Light Image Enhancement
    Zhang, Kaibing
    Yuan, Cheng
    Li, Jie
    Gao, Xinbo
    Li, Minqi
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2023, 32 : 2295 - 2308
  • [48] One-Dimensional Convolutional Multi-branch Fusion Network for EEG-Based Motor Imagery Classification
    Liu, Xiaoguang
    Zhang, Mingjin
    Xiong, Shicheng
    Wang, Xiaodong
    Liang, Tie
    Li, Jun
    Xiong, Peng
    Wang, Hongrui
    Liu, Xiuling
    [J]. IRBM, 2023, 44 (06)
  • [49] Multi-hop double-branch network for small sample hyperspectral image classification
    Shuang, Murong
    Wang, Yulong
    Zhang, Yanjun
    [J]. JOURNAL OF SPATIAL SCIENCE, 2024,
  • [50] Two-branch global spatial-spectral fusion transformer network for hyperspectral image classification
    Xie, Erxin
    Chen, Na
    Zhang, Genwei
    Peng, Jiangtao
    Sun, Weiwei
    [J]. PHOTOGRAMMETRIC RECORD, 2024, 39 (186): : 392 - 411