过度拟合
计算机科学
目标检测
人工智能
一般化
对象(语法)
机器学习
蒸馏
弹丸
模式识别(心理学)
计算机视觉
人工神经网络
数学
数学分析
有机化学
化学
作者
Yang Li,Yicheng Gong,Zhuo Zhang
标识
DOI:10.1109/mis.2022.3205686
摘要
In many fields, due to the lack of large-scale training data, the traditional object detection methods cannot complete the actual work well. The main reason is the overfitting problem and lack of the generalization ability. In this work, we propose a general method to alleviate the overfitting problem in the few-shot object detection. Our work extends Faster R-CNN with self-knowledge distillation algorithm and designs the loss function with attention mechanism, which can improve true detection in the foreground. In this way, object detector can learn an approximate mapping relationship from few samples, which makes the network possess a stronger generalization ability when tackling few images. Through numerous comparative experiments, we demonstrate that our method is general and feasible on VOC and COCO benchmarks datasets with different settings. We provide a new idea for solving the problem of few-shot object detection, and produce an excellent performance of recall rate on few-shot object detection.
科研通智能强力驱动
Strongly Powered by AbleSci AI