Synthetic-NLI-VE
收藏资源简介:
本文介绍了一种新的合成数据集Synthetic-NLI-VE,用于训练视觉蕴含模型。现有的视觉蕴含数据集相对于文本蕴含数据集来说较小且稀疏。本文通过将SNLI数据集中的前提文本作为生成图像模型的输入提示,使用Stable Diffusion模型为每个文本前提创建一个图像,从而生成一个数据集。该数据集在内部和外部评估中均表现出良好的性能,证明了在数据稀缺的情况下,合成数据可以成为训练视觉蕴含模型的可行解决方案。
This paper introduces a novel synthetic dataset, Synthetic-NLI-VE, for training visual entailment models. Existing visual entailment datasets are smaller and sparser compared to textual entailment datasets. This work constructs the Synthetic-NLI-VE dataset by taking the premise texts from the SNLI dataset as input prompts for image generation models, and leveraging the Stable Diffusion model to generate an image for each textual premise. This dataset exhibits excellent performance in both internal and external evaluations, demonstrating that synthetic data can serve as a feasible solution for training visual entailment models in scenarios with scarce training data.
Synthetic-NLI-VE数据集概述
数据集来源
- 基于SNLI-VE数据集或其底层Flickr30k数据集的标题生成合成图像。
数据内容
- 每张原始SNLI-VE图像生成五张合成图像。
- 图像分辨率:512×512像素。
文件结构
- 生成图像存储在
data/文件夹中。 - 文件名前缀为原始SNLI-VE图像ID以保持可追溯性。
- 数据文件夹分为多个部分(因单文件夹推送超过10000张图像存在问题)。
注意事项
dataset_dict.py和dataset_info.json文件为占位符,尚需实现(用于定义如何通过Hugging Face Datasets库加载和描述数据集)。
适用任务
- 视觉蕴含任务。
- 多模态推理任务。
- 作为真实世界图像-文本数据集的合成对应物。

- 1Dataset Creation for Visual Entailment using Generative AI莱顿大学 · 2025年



