atlas

Research map / Architectures and efficiency / CNN design

AsCAN: Asymmetric Convolution-Attention Networks for Efficient Recognition and Generation

NeurIPS 2024 · Architectures and efficiency › CNN design · approach: Efficient CNN scaling and dynamic inference · 1 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 AsCAN, a hybrid convolution-transformer architecture with an asymmetric macro design (more conv blocks early, more transformer blocks late), evaluated on ImageNet classification, ADE20K segmentation, class-conditional and text-to-image diffusion generation.

Problem

Hybrid convolution-transformer architectures are typically designed at the block level with efficient attention approximations, while the macro arrangement of convolutional and transformer blocks across stages is under-explored, leaving latency-performance trade-offs suboptimal across recognition an

Approach

Use vanilla attention and FusedMBConv blocks but distribute them asymmetrically across stages: first stage only convolutions, early stages conv-heavy, later stages transformer-heavy. This macro design gives better throughput than efficient-attention hybrids without optimizing attention kernels, and the same architecture is scaled to a UNet for text-to-image diffusion with a multi-stage training pi

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use an asymmetric rather than symmetric distribution of convolution and transformer blocks (ablation)AsCAN image classifier · ImageNet-1Ktop-1 accuracy: 82.8% -> 83.4%Helped
Put convolution blocks before transformer blocks within stages (ablation)AsCAN image classifier · ImageNet-1Kthroughput, A100 B=16 (images/s): 1428 -> 3224Helped
Keep convolution blocks in the stem instead of transformer blocks (ablation)AsCAN image classifier · ImageNet-1Kthroughput, A100 B=64 (images/s): 1584 -> 4295Helped
Replace MBConv with FusedMBConv (ablation)AsCAN hybrid block · ImageNet-1Kthroughput, A100 B=64 (images/s): 3013 -> 4295Helped
Use vanilla attention instead of multi-axial attention (ablation)AsCAN transformer block · ImageNet-1Kthroughput, A100 B=64 (images/s): 3541 -> 4295Helped
Increase transformer blocks in the network from configuration C6 to C9 (ablation)AsCAN image classifier · ImageNet-1Ktop-1 accuracy: 82.8% -> 82.7%No clear effect
Use sampled cfg (ablation)AsCAN class-conditional UNet · ImageNet-1K 256×256 class-conditional generationFID: 2.41 -> 2.23Helped
Use AsCAN rather than FasterViT-1ImageNet-1K classification · ImageNet-1Kthroughput, A100 B=16 (images/s): 1123 -> 3224Helped
Use AsCAN rather than SDXLText-to-image generation · GenEvaloverall score: 0.55 -> 0.64Helped
Use AsCAN rather than SDXLText-to-image generation · Validation Set-B-10KFID: 35.86 -> 15.45Helped
Use AsCAN-T rather than Swin-T as the UPerNet backboneUPerNet · ADE20Klatency, A100 (frames/s): 44 -> 64Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1K ImageNet-21K ADE20K GenEval proprietary/licensed T2I image-text data

Benchmarks: ImageNet-1K ImageNet-21K ADE20K GenEval ImageNet-1K class-conditional FID

Metrics: top-1 accuracy throughput (images/s) FID mIoU GenEval overall score user preference rate

Baselines: MaxViT CoAtNet FasterViT PixArt-alpha PixArt-Sigma SDXL DiT-XL/2-G U-ViT ADM LDM Swin SMT

Builds on: CoAtNet MaxViT ConvNeXt PixArt-alpha DDPM

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