arXiv Machine Learning

Efficient Image Restoration with State-Dependent Forward Diffusion

arXiv:2505. 16733v3 Announce Type: replace Abstract: This paper proposes to perform image restoration through a state-dependent mean-reverting forward diffusion (FoD) process.

arXiv Computer Vision
Aug 25

Pixel-Space Diffusion via Observation Operators

Pixel‑Space Diffusion via Observation Operators introduces a new framework for pixel‑space diffusion models that addresses a scale‑time mismatch in existing methods. By replacing fixed full‑image supervision with a time‑indexed observation trajectory that progresses from coarse structures to the full image, the model aligns supervision with the natural recovery order of image details. The approach employs Gaussian‑Lanczos operators and a GL‑CoDA decoder to refine features progressively, resulting in faster convergence and higher generation quality, achieving an FID of 1.52 on ImageNet‑256.

By Shaojie Guo, Lichen Ma, Haoyang Tong, Yu He, Zipeng Guo, Xiaoan Liu, Feng Yan, Yu Guo, Fei Wang, Junshi Huang, Yan Wang
arXiv Machine Learning
Jun 29

StableMotion: One-Step Motion Estimation with Diffusion Prior

arXiv:2505. 06668v2 Announce Type: replace-cross Abstract: We present StableMotion, a novel framework that leverages geometric and content priors from pretrained large-scale image diffusion models for motion estimation in single-image rectification tasks such as Stitched Image Rectangling (SIR) and Rolling Shutter Correction (RSC).

By Ziyi Wang, Haipeng Li, Lin Sui, Tianhao Zhou, Hai Jiang, Lang Nie, Bing Zeng, Shuaicheng Liu
arXiv Computer Vision
Aug 31

Physics-Guided Flow Matching for CT Image Reconstruction

The paper introduces a high‑resolution Rectified Flow Matching model trained on 256×256 chest CT images to serve as a generative prior for CT reconstruction. A two‑stage training strategy—initial strong anatomically informed augmentation followed by fine‑tuning—helps mitigate overfitting and improve structural fidelity. When evaluated on various CT inverse problems, Flow Matching‑based reconstruction methods outperform diffusion‑based algorithms in PSNR, SSIM, and perceptual quality while requiring fewer sampling steps.

By Davide Evangelista
arXiv AI
Jul 7

Efficient Flow Matching for Sparse-View CT Reconstruction

arXiv:2603. 00205v2 Announce Type: replace-cross Abstract: Generative models, particularly Diffusion Models (DM), have shown strong potential for Computed Tomography (CT) reconstruction serving as expressive priors for solving ill-posed inverse problems.

By Jiayang Shi, Lincen Yang, Zhong Li, Tristan van Leeuwen, Daniel M. Pelt, K. Joost Batenburg
arXiv Computer Vision
Sep 3

Local Epistemic Uncertainty Guided Active Sampling for Plug-and-play Diffusive Image Restoration

The paper introduces LEADer, a framework that uses local epistemic uncertainty to guide active sampling in diffusion-based image restoration. By adjusting prior strength per pixel and pruning sampling trajectories based on uncertainty traces, LEADer balances detail preservation with artifact suppression and accelerates convergence. The method is plug‑and‑play, theoretically guarantees data consistency and stable convergence, and improves performance across multiple state‑of‑the‑art diffusion models with minimal memory overhead.

By Jiaqi Zhang, Zheng Pang, Rongrong Gao, Qiyuan Zhang, Yang Yang