Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Attention Convolutional Binary Neural Tree for Fine-Grained Visual Categorization

About

Fine-grained visual categorization (FGVC) is an important but challenging task due to high intra-class variances and low inter-class variances caused by deformation, occlusion, illumination, etc. An attention convolutional binary neural tree architecture is presented to address those problems for weakly supervised FGVC. Specifically, we incorporate convolutional operations along edges of the tree structure, and use the routing functions in each node to determine the root-to-leaf computational paths within the tree. The final decision is computed as the summation of the predictions from leaf nodes. The deep convolutional operations learn to capture the representations of objects, and the tree structure characterizes the coarse-to-fine hierarchical feature learning process. In addition, we use the attention transformer module to enforce the network to capture discriminative features. The negative log-likelihood loss is used to train the entire network in an end-to-end fashion by SGD with back-propagation. Several experiments on the CUB-200-2011, Stanford Cars and Aircraft datasets demonstrate that the proposed method performs favorably against the state-of-the-arts.

Ruyi Ji, Longyin Wen, Libo Zhang, Dawei Du, Yanjun Wu, Chen Zhao, Xianglong Liu, Feiyue Huang• 2019

Related benchmarks

TaskDatasetResultRank
Fine-grained Image ClassificationCUB200 2011 (test)
Accuracy88.1
536
Fine-grained Image ClassificationStanford Cars (test)
Accuracy94.6
348
Image ClassificationStanford Cars (test)
Accuracy94.6
306
Fine-grained visual classificationFGVC-Aircraft (test)
Top-1 Acc92.4
287
Image ClassificationCUB-200-2011 (test)
Top-1 Acc88.1
276
Image ClassificationFGVC-Aircraft (test)
Accuracy92.4
231
Fine-grained Image ClassificationCUB-200 2011
Accuracy88.1
222
Fine-grained Image ClassificationStanford Cars
Accuracy94.6
206
Fine-grained visual classificationFGVC Aircraft
Top-1 Accuracy92.4
41
Fine grained classificationAircraft 1.0 (test)
Top-1 Accuracy92.4
18
Showing 10 of 10 rows

Other info

Code

Follow for update