Enhancing Few-shot Image Classification with Cosine Transformer

This repo contains the official implementation code for the paper Enhancing Few-shot Image Classification with Cosine Transformer (IEEE Access). In this project, we developed a transformer-based algorithm FS-CT for few-shot classification and cross-attention mechansim, where we proved that cosine similarity benefits attention mechanism and and improve few-shot algorithms across settings and datasets. In particular, with the proposed Cosine attention, we achieve a more stable and consistent output as correlation map between support and query feature and thus improve ViT-bases few-shot algorithms' performance greatly.

Table of Content

Enhancing Few-shot Image Classification with Cosine Transformer

Few-shot Cosine Transformer

The overall architecture of the proposed Few-shot Cosine Transformer, which includes two main components: (a) learnable prototypical embedding that calculates the categorical proto representation given random support features that might be either in the far margin of the distribution or very close to each other and (b) Cosine transformer that determines the similarity matrix between proto representations and query samples for the few-shot classification tasks. The heart of the transformer architecture is Cosine attention, an attention mechanism with cosine similarity and no softmax function to deal with two different sets of features. The Cosine transformer shares a similar architecture with a standard transformer encoder block, with two skip connections to preserve information, a two-layer feed-forward network, and layer normalization between them to reduce noise. The outcome value is through a cosine linear layer, with cosine similarity replacing the dot-product, before feeding to softmax for query prediction.

Experiments

Dependencies environment

pip install -r requirements.txt

Dataset

mini-ImageNet:
- Go to /dataset/miniImagenet/
- Download the dataset from download_miniImagenet.txt
- Run source download_miniImagenet.sh for processing dataset.
- When complete, there are three JSON files base.json, val.json, and novel.json for experiments
CUB-200:
- Go to /dataset/CUB/
- Processing data similar to mini-Imagenet dataset
CIFAR-FS:
- Go to /dataset/CIFAR_FS/
- Processing data similar to mini-Imagenet dataset
Omniglot:
- Go to /dataset/Omniglot/
- Run source download_Omniglot.sh
Yoga:
- This is our custom dataset with 50 yoga pose categories and 2480 images, including 50 categiores for training, 13 for validating, and 12 for testing set
- Go to /dataset/Yoga/
- Run source yoga_processing.sh
Custom dataset:
- Require three data split json file: base.json, val.json, novel.json
- The format should follow:
```
{
  "label_names": `["class0","class1",...]`,
  "image_names": `["filepath1","filepath2",...]`,
  "image_labels": `[l1,l2,l3,...]`
}
```
- Put these file in the same folder and change data_dir ['DATASETNAME'] in configs.py to the corresponded folder path
- See other datasets folder for examples

Running scripts

Python scripts
- Testing only: test.py (does not support WandB )
- Training and testing: train_test.py
Configurations pool:
- Backbones: Conv4/Conv6/ResNet18/ResNet34
- Methods: CTX_softmax/CTX_cosine/FSCT_softmax/FSCT_cosine
  - softmax is the baseline scaled dot-product attention mechanism
  - cosine is our proposed Cosine attention mechanism
- Dataset: miniImagenet/CUB/CIFAR/Omniglot/Yoga
Main parameters:
- --backbone: backbone model (default ResNet34)
- --FETI: Using FETI (Feature Extractor Trained partially on ImageNet) for ResNet Backbone if 1, none if 0 (default 0)
- --method: few-shot method algorithm (default FSCT_cosine)
- --n_way: number of catergories for classification (default 5)
- --k_shot: number of shot per category in Support set (default 5)
- --n_episode: number of training/validating episodic batch per epoch
- --train_aug: apply augmentation if 1, none if 0 (default 0)
- --num_epoch: number of training epoch (default 50)
- --wandb: saving training log and plot visualization into WandB server if 1, none if 0 (default 0)
- For other parameters, please read io_utils.py for detail information.
Example:
python train_test.py --method FSCT_cosine --dataset miniImagenet --backbone ResNet34 --FETI 1 --n_way 5 --k_shot 5 --train_aug 0 --wandb 1
Bash script for multiple running:
- source run_script.sh
- Parameters can be modified within the script for specific experiments, including dataset, backbone, method, n_way, k_shot, augmentation
- All the method automatically push the training loss/val logs into WandB server. Set --wandb 0 to turn it off
Result logs after testing will be saved in record/results.txt

Results

Our method Few-Shot TransFormer achieves the following performances on:

Dataset	1-shot Accuracy	5-shot Accuracy
mini-ImageNet	55.87+-0.86%	73.42+-0.67%
CIFAR-FS	67.06+-0.89%	82.89+-0.61%
CUB	81.23+-0.77%	92.25+-0.37%

All results are stored in record/official_results.txt

References

Acknowledgment

This repository is mainly based on "A Closer Look at Few-shot Classification" official GitHub Repository: wyharveychen/CloserLookFewShot
The CrossTransformer CTX implemented code in this repository is modified from lucidrains/cross-transformers-pytorch

Citation

If you find our code useful, please consider citing our work using the bibtex:

@article{nguyen2023FSCT,
  author={Nguyen, Quang-Huy and Nguyen, Cuong Q. and Le, Dung D. and Pham, Hieu H.},
  journal={IEEE Access}, 
  title={Enhancing Few-Shot Image Classification With Cosine Transformer}, 
  year={2023},
  volume={11},
  number={},
  pages={79659-79672},
  doi={10.1109/ACCESS.2023.3298299}}

Contact

If you have any concerns or support need on this repository, please send me an email at quanghuy0497@gmail.com

vinuni-vishc/Few-Shot-Cosine-Transformer