RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks
This study aims at improving the efficiency of remote sensing scene classification (RSSC) through lightweight neural networks and to provide a possibility for large-scale, intelligent and real-time computation in performing RSSC for common devices. In this study, a lightweight RSSC model is proposed...
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2022-11-01
|
| Series: | Electronics |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2079-9292/11/22/3727 |
| _version_ | 1797465499118862336 |
|---|---|
| author | Zhichao Chen Jie Yang Zhicheng Feng Lifang Chen |
| author_facet | Zhichao Chen Jie Yang Zhicheng Feng Lifang Chen |
| author_sort | Zhichao Chen |
| collection | DOAJ |
| description | This study aims at improving the efficiency of remote sensing scene classification (RSSC) through lightweight neural networks and to provide a possibility for large-scale, intelligent and real-time computation in performing RSSC for common devices. In this study, a lightweight RSSC model is proposed, which is named RSCNet. First, we use the lightweight ShuffleNet v2 network to extract the abstract features from the images, which can guarantee the efficiency of the model. Then, the weights of the backbone are initialized using transfer learning, allowing the model to learn by drawing on the knowledge of ImageNet. Second, to further improve the classification accuracy of the model, we propose to combine ShuffleNet v2 with an efficient channel attention mechanism that allows the features of the input classifier to be weighted. Third, we use a regularization technique during the training process, which utilizes label smoothing regularization to replace the original loss function. The experimental results show that the classification accuracy of RSCNet is 96.75% and 99.05% on the AID and UCMerced_LandUse datasets, respectively. The floating-point operations (FLOPs) of the proposed model are only 153.71 M, and the time spent for a single inference on the CPU is about 2.75 ms. Compared with existing RSSC methods, RSCNet achieves relatively high accuracy at a very small computational cost. |
| first_indexed | 2024-03-09T18:22:22Z |
| format | Article |
| id | doaj.art-4d6805476d8e4cbda2b8af187675091e |
| institution | Directory Open Access Journal |
| issn | 2079-9292 |
| language | English |
| last_indexed | 2024-03-09T18:22:22Z |
| publishDate | 2022-11-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Electronics |
| spelling | doaj.art-4d6805476d8e4cbda2b8af187675091e2023-11-24T08:09:32ZengMDPI AGElectronics2079-92922022-11-011122372710.3390/electronics11223727RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural NetworksZhichao Chen0Jie Yang1Zhicheng Feng2Lifang Chen3Department of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou 341000, ChinaDepartment of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou 341000, ChinaDepartment of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou 341000, ChinaDepartment of Science, Jiangxi University of Science and Technology, Ganzhou 341000, ChinaThis study aims at improving the efficiency of remote sensing scene classification (RSSC) through lightweight neural networks and to provide a possibility for large-scale, intelligent and real-time computation in performing RSSC for common devices. In this study, a lightweight RSSC model is proposed, which is named RSCNet. First, we use the lightweight ShuffleNet v2 network to extract the abstract features from the images, which can guarantee the efficiency of the model. Then, the weights of the backbone are initialized using transfer learning, allowing the model to learn by drawing on the knowledge of ImageNet. Second, to further improve the classification accuracy of the model, we propose to combine ShuffleNet v2 with an efficient channel attention mechanism that allows the features of the input classifier to be weighted. Third, we use a regularization technique during the training process, which utilizes label smoothing regularization to replace the original loss function. The experimental results show that the classification accuracy of RSCNet is 96.75% and 99.05% on the AID and UCMerced_LandUse datasets, respectively. The floating-point operations (FLOPs) of the proposed model are only 153.71 M, and the time spent for a single inference on the CPU is about 2.75 ms. Compared with existing RSSC methods, RSCNet achieves relatively high accuracy at a very small computational cost.https://www.mdpi.com/2079-9292/11/22/3727remote sensingclassificationlightweight neural networksattention mechanismartificial intelligencetransfer learning |
| spellingShingle | Zhichao Chen Jie Yang Zhicheng Feng Lifang Chen RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks Electronics remote sensing classification lightweight neural networks attention mechanism artificial intelligence transfer learning |
| title | RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks |
| title_full | RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks |
| title_fullStr | RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks |
| title_full_unstemmed | RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks |
| title_short | RSCNet: An Efficient Remote Sensing Scene Classification Model Based on Lightweight Convolution Neural Networks |
| title_sort | rscnet an efficient remote sensing scene classification model based on lightweight convolution neural networks |
| topic | remote sensing classification lightweight neural networks attention mechanism artificial intelligence transfer learning |
| url | https://www.mdpi.com/2079-9292/11/22/3727 |
| work_keys_str_mv | AT zhichaochen rscnetanefficientremotesensingsceneclassificationmodelbasedonlightweightconvolutionneuralnetworks AT jieyang rscnetanefficientremotesensingsceneclassificationmodelbasedonlightweightconvolutionneuralnetworks AT zhichengfeng rscnetanefficientremotesensingsceneclassificationmodelbasedonlightweightconvolutionneuralnetworks AT lifangchen rscnetanefficientremotesensingsceneclassificationmodelbasedonlightweightconvolutionneuralnetworks |