arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.06566cs.CVcs.AI

面向回归的知识蒸馏用于光学遥感图像轻量化船舶朝向角预测

Regression-Oriented Knowledge Distillation for Lightweight Ship Orientation Angle Prediction with Optical Remote Sensing Images

Nanjing Research Institute of Electronic Engineering(南京电子工程研究所);Nanjing University of Information Science & Technology(南京信息工程大学)

Zhan Shi, Xin Ding, Peng Ding, Chun Yang, Ru Huang, Xiaoxuan Song

AI总结:

本文提出Mobile-SOAP模型与SOAP-KD知识蒸馏框架,在FGSC-23上以很小精度损失显著降低船舶朝向角预测模型的参数量与计算量。

摘要:

利用光学遥感图像进行船舶朝向角预测(SOAP)是一项重要的图像处理任务,通常依赖深度卷积神经网络(CNN)来实现准确预测。本文提出一种新颖框架,旨在在不损害预测精度的情况下减小SOAP模型的模型规模和计算成本。首先,基于MobileNetV2设计了一个名为Mobile-SOAP的新型SOAP模型,取得了当前最优的预测精度。通过分别用四种小规模网络替换Mobile-SOAP中的卷积块,还构建了四个微型SOAP模型。随后,为将知识从Mobile-SOAP迁移到四个轻量化模型,提出了一种名为SOAP-KD的新型知识蒸馏(KD)框架,该框架包含新颖的基于特征的引导损失以及经过优化的、基于合成样本的知识迁移机制。最后,在FGSC-23数据集上开展的大量实验证实了Mobile-SOAP相较现有模型的优越性,并证明了SOAP-KD在提升四个专门设计的微型模型预测性能方面的有效性。值得注意的是,通过使用SOAP-KD,基于ShuffleNetV2x1.0的模型测试平均绝对误差仅比Mobile-SOAP高8%,但其参数量和乘累加运算(MACs)分别减少了61.6%和60.8%。

英文摘要:

Ship orientation angle prediction (SOAP) with optical remote sensing images is an important image processing task, which often relies on deep convolutional neural networks (CNNs) to make accurate predictions. This paper proposes a novel framework to reduce the model sizes and computational costs of SOAP models without harming prediction accuracy. First, a new SOAP model called Mobile-SOAP is designed based on MobileNetV2, achieving state-of-the-art prediction accuracy. Four tiny SOAP models are also created by replacing the convolutional blocks in Mobile-SOAP with four small-scale networks, respectively. Then, to transfer knowledge from Mobile-SOAP to four lightweight models, we propose a novel knowledge distillation (KD) framework termed SOAP-KD consisting of a novel feature-based guidance loss and an optimized synthetic samples-based knowledge transfer mechanism. Lastly, extensive experiments on the FGSC-23 dataset confirm the superiority of Mobile-SOAP over existing models and also demonstrate the effectiveness of SOAP-KD in improving the prediction performance of four specially designed tiny models. Notably, by using SOAP-KD, the test mean absolute error of the ShuffleNetV2x1.0-based model is only 8% higher than that of Mobile-SOAP, but its number of parameters and multiply-accumulate operations (MACs) are respectively 61.6% and 60.8% less.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑