Human brains are natural learning systems which inherently recognise image objects in a hierarchical pattern. Similar association exists among different categories of images which interact while training a deep learning model, leading to misclassification and indistinct features. As semantically related classes form clusters in the embedding space, we can list these classes under a single aggregate class, and make the learning faster by leveraging this knowledge while training a model. In this paper, we introduce a hierarchical architecture that simultaneously learns both broad-class and fine-class embeddings using transformer-based encoders. These encoders are trained alternately, and the hierarchical knowledge is encoded through tensor product between the coarse and fine class representations. Our approach has been able to achieve state-of-the-art performance on benchmark datasets with very less epochs.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

DualViT: A Hierarchical Vision Transformer for Broad and Fine Class Embeddings

  • Ankita Chatterjee,
  • Sandip Dutta,
  • Jayanta Mukhopadhyay,
  • Partha Pratim Das

摘要

Human brains are natural learning systems which inherently recognise image objects in a hierarchical pattern. Similar association exists among different categories of images which interact while training a deep learning model, leading to misclassification and indistinct features. As semantically related classes form clusters in the embedding space, we can list these classes under a single aggregate class, and make the learning faster by leveraging this knowledge while training a model. In this paper, we introduce a hierarchical architecture that simultaneously learns both broad-class and fine-class embeddings using transformer-based encoders. These encoders are trained alternately, and the hierarchical knowledge is encoded through tensor product between the coarse and fine class representations. Our approach has been able to achieve state-of-the-art performance on benchmark datasets with very less epochs.