English

Can We Achieve Efficient Diffusion without Self-Attention? Distilling Self-Attention into Convolutions

Computer Vision and Pattern Recognition 2025-05-01 v1

Abstract

Contemporary diffusion models built upon U-Net or Diffusion Transformer (DiT) architectures have revolutionized image generation through transformer-based attention mechanisms. The prevailing paradigm has commonly employed self-attention with quadratic computational complexity to handle global spatial relationships in complex images, thereby synthesizing high-fidelity images with coherent visual semantics.Contrary to conventional wisdom, our systematic layer-wise analysis reveals an interesting discrepancy: self-attention in pre-trained diffusion models predominantly exhibits localized attention patterns, closely resembling convolutional inductive biases. This suggests that global interactions in self-attention may be less critical than commonly assumed.Driven by this, we propose Δ\DeltaConvFusion to replace conventional self-attention modules with Pyramid Convolution Blocks (Δ\DeltaConvBlocks).By distilling attention patterns into localized convolutional operations while keeping other components frozen, Δ\DeltaConvFusion achieves performance comparable to transformer-based counterparts while reducing computational cost by 6929×\times and surpassing LinFusion by 5.42×\times in efficiency--all without compromising generative fidelity.

Keywords

Cite

@article{arxiv.2504.21292,
  title  = {Can We Achieve Efficient Diffusion without Self-Attention? Distilling Self-Attention into Convolutions},
  author = {ZiYi Dong and Chengxing Zhou and Weijian Deng and Pengxu Wei and Xiangyang Ji and Liang Lin},
  journal= {arXiv preprint arXiv:2504.21292},
  year   = {2025}
}
R2 v1 2026-06-28T23:16:13.637Z