METHOD AND SYSTEM FOR IMPROVED ATTENTION MAP GUIDANCE FOR VISUAL RECOGNITION IN IMAGES

摘要

A method for generating an improved attention map in an image includes processing an input image in a trained neural network to produce a classification loss, back-propagating the classification loss through the neural network to produce a weight vector in a last convolutional layer of the neural network, applying the weight vector to the input image to produce a attention map, passing the attention map through the neural network to produce an attention loss, and using the classification loss and the attention loss to supervision to the neural network to produce an improved attention map. In some embodiments, a minority percentage of training images may be provided with low-level labels in addition to image-level labels. Low-level labels may include bounding boxes enclosing surface regions of a target object or segmentations mask labels for each pixel identifiable as part of the target object.