atlas

Research map / Architectures and efficiency / Attention and transformers

SparseFormer: Sparse Visual Recognition via Limited Latent Tokens

ICLR 2024 · Architectures and efficiency › Attention and transformers · approach: Efficient and robust vision transformers · 6 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SparseFormer, a vision transformer that represents images with a small set of latent tokens (down to 49) that adaptively sample image features via bilinear interpolation and RoI adjustment, trained end-to-end with only classification labels.

Problem

Standard vision networks densely process every pixel or patch, incurring computation that scales with input resolution; the paper asks whether recognition can be done sparsely, as human vision appears to do.

Approach

Learnable latent tokens, each with an RoI descriptor, iteratively adjust their RoIs and sample image features at a few bilinear points; a focusing transformer decodes these samples into tokens, and a standard transformer (cortex) operates only on the limited tokens, so cost is nearly independent of input resolution. The model learns where to look from classification loss alone.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use SparseFormer-T instead of Swin-TImageNet-1K classification · ImageNet-1Ktop-1 accuracy: 81.3 -> 81.0Hurt
use SparseFormer-T instead of Swin-TImageNet-1K classification · ImageNet-1KFLOPs: 4.5G versus 2.0GHelped
use SparseFormer-T instead of Swin-TImageNet-1K classification · ImageNet-1Kthroughput: 1270 versus 726Helped
pre-train SparseFormer-B on ImageNet-21KSparseFormer-B · ImageNet-1Ktop-1 accuracy: 1.0 top-1 accuracy improvementHelped
fine-tune with 384² input resolutionSparseFormer-B · ImageNet-1Ktop-1 accuracy: 0.5↑Helped
increase latent tokens to 81 (ablation)SparseFormer-T · ImageNet-1Ktop-1 accuracyHelped
remove RoI adjustment and use fixed token RoIs and sampling positions (ablation)SparseFormer-T · ImageNet-1Ktop-1 accuracyHurt
increase sampling points from 36 to 64 (ablation)SparseFormer-T · ImageNet-1Ktop-1 accuracy: 81.3 accuracyHelped
replace early-convolution image features with raw RGB input (ablation)SparseFormer-T · ImageNet-1Ktop-1 accuracy: about 60 top-1 accuracyHurt
replace early-convolution features with ViT-like patch embeddings (ablation)SparseFormer-T · ImageNet-1Ktop-1 accuracyHurt
use adaptive mixing instead of static mixing to decode sampled features (ablation)SparseFormer-T · ImageNet-1Ktop-1 accuracyHelped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1K ImageNet-21K Kinetics-400 COCO

Benchmarks: ImageNet-1K Kinetics-400 COCO

Metrics: top-1 accuracy throughput FLOPs

Baselines: Swin-T Perceiver ResNet ConvNeXt TimeSFormer ViViT I3D SlowFast

Builds on: Perceiver DETR AdaMixer Swin Transformer dynamic convolution

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy