A Network Pruning Method for Remote Sensing Image Scene Classification

Baogui Qi, He Chen, Yin Zhuang, Shaorong Liu, Liang Chen

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

2 Citations (Scopus)

Abstract

Deep convolutional neural networks have been widely used to improve remote sensing image scene classification performance. However, most of these networks include many parameters and need many computational resources. Which hinders the applications of these networks when facing the satellite, plane or other mobile platforms. In this paper, we introduce a network pruning framework which can reduce the size of the network model and maintain the classification accuracy. In this framework, we train the pruned model using both the original unpruned model's output and training dataset. Which can learn more information than retrain using dataset only. In our experiments, we evaluate our method for remote sensing scene classification on NWPU-RESISC45 dataset. The results demonstrate that our method was effective and maintained the model classification accuracy.

Original languageEnglish
Title of host publicationICSIDP 2019 - IEEE International Conference on Signal, Information and Data Processing 2019
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728123455
DOIs
Publication statusPublished - Dec 2019
Event2019 IEEE International Conference on Signal, Information and Data Processing, ICSIDP 2019 - Chongqing, China
Duration: 11 Dec 201913 Dec 2019

Publication series

NameICSIDP 2019 - IEEE International Conference on Signal, Information and Data Processing 2019

Conference

Conference2019 IEEE International Conference on Signal, Information and Data Processing, ICSIDP 2019
Country/TerritoryChina
CityChongqing
Period11/12/1913/12/19

Keywords

  • compression
  • convolutional neural networks
  • deep learning
  • pruning
  • scene classification

Fingerprint

Dive into the research topics of 'A Network Pruning Method for Remote Sensing Image Scene Classification'. Together they form a unique fingerprint.

Cite this

Qi, B., Chen, H., Zhuang, Y., Liu, S., & Chen, L. (2019). A Network Pruning Method for Remote Sensing Image Scene Classification. In ICSIDP 2019 - IEEE International Conference on Signal, Information and Data Processing 2019 Article 9173116 (ICSIDP 2019 - IEEE International Conference on Signal, Information and Data Processing 2019). Institute of Electrical and Electronics Engineers Inc.. https://doi.org/10.1109/ICSIDP47821.2019.9173116