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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Fine-grained Image Classification | CUB200 2011 (test) | Accuracy88.1 | 536 | |
| Fine-grained Image Classification | Stanford Cars (test) | Accuracy94.6 | 348 | |
| Image Classification | Stanford Cars (test) | Accuracy94.6 | 306 | |
| Fine-grained visual classification | FGVC-Aircraft (test) | Top-1 Acc92.4 | 287 | |
| Image Classification | CUB-200-2011 (test) | Top-1 Acc88.1 | 276 | |
| Image Classification | FGVC-Aircraft (test) | Accuracy92.4 | 231 | |
| Fine-grained Image Classification | CUB-200 2011 | Accuracy88.1 | 222 | |
| Fine-grained Image Classification | Stanford Cars | Accuracy94.6 | 206 | |
| Fine-grained visual classification | FGVC Aircraft | Top-1 Accuracy92.4 | 41 | |
| Fine grained classification | Aircraft 1.0 (test) | Top-1 Accuracy92.4 | 18 |