Attention Mining Branch for Optimizing Attention Map
- Takaaki Iwayoshi, Masahiro Mitsuhara, Masayuki Takada, Tsubasa Hirakawa, Takayoshi Yamashita, Hironobu Fujiyoshi
- International Conference on Machine Vision Applications, 2021
Download: PDF (English)
Attention branch network (ABN) achieves high accuracy by visualizing the attention area of the network during inference and utilizing it in the recognition process. Meanwhile, if the attention area does not highlight the target object to be recognized, it may causes failure recognition. To solve this problem, a fine-tuning method of the ABN using modified attention maps by human knowledge has been proposed. However, this method takes human and time costs because the attention map must be modified by humans. In this paper, we propose a method that automatically optimize the attention map by introducing an attention mining branch to the ABN.