返回资源中心

LAION-5B Open Dataset

数据集
机器学习
10 次浏览
1 个赞
datasetvisionmultimodal

资源描述

LAION-5B 是由 LAION 组织开源的超大规模图文对数据集,包含约 58.5 亿条经过 CLIP 模型过滤的图像-文本链接。该数据集源自 Common Crawl 网页抓取数据,为 Stable Diffusion 等主流文生图模型及多模态大模型提供了核心预训练语料。适用于视觉生成、跨模态检索、VLM 预训练等研究场景,完全免费开放,大幅降低了大规模多模态模型的训练门槛。

详细内容

## 数据集背景与来源 LAION-5B(Large-scale Artificial Intelligence Open Network)由德国非营利研究组织 LAION 构建并发布。面对多模态大模型训练对海量数据的渴求,团队从 Common Crawl 公开网页快照中提取原始图文数据,并创新性地引入 OpenAI 的 CLIP 模型进行大规模语义对齐过滤。该项目旨在打破商业公司对高质量训练数据的垄断,为学术界和开源社区提供完全免费、可复现的基础多模态语料。 ## 数据规模与标注信息 - **总体规模**:包含约 58.5 亿个图像-文本对,按语言划分为三个核心子集:`laion2B-en`(23.2亿英文)、`laion2B-multi`(22.6亿多语言)和 `laion1B-nolang`(12.7亿未知语言)。 - **数据格式**:数据集**不直接托管图片文件**,而是提供结构化的 Parquet 元数据索引。字段包含图片 URL、网页 Alt-text/描述文本、CLIP 相似度分数、NSFW 概率、水印概率、美学评分及图像宽高比等。 - **标注与过滤**:采用 CLIP ViT-B/32 模型计算图文相似度,英文子集阈值设定为 >0.28,多语言子集 >0.26。同时结合美学评分与水检测模型,有效剔除了低质、无关或过度商业化的样本,确保语义对齐质量。 ## 典型应用场景 1. **文生图/图生文模型预训练**:作为 Stable Diffusion、Kandinsky 等主流扩散模型(Diffusion Models)的核心训练基座。 2. **多模态表征学习**:用于训练或微调 CLIP、ALIGN、OpenCLIP 等视觉-语言双塔对齐模型。 3. **跨模态检索与零样本任务**:支持大规模图文互搜、零样本图像分类及视觉问答(VQA)基准测试。 4. **数据工程与算法研究**:为数据清洗策略、去重算法、大规模分布式训练基础设施提供标准测试床。 ## 使用注意事项 - **下载与存储**:仅提供 URL 索引,需自行编写爬虫或使用 `img2dataset` 等工具下载。完整图片集体积可达数百 TB,建议优先使用 Hugging Face / AWS 提供的 Parquet 元数据快照或按需抽样下载。 - **链接失效处理**:网页数据具有时效性,部分 URL 可能已失效(Dead Links)。实际下载时需配置重试机制与超时过滤,通常有效下载率在 70%-80% 左右。 - **合规与安全**:尽管官方提供了 NSFW 过滤标签,但原始网页数据仍可能包含敏感、偏见或版权争议内容。用于商业产品或公开发布前,务必进行二次内容安全审查与版权合规评估。 - **引用规范**:遵循开放数据协议,学术或工程使用时请引用 LAION-5B 官方论文(NeurIPS 2022 Datasets and Benchmarks Track),并注明数据来源为 Common Crawl。