Diffusion Probe: Generated Image Result Prediction Using CNN Probes
Diffusion Probe introduces a lightweight framework that predicts final image quality by analyzing internal cross-attention maps from early diffusion steps. By training a small network to map these early attention statistics to a quality score, it allows for fast, cheap forecasting of image fidelity before full generation is complete. This enables efficient decision-making in tasks like prompt optimization and seed selection.