QKVA grid: Attention in Image Perspective and Stacked DETR

07/09/2022

∙

We present a new model named Stacked-DETR(SDETR), which inherits the main ideas in canonical DETR. We improve DETR in two directions: simplifying the cost of training and introducing the stacked architecture to enhance the performance. To the former, we focus on the inside of the Attention block and propose the QKVA grid, a new perspective to describe the process of attention. By this, we can step further on how Attention works for image problems and the effect of multi-head. These two ideas contribute the design of single-head encoder-layer. To the latter, SDETR reaches great improvement(+1.1AP, +3.4APs) to DETR. Especially to the performance on small objects, SDETR achieves better results to the optimized Faster R-CNN baseline, which was a shortcoming in DETR. Our changes are based on the code of DETR. Training code and pretrained models are available at https://github.com/shengwenyuan/sdetr.

READ FULL TEXT

QKVA grid: Attention in Image Perspective and Stacked DETR

Grid R-CNN Plus: Faster and Better

Transformer in Convolutional Neural Networks

Deep Language Networks: Joint Prompt Training of Stacked LLMs using Variational Inference

SimA: Simple Softmax-free Attention for Vision Transformers

A Parameter-efficient Multi-subject Model for Predicting fMRI Activity

Superpixel Image Classification with Graph Attention Networks

Multi-Outputs Is All You Need For Deblur

QKVA grid: Attention in Image Perspective and Stacked DETR

Related Research

Grid R-CNN Plus: Faster and Better

Transformer in Convolutional Neural Networks

Deep Language Networks: Joint Prompt Training of Stacked LLMs using Variational Inference

SimA: Simple Softmax-free Attention for Vision Transformers

A Parameter-efficient Multi-subject Model for Predicting fMRI Activity

Superpixel Image Classification with Graph Attention Networks

Multi-Outputs Is All You Need For Deblur