一种用于遥感图像分类的轻量级深度学习模型
计算机视觉与模式识别
2023-02-28 v1 人工智能
机器学习
摘要
本文提出了一种高性能轻量级深度学习模型用于遥感图像分类（RSIC），即识别遥感图像的航拍场景任务。为此，我们首先评估了多种基准卷积神经网络（CNN）架构：MobileNet V1/V2、ResNet 50/151V2、InceptionV3/InceptionResNetV2、EfficientNet B0/B7、DenseNet 121/201、ConNeXt Tiny/Large。然后，选取性能最佳的模型，以师生结构训练一个紧凑模型。来自教师的知识蒸馏旨在以显著降低的复杂度实现高性能。通过在 NWPU-RESISC45 基准上进行大量实验，我们提出的师生模型优于最先进系统，并有潜力应用于广泛的边缘设备。
引用
@article{arxiv.2302.13028,
title = {A Light-weight Deep Learning Model for Remote Sensing Image Classification},
author = {Lam Pham and Cam Le and Dat Ngo and Anh Nguyen and Jasmin Lampert and Alexander Schindler and Ian McLoughlin},
journal= {arXiv preprint arXiv:2302.13028},
year = {2023}
}