arrow
返回

Dynamic multi-scale loss optimization for object detection

delete2022-06-23
delete1
PRE
AI
Y
Yihao Luo
X
Xiang Cao
J
Juntao Zhang
P
Peng Cheng
王
王天江 (Tianjiang Wang)
Q
Qi Feng *
DOI:10.1007/s11042-022-13164-9delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
With the continuous improvement of deep object detectors via advanced model architectures, imbalance problems in the training process have received more attention. It is a common paradigm in object detection frameworks to perform multi-scale detection. However, each scale is treated equally during training. In this paper, we carefully study the objective imbalance of multi-scale detector training. We argue that the loss in each scale level is neither equally important nor independent. Different from the existing solutions of setting multi-task weights, we dynamically optimize the loss weight of each scale level in the training process. Specifically, we propose an Adaptive Variance Weighting (AVW) to balance multi-scale loss according to the statistical variance. Then we develop a novel Reinforcement Learning Optimization (RLO) to decide the weighting scheme probabilistically during training. It makes better utilization of multi-scale training loss without extra computational complexity and learnable parameters for backpropagation. Without bells and whistles, the proposed method improves ATSS by 0.9 AP on the MS COCO benchmark. And it achieves 82.1 mAP on Pascal VOC 2007 test set, which outperforms other reinforcement-learning-based methods.
Keyword:
Object detection
Multi-scale imbalance
Reinforcement learning
Multi-task

期刊

Multimedia Tools and Applications 封面图
Multimedia Tools and Applications
IF:
3
论文数:
2.0W
被引数:
3.2W

机构

暂无机构信息
引用论文

引用论文

Adjuvant 5-FU-Based Chemoradiotherapy for Patients Undergoing R-1/R-2 Resections for Pancreatic Cancer
err2006-02-23
err0
PREAI
errHans G. Smeenk; Luca Incrocci; Geert Kazemier; Herman van Dekken; Khe T.C. Tran; Johannes Jeekel; Casper H.J. van Eijck
err分享
err收藏
Engineering Antibodies with C-Terminal Sortase-Mediated Modification for Targeted Nanomedicine
err2019-07-23
err0
PREAI
errRania A. Hashad; Jaclyn L. Lange; Natasha C. W. Tan; Karen Alt; Christoph E. Hagemeyer
err分享
err收藏
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容