Global–local feature learning for fine-grained food classification based on Swin Transformer

Jun Hwa Kim, Namho Kim, Chee Sun Won

Research output: Contribution to journalArticlepeer-review

12 Scopus citations

Abstract

Separable object parts, such as the head and tail in a bird, are vital for fine-grained visual classifications. For those objects without separable parts, the classification task relies only on local and global textural image features. Although the Swin Transformer architecture was proposed to efficiently capture both local and global visual features, it still exhibits a bias towards global features. Therefore, our goal is to enhance the local feature learning capability of the Swin Transformer by adding four new modules of the Local Feature Extraction Network (L-FEN), Convolution Patch-Merging (CP), Multi-Path (MP), and Multi-View (MV). The L-FEN enhances the Swin transformer with the improved local feature capture. The CP is a localized and hierarchical adaptation of the Swin's Patch Merging technique. The MP method integrates features across various Swin stages to accentuate local details. Meanwhile, the MV Swin transformer block supersedes traditional Swin blocks with those incorporating varied receptive fields, ensuring a broader scope of local feature capture. Our enhanced architecture, named Global–Local Swin Transformer (GL-Swin), is applied to solve a fine-grained food classification task. On three major food datasets: ISIA Food-500 UEC Food-256, and Food-101, our GL-Swin achieved accuracies of 66.75%, 85.78%, and 92.93% respectively, consistently outperforming other leading methods.

Original languageEnglish
Article number108248
JournalEngineering Applications of Artificial Intelligence
Volume133
DOIs
StatePublished - Jul 2024

Keywords

  • CNN
  • Deep learning
  • Fine-grained visual classification
  • Food dataset
  • Vision transformer

Fingerprint

Dive into the research topics of 'Global–local feature learning for fine-grained food classification based on Swin Transformer'. Together they form a unique fingerprint.

Cite this