English

ReDiffuse: Rotation Equivariant Diffusion Model for Multi-focus Image Fusion

Computer Vision and Pattern Recognition 2026-03-24 v1

Abstract

Diffusion models have achieved impressive performance on multi-focus image fusion (MFIF). However, a key challenge in applying diffusion models to the ill-posed MFIF problem is that defocus blur can make common symmetric geometric structures (e.g., textures and edges) appear warped and deformed, often leading to unexpected artifacts in the fused images. Therefore, embedding rotation equivariance into diffusion networks is essential, as it enables the fusion results to faithfully preserve the original orientation and structural consistency of geometric patterns underlying the input images. Motivated by this, we propose ReDiffuse, a rotation-equivariant diffusion model for MFIF. Specifically, we carefully construct the basic diffusion architectures to achieve end-to-end rotation equivariance. We also provide a rigorous theoretical analysis to evaluate its intrinsic equivariance error, demonstrating the validity of embedding equivariance structures. ReDiffuse is comprehensively evaluated against various MFIF methods across four datasets (Lytro, MFFW, MFI-WHU, and Road-MF). Results demonstrate that ReDiffuse achieves competitive performance, with improvements of 0.28-6.64\% across six evaluation metrics. The code is available at https://github.com/MorvanLi/ReDiffuse.

Keywords

Cite

@article{arxiv.2603.21129,
  title  = {ReDiffuse: Rotation Equivariant Diffusion Model for Multi-focus Image Fusion},
  author = {Bo Li and Tingting Bao and Lingling Zhang and Weiping Fu and Yaxian Wang and Jun Liu},
  journal= {arXiv preprint arXiv:2603.21129},
  year   = {2026}
}

Comments

10 pages, 9 figures

R2 v1 2026-07-01T11:32:01.142Z