文献类型：期刊文献

中文题名：基于MAML算法的YOLOv3目标检测模型

英文题名：YOLOv3 Object Detection Model Based on MAML Algorithm

作者：沈震宇[1];朱昌明[1];王喆[2]

机构：[1]上海海事大学信息工程学院,上海202010;[2]华东理工大学信息科学与工程学院,上海200237

年份：2022

卷号：48

期号：1

起止页码：112

中文期刊名：华东理工大学学报（自然科学版）

外文期刊名：Journal of East China University of Science and Technology

收录：Scopus;北大核心:【北大核心2020】；CSCD:【CSCD_E2021_2022】；

基金：晨光计划(18CG54);中国博士后科学基金(2019M651576);国家自然科学青年基金(61602296);上海市自然科学基金(16ZR1414500)。

语种：中文

中文关键词：计算机视觉;图像识别;特征提取;目标检测;小样本学习

外文关键词：computer vision;image recognition;feature extraction;object detection;few shot learning

摘要：作为典型的一体化卷积神经网络,YOLOv3模型的网路传输途径简单,检测速度相对较快,但检测精度较低。当遇到新的目标在训练数据集中存在的样本较少时,模型检测会更加不准确,甚至会出现检测不到的情况。本文基于与模型不相关的元学习算法(MAML)改进了YOLOv3主干网络的结构,使其具有内循环和外循环的梯度下降,在初始参数基础上进行多步的梯度调整,达到仅用小样本数据就能快速收敛的目的。实验结果表明,该方法使得YOLOv3模型的检测精度提升了5.24%,且可以使梯度下降保持稳定,有效地满足YOLOv3模型在小样本数据训练情况下识别目标位置的精准性和泛化性。
Object detection has been a research hotspot in the field of computer vision in recent years.Due to the extensive application of deep learning,the target detection technology combined with deep learning has been developing and making continuous breakthroughs.In the field of target detection,it is difficult to solve the problem of target detection with few sample categories,and detect small targets with high accuracy via the training with few sample categories.By means of the model-agnostic meta-learning(MAML)algorithm in meta learning,this paper improves the information transmission form of the backbone network in YOLOv3 to make Darknet-53 achieve two stages of parameter internal update and external update in gradient descent.By multi-step gradient adjustment on the initial parameters,the trained weights can focus more on the feature information of the target.Even only via a small number of sample categories,it can also maintain the sensitivity to the target in the new task.It is shown via the experiment results that the mean average precision(mAP)value of YOLOv3 model attains 74.81%,and the mAP value of YOLOv3 model based on MAML algorithm can reach 80.05%,which improves the accuracy by 5.24%.The network structure and training mechanism of the modified YOLOv3 via MAML can improve the accuracy of detection in training and prediction,and the trained weights can make the model have high detection accuracy and high generalization.