atlas

Research map / Image generation / Autoregressive and latent generation

Exploring DCN-like architecture for fast image generation with arbitrary resolution

NeurIPS 2024 · Image generation › Autoregressive and latent generation · approach: High-resolution diffusion synthesis · 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 FlowDCN, a purely convolution-based flow-matching diffusion model built on group-wise multiscale deformable convolution, for fast arbitrary-resolution image generation with linear complexity.

Problem

Arbitrary-resolution image generation with diffusion transformers suffers from quadratic attention cost and poor resolution extrapolation due to absolute position embeddings.

Approach

Decouples deformable convolution's sampling field into a learnable scale and a fixed direction, assigns different scale priors to channel groups (group-wise multiscale deformable block), removes long residual connections, and rescales the maximum sampling scale S_max by the train/test resolution ratio for resolution extrapolation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use FlowDCN-S/2 instead of SiT-S/2CIFAR10 32×32 image generation · CIFAR10 32×32FID: 7.42 -> 5.47Helped
remove group-wise multiscale deformable convolution (ablation)FlowDCN-S/2 · CIFAR10 32×32FID: 5.47 -> 5.72Hurt
use random prior initialization instead of predefined direction grids and linearly increased scale priors (ablation)FlowDCN-S/2 · CIFAR10 32×32FID: 5.47 -> 5.68Hurt
increase kernel size from 9 to 32 sampling points (ablation)FlowDCN-S/2 · CIFAR10 32×32FID: 5.47 -> 5.13Helped
learn direction priors instead of fixing them (ablation)FlowDCN-S/2 with fixed relative scale · CIFAR10 32×32FID: 5.6 -> 6.01Hurt
learn relative scale rather than fixing it (ablation)FlowDCN-S/2 with fixed direction priors · CIFAR10 32×32FID: 5.6 -> 5.47Helped
replace RMSNorm and SwiGLU with LayerNorm and FFN (ablation)FlowDCN-B/2 · ImageNet 256×256, 400k stepsFID: 28.5 -> 29.1Hurt
replace deformable convolution with standard group-wise convolution (ablation)FlowDCN-S/2 · ImageNet 256×256, 400k stepsFID: 54.6 -> 59.0; 54.6 -> 63.0Hurt
adjust Smax for inference resolution (ablation)FlowDCN-B/2 · ImageNet 320×320 and 224×448FID: 34.4 -> 35.7; 71.7 -> 81.1Mixed
use FlowDCN-XL/2 instead of SiT-XL/2ImageNet 256×256 generation with CFG=1.375 and ODE · ImageNet 256×256sFID: 4.50 -> 4.30Helped
increase FlowDCN training from 384M to 486M total imagesFlowDCN-XL/2 with CFG=1.375 and ODE · ImageNet 256×256FID: 2.13 -> 2.01Helped
use FlowDCN-XL/2 instead of SiT-XL/2-GImageNet 512×512 generation · ImageNet 512×512FID: 2.62 -> 2.44Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet CIFAR-10

Benchmarks: ImageNet 256x256 ImageNet 512x512 CIFAR-10 32x32 ImageNet arbitrary-resolution extrapolation (320x320, 224x448, 160x480)

Metrics: FID sFID Inception Score Precision Recall FLOPs Latency

Baselines: DiT-XL/2 SiT-XL/2 FiT-XL/2 DiffusionSSM-XL ADM-U LDM-4 U-ViT-H/2 StyleGAN-XL BigGAN-deep

Builds on: DiT SiT deformable convolution (DCN) DCNv4 FiT

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