使用增强学习训练多焦点聚焦模型

摘要

聚焦模型(Attention model,AM)将计算资源集中于输入数据特定区域,相比卷积神经网络,AM具有参数少、计算量独立输入和高噪声下正确率较高等优点.相对于输入图像和识别目标,聚焦区域通常较小;如果聚焦区域过小,就会导致过多的迭代次数,降低了效率,也难以在同一输入中寻找多个目标.因此本文提出多焦点聚焦模型,同时对多处并行聚焦.使用增强学习(Reinforce learning,RL)进行训练,将所有焦点的行为统一评分训练.与单焦点聚焦模型相比,训练速度和识别速度提高了25％.同时本模型具有较高的通用性.%Attention model (AM) concentrates computing resources on specific areas of the input data.Compared with the convolutional neural network,AM has many advantages:fewer parameters,the amount of computation being independent of the input,higher tolerance for noise input,etc.Generally,the focused area is smaller than the input image and target.However,if the focused area is too small,it will lead to more iterations and a low efficiency;besides,it is difficult to recognize multiple targets in the same input.Therefore,this paper proposes a multi-focus model.However,if on multiple focuses in parallel.This model uses reinforce learning (RL) to train,and scores the behaviors of all focuses uniformly during training.Compared with the single focus model,both the training and recognition speeds are improved by 25 ％.At the same time,the model has good generality.