Refined Probability Distribution Module for Fine-Grained Visual Categorization

Peipei Zhao,Qiguang Miao,Hongsheng Li,Ruyi Liu,Yining Quan,Jianfeng Song
DOI: https://doi.org/10.1016/j.neucom.2022.10.004
IF: 6
2022-01-01
Neurocomputing
Abstract:Fine-grained visual categorization is an important task in computer vision. Prior works on fine-grained visual categorization have paid much attention to addressing intra-class variation and inter-class similarity. However, they rarely study that task from the perspective of probability distribution. In this paper, we propose a novel refined probability distribution module based on deep convolutional neural network. Our module computes the probability of an image by fully utilizing the similarity information between images. Firstly, we use deep neural networks to obtain the initial probability distribution and extract features. Then, we build a network whose inputs are features for calculating image-to-image similarity scores. Finally, our module refines the initial probability distribution based on an effective batch random walk operation with similarity scores. Our module can be plugged into many deep convolutional neural networks. Experimental results show that our approach outperforms state-of-the-art methods on the CUB-200–2011, FGVC-Aircraft and Stanford Cars datasets respectively.
What problem does this paper attempt to address?