arXiv:2502.04378cs.CVcs.GR2025-02被引 5

用大模型和扩散模型生成逼真测试图像,提升视觉模型鲁棒性。

DILLEMA: Diffusion and Large Language Models for Multi-Modal Augmentation

  • 先用文本描述图像,再让语言模型生成反事实描述。
  • 通过控制扩散模型生成保留场景结构的新图像。
  • 在ImageNet1K和SHIFT数据集上验证有效,人类评估认可真实性。

确保深度学习模型的鲁棒性需要全面且多样的测试。现有方法多依赖简单的数据增强或生成对抗网络,难以生成真实且多样化的测试样本。为此,我们提出一种新型框架,利用大语言模型(LLM)和控制条件扩散模型(Diffusion Models)生成高保真度的合成测试用例。该方法首先通过图像描述模型将图像转换为详细文本描述,使语言模型识别可修改部分并生成反事实描述;随后,基于这些描述通过文本到图像扩散过程生成新图像,保持空间一致性并保留场景关键元素。我们在ImageNet1K(图像分类)和SHIFT(自动驾驶语义分割)两个数据集上验证了该方法的有效性。结果表明,该方法能生成显著的测试案例,通过针对性重训练揭示模型弱点并提升鲁棒性。我们通过Mechanical Turk进行人工评估,参与者高度一致认为生成图像真实有效。

原文摘要 · Abstract (English)

Ensuring the robustness of deep learning models requires comprehensive and diverse testing. Existing approaches, often based on simple data augmentation techniques or generative adversarial networks, are limited in producing realistic and varied test cases. To address these limitations, we present a novel framework for testing vision neural networks that leverages Large Language Models and control-conditioned Diffusion Models to generate synthetic, high-fidelity test cases. Our approach begins by translating images into detailed textual descriptions using a captioning model, allowing the language model to identify modifiable aspects of the image and generate counterfactual descriptions. These descriptions are then used to produce new test images through a text-to-image diffusion process that preserves spatial consistency and maintains the critical elements of the scene. We demonstrate the effectiveness of our method using two datasets: ImageNet1K for image classification and SHIFT for semantic segmentation in autonomous driving. The results show that our approach can generate significant test cases that reveal weaknesses and improve the robustness of the model through targeted retraining. We conducted a human assessment using Mechanical Turk to validate the generated images. The responses from the participants confirmed, with high agreement among the voters, that our approach produces valid and realistic images.

多模态扩散模型数据增强鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。