five

fschieber/wit-captions

收藏
Hugging Face2024-03-15 更新2024-06-11 收录
下载链接:
https://hf-mirror.com/datasets/fschieber/wit-captions
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是从原始的WIT数据集派生而来的,经过了一系列的修改。首先,删除了除image_url、caption_reference_description、caption_attribution_description、mime_type、original_height、original_width之外的所有列。然后,移除了所有没有caption_reference_description或caption_attribution_description的行。接着,对image_url进行了去重处理,保留了具有最长caption_reference_description的条目。新增了一个text列,该列包含caption_reference_description和caption_attribution_description中最长的描述。最后,数据集被分割成8个文件,总共包含11,419,525行。该数据集的目的是用于生成一个新的数据集,其中包含text列的嵌入向量。

该数据集是从原始的WIT数据集派生而来的,经过了一系列的修改。首先,删除了除image_url、caption_reference_description、caption_attribution_description、mime_type、original_height、original_width之外的所有列。然后,移除了所有没有caption_reference_description或caption_attribution_description的行。接着,对image_url进行了去重处理,保留了具有最长caption_reference_description的条目。新增了一个text列,该列包含caption_reference_description和caption_attribution_description中最长的描述。最后,数据集被分割成8个文件,总共包含11,419,525行。该数据集的目的是用于生成一个新的数据集,其中包含text列的嵌入向量。
提供机构:
fschieber
原始信息汇总

数据集概述

数据集来源与处理

  • 该数据集源自原始的WIT数据集,经过以下处理:
    • 保留了image_url, caption_reference_description, caption_attribution_description, mime_type, original_height, original_width列,移除了其他所有列。
    • 移除了没有caption_reference_description或caption_attribution_description的行。
    • 根据image_url去重,保留了具有最长caption_reference_description的记录。
    • 新增了text列,该列包含caption_reference_description和caption_attribution_description中较长的描述文本。
      • 注意:caption_attribution_description可能包含多种语言,使用简单正则表达式解析并选择最长语言描述,可能不完全准确或有遗漏。

数据集结构

  • 数据集最终被分割成8个文件。
  • 所有8个文件总计包含11,419,525行数据。

数据集用途

  • 该数据集旨在用于生成text列的嵌入向量。
搜集汇总
数据集介绍
main_image_url
构建方式
在多媒体信息检索与跨模态学习领域,图文配对数据的质量直接影响模型的表达能力。fschieber/wit-captions数据集源自原始WIT数据集,经过一系列精细的过滤与重构流程构建而成。首先,保留图像链接、参考描述、归属描述、MIME类型、原始高度与宽度等关键字段,剔除其余冗余列。其次,移除所有缺失参考描述或归属描述的行,确保每条记录均具备语义信息。随后,基于图像链接进行去重处理,优先保留参考描述最长的条目,以提升数据丰富度。最后,新增一个文本列,通过正则表达式从归属描述中提取并选取最长文本,最终将数据集划分为8个文件,共计约1142万条记录。
特点
该数据集的核心特点在于其专注于图文对齐任务中的文本质量优化。通过去除无描述样本与重复图像,数据集的噪声显著降低,每条记录均包含至少一个高质量的自然语言描述。新增的文本列整合了参考描述与归属描述中的最长片段,兼顾了多语言场景下的信息完整性。尽管正则解析可能遗留少量语言识别误差,但整体上为跨模态嵌入向量的生成提供了标准化且规模适中的基础语料。数据集以图像链接为唯一标识,支持高效的外链图像检索,便于下游模型直接利用。
使用方法
该数据集专为生成文本嵌入向量而设计,适用于训练图文匹配、图像描述生成或多模态对比学习模型。使用时,可通过HuggingFace的datasets库直接加载各分片文件,利用图像链接下载对应图片,并与文本列配对构建训练样本。建议在预处理阶段对文本进行语言标准化或去噪处理,以提升嵌入质量。数据集可直接用于计算文本特征的向量表示,或作为预训练阶段的数据源,进一步扩展至视觉-语言联合任务。
背景与挑战
背景概述
在视觉与语言交叉领域,图像描述生成任务长期受限于多语言、多模态数据的稀缺性与不均衡性。2019年,谷歌研究团队推出了Wikipedia-based Image Text(WIT)数据集,旨在通过整合维基百科中丰富的图像-文本对,推动跨语言、跨领域的多模态理解研究。该数据集由多位来自谷歌的研究人员主导,核心研究问题聚焦于如何从大规模、多源异构的网页数据中提取高质量的图像描述,以支撑多语言图像描述、视觉问答等下游任务。fschieber/wit-captions作为WIT的精炼版本,通过剔除冗余列、去重图像引用并统一文本字段,进一步优化了数据质量,为后续生成文本嵌入向量提供了更简洁、高效的基准。这一数据集不仅降低了多模态研究的预处理门槛,还通过保留最长描述的策略,强化了语义信息的完整性,对低资源语言的图像理解研究具有重要推动作用。
当前挑战
该数据集面临的挑战主要体现在两个层面。在领域问题层面,尽管WIT原始数据覆盖数百种语言,但图像描述的质量与语言分布极不均衡——英语描述占据主导,而许多小语种描述存在噪声或缺失,这导致多语言图像描述模型的泛化能力受限,尤其在低资源语言场景下表现不佳。在构建过程中,数据清洗面临显著困难:原始caption_attribution_description字段常混杂多种语言,虽采用正则表达式提取最长描述,但该策略可能误判语言边界,遗留包含多语言混合的“伪影”文本;同时,基于图像URL的去重操作仅保留最长描述,可能丢弃其他有效语义变体,造成信息损失。此外,11.4百万条数据分拆为8个文件,虽便于分布式处理,却增加了跨文件检索与整合的复杂度,对下游任务的批处理效率构成隐性制约。
常用场景
经典使用场景
在跨模态智能研究领域,fschieber/wit-captions数据集作为维基百科图像文本描述的精炼版本,被广泛用于训练和评估图像描述生成模型。其核心价值在于提供了经过去重和文本筛选的高质量图文对,研究者可将其作为基准数据集,开发能够从图像中自动生成自然语言描述的深度学习架构,如基于Transformer的编码器-解码器模型,从而推动视觉与语言理解技术的协同发展。
实际应用
在实际部署中,该数据集支撑了搜索引擎中的图像检索优化、辅助视障人士的自动场景叙述系统以及社交媒体内容自动摘要工具。例如,基于其训练的模型可用于自动生成电商产品图片的详细说明,或为新闻配图提供多语言描述,显著提升信息无障碍获取效率。其精简结构也便于边缘设备上的轻量化部署,降低计算资源消耗。
衍生相关工作
围绕该数据集衍生了一系列经典工作,包括基于预训练视觉语言模型(如CLIP、BLIP)的微调方法,以及针对长尾概念描述的生成对抗网络改进。研究者还利用其嵌入向量生成特性,开发了跨模态检索的对比学习框架,并催生了如WIT-Retrieval等下游任务基准。此外,针对多语言描述解析的启发,相关研究进一步优化了正则表达式与语言检测算法,推动了多模态数据清洗工具的标准化。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务