---
license: cc-by-nc-4.0
pipeline_tag: image-to-image
library_name: diffusers
---

# The Consistency Critic: Correcting Inconsistencies in Generated Images via Reference-Guided Attentive Alignment

This repository hosts **ImageCritic**, a reference-guided post-editing approach designed to correct inconsistencies in generated images. It aims to solve the inconsistency problem in generated images by applying attention alignment and a detail encoder, providing significant improvements over existing methods in various customized generation scenarios.

The model was presented in the paper [The Consistency Critic: Correcting Inconsistencies in Generated Images via Reference-Guided Attentive Alignment](https://huggingface.co/papers/2511.20614).

*   📚 [Paper (arXiv)](https://arxiv.org/abs/2511.20614)
*   🌐 [Project Page](https://ouyangziheng.github.io/ImageCritic-Page/)
*   💻 [Code (GitHub)](https://github.com/HVision-NKU/ImageCritic)
*   🤗 [Hugging Face Space Demo](https://huggingface.co/spaces/ziheng1234/ImageCritic)
*   📦 [Hugging Face Dataset](https://huggingface.co/datasets/ziheng1234/Critic-10K)

<img src='https://raw.githubusercontent.com/HVision-NKU/ImageCritic/refs/heads/master/figure/teaser.png' width='100%' />

## 🖼️ Visual Results

ImageCritic can effectively resolve detail-related issues in various customized generation scenarios, providing significant improvements over existing methods.

<img src='https://github.com/HVision-NKU/ImageCritic/blob/master/figure/compare.png?raw=true' width='100%' />

### Online HuggingFace Demo
You can try ImageCritic demo on [HuggingFace](https://huggingface.co/spaces/ziheng1234/ImageCritic).

## Citation

If you find this project useful for your research, please consider citing our paper:
```bibtex
@article{ouyang2025consistency,
  title={The Consistency Critic: Correcting Inconsistencies in Generated Images via Reference-Guided Attentive Alignment},
  author={Ouyang, Ziheng and Song, Yiren and Liu, Yaoli and Zhu, Shihao and Hou, Qibin and Cheng, Ming-Ming and Shou, Mike Zheng},
  journal={arXiv preprint arXiv:2511.20614},
  year={2025}
}
```

## 📧 Contact
If you have any comments or questions, please [open a new issue](https://github.com/HVision-NKU/ImageCritic/issues) or contact [Ziheng Ouyang](mailto:zihengouyang666@gmail.com) 

## License
Licensed under a [Creative Commons Attribution-NonCommercial 4.0 International](https://creativecommons.org/licenses/by-nc/4.0/) for Non-commercial use only.
Any commercial use should get formal permission first.