Research map / Image generation / Autoregressive and latent generation
DiCo: Revitalizing ConvNets for Scalable and Efficient Diffusion Modeling
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 DiCo, a fully convolutional diffusion backbone (1x1 pointwise + 3x3 depthwise convs with compact channel attention) that matches or beats DiT and linear-attention/Mamba diffusion models on ImageNet at lower compute.
Problem
Diffusion Transformers are strong but computationally expensive due to quadratic self-attention, and recent linear/RNN-style alternatives (Mamba, linear attention) do not yield much runtime gain at high resolution. The paper asks whether a hardware-efficient ConvNet backbone can match DiT generation
Approach
Analysis of pretrained DiTs shows attention is mostly local; replacing self-attention with convolutions causes channel redundancy (many inactive channels), which a compact channel attention (GAP + 1x1 conv + sigmoid gating) fixes, yielding a U-shaped pure-ConvNet diffusion model.
Key results
- DiCo-XL with CFG reaches FID 2.05 at 256x256 and 2.53 at 512x512, with 2.7x and 3.1x speedup over DiT-XL/2 respectively.
- DiCo-H (1B parameters) reaches FID 1.90 on ImageNet 256x256 without extra modern tricks.
- Removing compact channel attention raises FID from 49.97 to 54.78 on the small model (ablation), i.e. CCA gives about 4.81 points improvement.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Replace self-attention with DiCo's convolutional backbone | DiT-XL/2 · ImageNet 256×256, with CFG | FID: 2.27 -> 2.05 | Helped |
| Replace self-attention with DiCo's convolutional backbone | DiT-XL/2 · ImageNet 512×512, with CFG | FID: 3.04 -> 2.53 | Helped |
| Use text-conditioned depthwise convolution instead of cross-attention | DiCo text-to-image · MS-COCO | FID: 4.87 -> 4.93 | Hurt |
| Replace GELU with ReLU (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 51.26 | Hurt |
| Increase DWC kernel from 3×3 to 5×5 (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 48.03 | Helped |
| Increase DWC kernel from 3×3 to 7×7 (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 47.49 | Helped |
| Add compact channel attention (CCA) (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 54.78 -> 49.97 | Helped |
| Replace CCA with an SE module (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 50.89 | Hurt |
| Replace CCA with channel self-attention (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 50.24 | Hurt |
| Replace the Conv Module with self-attention (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 54.00 | Hurt |
| Replace the Conv Module with window attention (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 53.23 | Hurt |
| Replace the Conv Module with focused linear attention (ablation) | DiCo-S · ImageNet 256×256, 400K, without CFG | FID: 49.97 -> 50.60 | Hurt |
Surprising findings
- Naive replacement of self-attention with convolution in DiT yields many inactive channels, causing substantial channel redundancy and degraded FID.
- Pretrained DiT attention maps are predominantly local, with only a few neighboring tokens contributing significantly to a given anchor token.
Limitations
- Scaling limited to 1B parameters due to compute; larger models not explored.
- Text-to-image experiments are small-scale, trained from scratch on MS-COCO rather than large-scale T2I.
Experimental setup
Datasets: ImageNet-1K MS-COCO
Benchmarks: ImageNet 256x256 ImageNet 512x512 ImageNet 1024x1024 MS-COCO
Metrics: FID IS Precision Recall sFID Gflops throughput
Baselines: DiT DiG DiC DiS DiM DiffuSSM DiMSUM U-ViT FlowDCN LightningDiT MMDiT+REPA
Builds on: DiT ConvNeXt iDDPM classifier-free guidance
Closest papers by meaning
- DiC: Rethinking Conv3x3 Designs in Diffusion Models (CVPR 2025 · similarity 0.92)
- CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Up (NeurIPS 2025 · similarity 0.92)
- SD-DiT: Unleashing the Power of Self-supervised Discrimination in Diffusion Transformer (CVPR 2024 · similarity 0.90)
- DiG: Scalable and Efficient Diffusion Models with Gated Linear Attention (CVPR 2025 · similarity 0.90)
- Faster Diffusion: Rethinking the Role of the Encoder for Diffusion Model Inference (NeurIPS 2024 · similarity 0.89)
- Why DiT Models Underperform as Representation Learners without Long Skip Connections (NeurIPS 2026 · similarity 0.89)
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.
