DIAMOND: Directed Inference for Artifact Mitigation in Flow Matching Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Polowczyk, Alicja, Polowczyk, Agnieszka, Borycki, Piotr, Waczyńska, Joanna, Tabor, Jacek, Spurek, Przemysław
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914298864861184
author Polowczyk, Alicja
Polowczyk, Agnieszka
Borycki, Piotr
Waczyńska, Joanna
Tabor, Jacek
Spurek, Przemysław
author_facet Polowczyk, Alicja
Polowczyk, Agnieszka
Borycki, Piotr
Waczyńska, Joanna
Tabor, Jacek
Spurek, Przemysław
contents Despite impressive results from recent text-to-image models like FLUX, visual and anatomical artifacts remain a significant hurdle for practical and professional use. Existing methods for artifact reduction, typically work in a post-hoc manner, consequently failing to intervene effectively during the core image formation process. Notably, current techniques require problematic and invasive modifications to the model weights, or depend on a computationally expensive and time-consuming process of regional refinement. To address these limitations, we propose DIAMOND, a training-free method that applies trajectory correction to mitigate artifacts during inference. By reconstructing an estimate of the clean sample at every step of the generative trajectory, DIAMOND actively steers the generation process away from latent states that lead to artifacts. Furthermore, we extend the proposed method to standard Diffusion Models, demonstrating that DIAMOND provides a robust, zero-shot path to high-fidelity, artifact-free image synthesis without the need for additional training or weight modifications in modern generative architectures. Code is available at https://gmum.github.io/DIAMOND/
format Preprint
id arxiv_https___arxiv_org_abs_2602_00883
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle DIAMOND: Directed Inference for Artifact Mitigation in Flow Matching Models
Polowczyk, Alicja
Polowczyk, Agnieszka
Borycki, Piotr
Waczyńska, Joanna
Tabor, Jacek
Spurek, Przemysław
Computer Vision and Pattern Recognition
Artificial Intelligence
Despite impressive results from recent text-to-image models like FLUX, visual and anatomical artifacts remain a significant hurdle for practical and professional use. Existing methods for artifact reduction, typically work in a post-hoc manner, consequently failing to intervene effectively during the core image formation process. Notably, current techniques require problematic and invasive modifications to the model weights, or depend on a computationally expensive and time-consuming process of regional refinement. To address these limitations, we propose DIAMOND, a training-free method that applies trajectory correction to mitigate artifacts during inference. By reconstructing an estimate of the clean sample at every step of the generative trajectory, DIAMOND actively steers the generation process away from latent states that lead to artifacts. Furthermore, we extend the proposed method to standard Diffusion Models, demonstrating that DIAMOND provides a robust, zero-shot path to high-fidelity, artifact-free image synthesis without the need for additional training or weight modifications in modern generative architectures. Code is available at https://gmum.github.io/DIAMOND/
title DIAMOND: Directed Inference for Artifact Mitigation in Flow Matching Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2602.00883