Saved in:
Bibliographic Details
Main Authors: Betti, Federico, Baraldi, Lorenzo, Cucchiara, Rita, Sebe, Nicu
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2409.10597
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916397056000000
author Betti, Federico
Baraldi, Lorenzo
Baraldi, Lorenzo
Cucchiara, Rita
Sebe, Nicu
author_facet Betti, Federico
Baraldi, Lorenzo
Baraldi, Lorenzo
Cucchiara, Rita
Sebe, Nicu
contents Diffusion models have significantly advanced generative AI, but they encounter difficulties when generating complex combinations of multiple objects. As the final result heavily depends on the initial seed, accurately ensuring the desired output can require multiple iterations of the generation process. This repetition not only leads to a waste of time but also increases energy consumption, echoing the challenges of efficiency and accuracy in complex generative tasks. To tackle this issue, we introduce HEaD (Hallucination Early Detection), a new paradigm designed to swiftly detect incorrect generations at the beginning of the diffusion process. The HEaD pipeline combines cross-attention maps with a new indicator, the Predicted Final Image, to forecast the final outcome by leveraging the information available at early stages of the generation process. We demonstrate that using HEaD saves computational resources and accelerates the generation process to get a complete image, i.e. an image where all requested objects are accurately depicted. Our findings reveal that HEaD can save up to 12% of the generation time on a two objects scenario and underscore the importance of early detection mechanisms in generative models.
format Preprint
id arxiv_https___arxiv_org_abs_2409_10597
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Optimizing Resource Consumption in Diffusion Models through Hallucination Early Detection
Betti, Federico
Baraldi, Lorenzo
Baraldi, Lorenzo
Cucchiara, Rita
Sebe, Nicu
Computer Vision and Pattern Recognition
Diffusion models have significantly advanced generative AI, but they encounter difficulties when generating complex combinations of multiple objects. As the final result heavily depends on the initial seed, accurately ensuring the desired output can require multiple iterations of the generation process. This repetition not only leads to a waste of time but also increases energy consumption, echoing the challenges of efficiency and accuracy in complex generative tasks. To tackle this issue, we introduce HEaD (Hallucination Early Detection), a new paradigm designed to swiftly detect incorrect generations at the beginning of the diffusion process. The HEaD pipeline combines cross-attention maps with a new indicator, the Predicted Final Image, to forecast the final outcome by leveraging the information available at early stages of the generation process. We demonstrate that using HEaD saves computational resources and accelerates the generation process to get a complete image, i.e. an image where all requested objects are accurately depicted. Our findings reveal that HEaD can save up to 12% of the generation time on a two objects scenario and underscore the importance of early detection mechanisms in generative models.
title Optimizing Resource Consumption in Diffusion Models through Hallucination Early Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.10597