LCA-CNN

LCA-CNN performs fine-grained image classification by using a Learning Cascade Attention Convolutional Neural Network to distinguish highly similar classes with minimal intra-class variation.


Key Features:

  • Cascade Attention Model: Employs a Deep Convolutional Neural Network (CNN) architecture with cascade attention to improve discrimination of fine-grained classes.
  • Spatial Confusion Attention: Identifies ambiguous regions within input images to focus analysis on areas requiring precise differentiation.
  • Constraint Loss Functions: Includes Spatial Mask Loss to penalize incorrect focus areas and Spatial And Loss to enforce consistent feature presence across class instances.
  • Cross-Network Attention: Uses a two-stream architecture with distinct pre-trained parameters to leverage diverse feature representations.
  • Novel Loss Functions: Implements Cross-network Similarity Loss to align features between streams and Satisfied Rank Loss to enforce prediction ranking consistent with ground truth.
  • Network Fusion Attention: Integrates intermediate results using an entropy add strategy to produce final predictions.
  • Weakly-supervised and Parallelized Training: Operates under a fully weakly-supervised regime and supports parallelized computation for training.
  • End-to-end Training: Architecture permits end-to-end model training.

Scientific Applications:

  • CUB-200-2011: Achieved a classification accuracy of 90.8% on this fine-grained bird dataset.
  • FGVC-Aircraft: Attained a classification accuracy of 92.1% on this fine-grained aircraft dataset.
  • Flower 102: Reached a classification accuracy of 98.5% on this fine-grained flower dataset.

Methodology:

Computational methods include a deep CNN with cascade attention and Spatial Confusion Attention, a two-stream Cross-Network Attention setup with distinct pre-trained parameters, constraint losses (Spatial Mask Loss, Spatial And Loss), Cross-network Similarity Loss, Satisfied Rank Loss, Network Fusion Attention using an entropy add strategy, and weakly-supervised, parallelized end-to-end training.

Topics

Details

Tool Type:
command-line tool
Programming Languages:
Python
Added:
1/14/2020
Last Updated:
12/22/2020

Operations

Publications

Zhu Y, Li R, Yang Y, Ye N. Learning Cascade Attention for fine-grained image classification. Neural Networks. 2020;122:174-182. doi:10.1016/j.neunet.2019.10.009. PMID:31683145.

PMID: 31683145
Funding: - National Key Research and Development Plan of China: 2016YFD0600101 - Jiangsu Provincial Department of Housing and Urban–Rural Development, PR China: 2016ZD44 - Innovation Training Program Projects for Jiangsu College Students, PR China: 201810298052Z