重庆工商大学学报(自然科学版)2025,Vol.42Issue(4):17-26,10.DOI:10.16055/j.issn.1672-058X.2025.0004.003
面向遥感图像场景分类的轻量级沙漏密集网络
Lightweight Hourglass Dense Network for Remote Sensing Image Scene Classification
摘要
Abstract
Objective In the task of remote sensing image scene classification with complex spatial structures and geographical layouts,although deep convolutional neural networks(CNNs)have better classification performance,they often have high complexity and are not suitable for mobile or embedded devices.Therefore,a new lightweight hourglass dense network(LHD-NET)was proposed to achieve a good balance between classification accuracy and model complexity.Methods Firstly,shallow information was extracted through a shallow mixed downsampling structure with a feature compensation mechanism.This structure can reduce the number of parameters in subsequent layers while ensuring sufficient information extraction,thus improving performance while keeping the model lightweight.Then,dense connections were used between hourglass structures to improve feature reuse and to some extent avoid gradient disappearance,which promoted information transfer.Finally,a convolutional layer feature with high semantic information was used to guide multi-layer feature aggregation,so as to improve the performance of the classifier.Meanwhile,during the training process,the cross-entropy loss function based on label smoothing was employed to smooth the true labels,which could effectively improve robustness and alleviate overfitting compared with ordinary cross-entropy loss functions.Results Experimental results showed that the model achieved significant classification performance with only 5.4 M parameters,obtaining average classification accuracies of 99.19%,97.75%,and 92.38%on three publicly available remote sensing datasets,namely UC Merced Land-Use,SIRI-WHU,and NWPU-RESISC45,respectively.Conclusion Experimental results demonstrate that the proposed model can achieve good classification performance with a small number of parameters.Compared with deep neural networks,the proposed model significantly reduces the number of model parameters while maintaining high classification accuracy,providing certain reference values for remote sensing image scene classification tasks and model lightweighting.关键词
遥感图像/场景分类/轻量级/卷积神经网络Key words
remote sensing image/scene classification/lightweight/convolutional neural network分类
信息技术与安全科学引用本文复制引用
刘向举,吴文彦,蒋社想..面向遥感图像场景分类的轻量级沙漏密集网络[J].重庆工商大学学报(自然科学版),2025,42(4):17-26,10.基金项目
安徽省重点实验室项目(ZKSYS202204). (ZKSYS202204)