Search papers, labs, and topics across Lattice.
The paper introduces ArtiAgent, a novel framework for automatically generating artifact-annotated image datasets by injecting realistic flaws into real images using a diffusion transformer. ArtiAgent employs a perception agent for grounding entities, a synthesis agent for artifact injection via patch-wise embedding manipulation, and a curation agent for filtering and explaining artifacts. Experiments on a synthesized dataset of 100K images demonstrate the efficacy of ArtiAgent in improving artifact awareness in VLMs and diffusion models.
Forget hand-labeled data: a new agent-based system automatically generates artifact-annotated datasets, enabling VLMs and diffusion models to better understand and fix visual flaws.
Despite recent advances in diffusion models, AI generated images still often contain visual artifacts that compromise realism. Although more thorough pre-training and bigger models might reduce artifacts, there is no assurance that they can be completely eliminated, which makes artifact mitigation a highly crucial area of study. Previous artifact-aware methodologies depend on human-labeled artifact datasets, which are costly and difficult to scale, underscoring the need for an automated approach to reliably acquire artifact-annotated datasets. In this paper, we propose ArtiAgent, which efficiently creates pairs of real and artifact-injected images. It comprises three agents: a perception agent that recognizes and grounds entities and subentities from real images, a synthesis agent that introduces artifacts via artifact injection tools through novel patch-wise embedding manipulation within a diffusion transformer, and a curation agent that filters the synthesized artifacts and generates both local and global explanations for each instance. Using ArtiAgent, we synthesize 100K images with rich artifact annotations and demonstrate both efficacy and versatility across diverse applications. Code is available at link.
