Efficient Rectified Flow for Image Fusion

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Zirui, Zhang, Jiayi, Guan, Tianwei, Zhou, Yuhan, Li, Xingyuan, Dong, Minjing, Liu, Jinyuan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918147034972160
author Wang, Zirui
Zhang, Jiayi
Guan, Tianwei
Zhou, Yuhan
Li, Xingyuan
Dong, Minjing
Liu, Jinyuan
author_facet Wang, Zirui
Zhang, Jiayi
Guan, Tianwei
Zhou, Yuhan
Li, Xingyuan
Dong, Minjing
Liu, Jinyuan
contents Image fusion is a fundamental and important task in computer vision, aiming to combine complementary information from different modalities to fuse images. In recent years, diffusion models have made significant developments in the field of image fusion. However, diffusion models often require complex computations and redundant inference time, which reduces the applicability of these methods. To address this issue, we propose RFfusion, an efficient one-step diffusion model for image fusion based on Rectified Flow. We incorporate Rectified Flow into the image fusion task to straighten the sampling path in the diffusion model, achieving one-step sampling without the need for additional training, while still maintaining high-quality fusion results. Furthermore, we propose a task-specific variational autoencoder (VAE) architecture tailored for image fusion, where the fusion operation is embedded within the latent space to further reduce computational complexity. To address the inherent discrepancy between conventional reconstruction-oriented VAE objectives and the requirements of image fusion, we introduce a two-stage training strategy. This approach facilitates the effective learning and integration of complementary information from multi-modal source images, thereby enabling the model to retain fine-grained structural details while significantly enhancing inference efficiency. Extensive experiments demonstrate that our method outperforms other state-of-the-art methods in terms of both inference speed and fusion quality. Code is available at https://github.com/zirui0625/RFfusion.
format Preprint
id arxiv_https___arxiv_org_abs_2509_16549
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Efficient Rectified Flow for Image Fusion
Wang, Zirui
Zhang, Jiayi
Guan, Tianwei
Zhou, Yuhan
Li, Xingyuan
Dong, Minjing
Liu, Jinyuan
Computer Vision and Pattern Recognition
Image fusion is a fundamental and important task in computer vision, aiming to combine complementary information from different modalities to fuse images. In recent years, diffusion models have made significant developments in the field of image fusion. However, diffusion models often require complex computations and redundant inference time, which reduces the applicability of these methods. To address this issue, we propose RFfusion, an efficient one-step diffusion model for image fusion based on Rectified Flow. We incorporate Rectified Flow into the image fusion task to straighten the sampling path in the diffusion model, achieving one-step sampling without the need for additional training, while still maintaining high-quality fusion results. Furthermore, we propose a task-specific variational autoencoder (VAE) architecture tailored for image fusion, where the fusion operation is embedded within the latent space to further reduce computational complexity. To address the inherent discrepancy between conventional reconstruction-oriented VAE objectives and the requirements of image fusion, we introduce a two-stage training strategy. This approach facilitates the effective learning and integration of complementary information from multi-modal source images, thereby enabling the model to retain fine-grained structural details while significantly enhancing inference efficiency. Extensive experiments demonstrate that our method outperforms other state-of-the-art methods in terms of both inference speed and fusion quality. Code is available at https://github.com/zirui0625/RFfusion.
title Efficient Rectified Flow for Image Fusion
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2509.16549