Remote Sensing Imagery Object Detection Model Compression via Tucker Decomposition

Lang Huyan, Ying Li (), Dongmei Jiang, Yanning Zhang, Quan Zhou, Bo Li, Jiayuan Wei, Juanni Liu, Yi Zhang, Peng Wang and Hai Fang
Additional contact information
Lang Huyan: School of Computer Science, National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, Shaanxi Provincial Key Laboratory of Speech & Image Information Processing, Northwestern Polytechnical University, Xi’an 710129, China
Ying Li: School of Computer Science, National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, Shaanxi Provincial Key Laboratory of Speech & Image Information Processing, Northwestern Polytechnical University, Xi’an 710129, China
Dongmei Jiang: School of Computer Science, National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, Shaanxi Provincial Key Laboratory of Speech & Image Information Processing, Northwestern Polytechnical University, Xi’an 710129, China
Yanning Zhang: School of Computer Science, National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, Shaanxi Provincial Key Laboratory of Speech & Image Information Processing, Northwestern Polytechnical University, Xi’an 710129, China
Quan Zhou: School of Computer Science, National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, Shaanxi Provincial Key Laboratory of Speech & Image Information Processing, Northwestern Polytechnical University, Xi’an 710129, China
Bo Li: Key Laboratory of Science and Technology on Space Microwave, CAST Xi’an, Xi’an 710100, China
Jiayuan Wei: Key Laboratory of Science and Technology on Space Microwave, CAST Xi’an, Xi’an 710100, China
Juanni Liu: Key Laboratory of Science and Technology on Space Microwave, CAST Xi’an, Xi’an 710100, China
Yi Zhang: Key Laboratory of Science and Technology on Space Microwave, CAST Xi’an, Xi’an 710100, China
Peng Wang: Key Laboratory of Science and Technology on Space Microwave, CAST Xi’an, Xi’an 710100, China
Hai Fang: Key Laboratory of Science and Technology on Space Microwave, CAST Xi’an, Xi’an 710100, China

Mathematics, 2023, vol. 11, issue 4, 1-26

Abstract: Although convolutional neural networks (CNNs) have made significant progress, their deployment onboard is still challenging because of their complexity and high processing cost. Tensors provide a natural and compact representation of CNN weights via suitable low-rank approximations. A novel decomposed module called DecomResnet based on Tucker decomposition was proposed to deploy a CNN object detection model on a satellite. We proposed a remote sensing image object detection model compression framework based on low-rank decomposition which consisted of four steps, namely (1) model initialization, (2) initial training, (3) decomposition of the trained model and reconstruction of the decomposed model, and (4) fine-tuning. To validate the performance of the decomposed model in our real mission, we constructed a dataset containing only two classes of objects based on the DOTA and HRSC2016. The proposed method was comprehensively evaluated on the NWPU VHR-10 dataset and the CAST-RS2 dataset created in this work. The experimental results demonstrated that the proposed method, which was based on Resnet-50, could achieve up to 4.44 times the compression ratio and 5.71 times the speedup ratio with merely a 1.9% decrease in the mAP (mean average precision) of the CAST-RS2 dataset and a 5.3% decrease the mAP of the NWPU VHR-10 dataset.

Keywords: Tucker decomposition; model compression; onboard object detection; remote sensing imagery; tensor decomposition; rank selection (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/11/4/856/pdf (application/pdf)
https://www.mdpi.com/2227-7390/11/4/856/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:11:y:2023:i:4:p:856-:d:1060949

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().