Dual-Path Deep Unsupervised Learning for Multi-Focus Image Fusion
Multi-focus image fusion (MFIF) aims at merging multiple images captured at different focal lengths to create an all-in-focus image. This paper introduces a fully unsupervised learning approach for MFIF that uses only pairs of defocused images for end-to-end training, bypassing the need for ground-truths in supervised learning. Unlike existing methods training via a similarity loss between fused and source images, we propose a dual-path learning framework comprising two networks: an image fuser and a mask predictor. The mask predictor is modeled as a self-supervised denoising network on imperfect fusion masks, trained with a masking-based unsupervised learning scheme. The image fuser, crafted with deep unrolling, leverages the output from the mask predictor to supervise its mask generation at each unrolled step. Moreover, we introduce a fusion consistency loss to ensure the alignment between the image fuser and the mask predictor. In extensive experiments, our proposed approach shows superiority over existing end-to-end unsupervised methods and competitive performance against the supervised ones.
Authors
- Hui Ji (ORCID: https://orcid.org/0000-0002-1674-6056)
- Yuhui Quan (ORCID: https://orcid.org/0000-0002-2564-7703)
- Yan Huang (ORCID: https://orcid.org/0000-0001-9136-195X)
- Xi Wan
- Tianxiang Zheng (ORCID: https://orcid.org/0000-0002-7133-7618)
Institutions
- National University of Singapore (SG)
- South China University of Technology (CN)
Publication Details
- Journal
- IEEE Transactions on Multimedia
- Published
- 2024-12-23
- DOI
- https://doi.org/10.1109/tmm.2024.3521817
- Citations
- 4
- Primary Topic
- Image Processing Techniques and Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.28
Funders
- National Natural Science Foundation of China