Search papers, labs, and topics across Lattice.
This paper introduces ReBridge-Flow, a novel method that enhances image restoration by addressing the issue of bridge mismatch in flow matching through a posterior bridge re-coupling approach. By decoding local source and clean endpoints and incorporating measurement information via clean-side anchoring, the method synchronously re-couples these endpoints to yield a measurement-aware pair that improves local bridge compatibility. Experimental results across various natural and medical image restoration tasks show that ReBridge-Flow significantly enhances structural consistency in restored images, outperforming existing methods.
ReBridge-Flow resolves critical bridge mismatches in flow matching, leading to significantly improved structural consistency in image restoration.
Flow Matching provides an efficient generative prior for image restoration by learning continuous transport between source and data distributions. However, existing methods typically incorporate measurement constraints through local corrections. Such corrections may disrupt the source-clean endpoint coupling implicitly encoded by the pretrained flow, making the corrected endpoint pair incompatible with the current state. To address this issue, we propose ReBridge-Flow, a posterior bridge re-coupling method. Specifically, given the current state, ReBridge-Flow first decodes the corresponding local source and clean endpoints. It then incorporates measurement information through clean-side anchoring and synchronously re-couples the source endpoint, yielding a measurement-aware endpoint pair with improved local bridge compatibility. The re-coupled endpoints further define a posterior-informed transport direction for advancing the sampling process. We also introduce the Posterior Bridge Defect, which jointly characterizes measurement error, deviation from the flow prior, and bridge mismatch, and leads to explicit updates for clean-side anchoring and source-side re-coupling. Extensive experiments on multiple natural and medical image restoration tasks demonstrate that ReBridge-Flow effectively alleviates bridge mismatch and improves the structural consistency of restored images.
