Optimizing Lightweight Convolutional Neural Networks for Hyperspectral Image Fusion

Authors

  • Weiheng Hong

DOI:

https://doi.org/10.61173/22876j03

Keywords:

hyperspectral image fusion, embedded devices, 8bit quantization, channel pruning

Abstract

This paper addresses the challenges associated with processing complex scenes and high-precision images on resource-constrained devices, particularly within the domain of hyperspectral image processing and super-resolution reconstruction techniques. We present an optimized model for existing hyperspectral image fusion models, leveraging network lightweight and channel pruning. Our proposed model, NestFuse Small, employs NestFuse as the primary network architecture and integrates a quantization pruning module. Experimental findings demonstrate that, in comparison to the original NestFuse model, NestFuse Small’s computation speed is 164.8% of the pre-optimization speed, a decrease in memory usage of 20.65%, resulting in a slight decrease in performance. This optimized model facilitates more efficient image processing on resource-constrained devices.

References

[1] Yang Kai. Research on Hyperspectral Image Classification Based on Attention Networks [D]. University of Chinese Academy of Sciences (Xi’an Institute of Optics and Precision Mechanics, Chinese Academy of Sciences), 2023. DOI: 10.27605/d.cnki.gkxgs.2023.000013.

[2] Rao Weiqiang. Research on Deep Learning Target Detection Method for Hyperspectral Image [D]. University of Chinese Academy of Sciences (Aerospace Information Research Institute, Chinese Academy of Sciences), 2022. DOI: 10.44231/ d.cnki.gktxc.2022.000074.

[3] L. Jian, X. Yang, Z. Liu, G. Jeon, M. Gao, and D. Chisholm, “SEDRFuse: A Symmetric Encoder–Decoder With Residual Block Network for Infrared and Visible Image Fusion,” in IEEE Transactions on Instrumentation and Measurement, vol. 70, pp. 1-15, 2021, Art no. 5002215, doi: 10.1109/TIM.2020.3022438.

[4] Xu H, Liang P, Yu W, et al. Learning a Generative Model for Fusing Infrared and Visible Images via Conditional Generative Adversarial Network with Dual Discriminators[C]//IJCAI. 2019: 3954-3960.

[5] Lin T Y, Maire M, Belongie S, et al. Microsoft coco: Common objects in context[C]//Computer Vision–ECCV 2014: 13th European Conference, Zurich, Switzerland, September 6-12, 2014, Proceedings, Part V 13. Springer International Publishing, 2014: 740-755.

[6] Toet A. The TNO multiband image data collection[J]. Data in Dean&Francis brief, 2017, 15: 249-251.

[7] Li H, Wu X J, Durrani T. NestFuse: An infrared and visible image fusion architecture based on nest connection and spatial/channel attention models[J]. IEEE Transactions on Instrumentation and Measurement, 2020, 69(12): 9645-9656

Downloads

Published

2024-04-16