MACS: Measurement-Aware Consistency Sampling for Inverse Problems

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Tanevardi, Amirreza, Moghadam, Pooria Abbas Rad, Eshtehardian, Seyed Mohammad, Amini, Sajjad, Khalaj, Babak
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912745391128576
author Tanevardi, Amirreza
Moghadam, Pooria Abbas Rad
Eshtehardian, Seyed Mohammad
Amini, Sajjad
Khalaj, Babak
author_facet Tanevardi, Amirreza
Moghadam, Pooria Abbas Rad
Eshtehardian, Seyed Mohammad
Amini, Sajjad
Khalaj, Babak
contents Diffusion models have emerged as powerful generative priors for solving inverse imaging problems. However, their practical deployment is hindered by the substantial computational cost of slow, multi-step sampling. Although Consistency Models (CMs) address this limitation by enabling high-quality generation in only one or a few steps, their direct application to inverse problems has remained largely unexplored. This paper introduces a modified consistency sampling framework specifically designed for inverse problems. The proposed approach regulates the sampler's stochasticity through a measurement-consistency mechanism that leverages the degradation operator, thereby enforcing fidelity to the observed data while preserving the computational efficiency of consistency-based generation. Comprehensive experiments on the Fashion-MNIST and LSUN Bedroom datasets demonstrate consistent improvements across both perceptual and pixel-level metrics, including the Fréchet Inception Distance (FID), Kernel Inception Distance (KID), peak signal-to-noise ratio (PSNR), and structural similarity index measure (SSIM), compared with baseline consistency and diffusion-based sampling methods. The proposed method achieves competitive or superior reconstruction quality with only a small number of sampling steps.
format Preprint
id arxiv_https___arxiv_org_abs_2510_02208
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MACS: Measurement-Aware Consistency Sampling for Inverse Problems
Tanevardi, Amirreza
Moghadam, Pooria Abbas Rad
Eshtehardian, Seyed Mohammad
Amini, Sajjad
Khalaj, Babak
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
Diffusion models have emerged as powerful generative priors for solving inverse imaging problems. However, their practical deployment is hindered by the substantial computational cost of slow, multi-step sampling. Although Consistency Models (CMs) address this limitation by enabling high-quality generation in only one or a few steps, their direct application to inverse problems has remained largely unexplored. This paper introduces a modified consistency sampling framework specifically designed for inverse problems. The proposed approach regulates the sampler's stochasticity through a measurement-consistency mechanism that leverages the degradation operator, thereby enforcing fidelity to the observed data while preserving the computational efficiency of consistency-based generation. Comprehensive experiments on the Fashion-MNIST and LSUN Bedroom datasets demonstrate consistent improvements across both perceptual and pixel-level metrics, including the Fréchet Inception Distance (FID), Kernel Inception Distance (KID), peak signal-to-noise ratio (PSNR), and structural similarity index measure (SSIM), compared with baseline consistency and diffusion-based sampling methods. The proposed method achieves competitive or superior reconstruction quality with only a small number of sampling steps.
title MACS: Measurement-Aware Consistency Sampling for Inverse Problems
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2510.02208