用大模型提升AI生成图缺陷检测的可解释性与精度
HEIE: MLLM-Based Hierarchical Explainable AIGC Image Implausibility Evaluator
- 通过分步推理分解复杂缺陷判断任务,增强结果可解释性
- 实现像素级缺陷热图生成,定位精度优于现有方法
- 适合需要可信、可解释图像质量评估的AI研发人员
AIGC图像在各领域广泛应用,但常存在伪影和不自然纹理等质量问题。专用模型虽能预测缺陷热图,却缺乏可解释性且难以利用常识与逻辑推理,泛化能力差。多模态大模型(MLLM)具备更强理解与推理能力,但在细粒度缺陷定位和生成像素级输出方面受限。为此,我们提出HEIE:一种基于MLLM的分层可解释图像不合理性评估器。引入基于思维链(CoT)的可解释三重评估机制,整合热图、评分与解释输出,通过分步推理提升可解释性。设计自适应分层不合理性映射器,融合低层图像特征与大模型高层映射标记,采用基于不确定性的自适应标记策略,实现从局部到全局的精确热图预测。此外,构建新数据集Expl-AIGI-Eval,支持可解释的AIGC图像不合理性评估。大量实验表明,该方法达到当前最优性能。
原文摘要 · Abstract (English)
AIGC images are prevalent across various fields, yet they frequently suffer from quality issues like artifacts and unnatural textures. Specialized models aim to predict defect region heatmaps but face two primary challenges: (1) lack of explainability, failing to provide reasons and analyses for subtle defects, and (2) inability to leverage common sense and logical reasoning, leading to poor generalization. Multimodal large language models (MLLMs) promise better comprehension and reasoning but face their own challenges: (1) difficulty in fine-grained defect localization due to the limitations in capturing tiny details, and (2) constraints in providing pixel-wise outputs necessary for precise heatmap generation. To address these challenges, we propose HEIE: a novel MLLM-Based Hierarchical Explainable Image Implausibility Evaluator. We introduce the CoT-Driven Explainable Trinity Evaluator, which integrates heatmaps, scores, and explanation outputs, using CoT to decompose complex tasks into subtasks of increasing difficulty and enhance interpretability. Our Adaptive Hierarchical Implausibility Mapper synergizes low-level image features with high-level mapper tokens from LLMs, enabling precise local-to-global hierarchical heatmap predictions through an uncertainty-based adaptive token approach. Moreover, we propose a new dataset: Expl-AIGI-Eval, designed to facilitate interpretable implausibility evaluation of AIGC images. Our method demonstrates state-of-the-art performance through extensive experiments. Our project is at https://yfthu.github.io/HEIE/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。