机器学习模型开发必读：开源数据库最全盘点
机器学习模型开发必读：开源数据库最全盘点
开发 AI 和机器学习系统从来没有像现在这样方便。类似于 TensorFlow、Torch 和 Spark 这样的开源工具，在 AI 开发者群体中已是无处不在。再加上亚马逊 AWS、Google Cloud 等云服务带来的海量计算能力，将来使用笔记本电脑来训练 ML 模型或许不再难以想象。
公众对 AI 的遐想，总忽视了数据的角色。但海量被标记、注解过的数据，是当下 AI 革命当之无愧的主要推手之一。业内研究团队和公司机构，均明白“数据民主化”的意义——使任何开发者都能获取高质量的数据来训练、测试模型，是加速 AI 技术发展的必要措施。
但就雷锋网所知，大多数涉及机器学习和 AI 的产品依赖于专有数据库（ proprietary datasets）。它们大多是不被公开的，以保护知识产权以及防范安全风险。
即便你幸运地找到了相关公共数据库，判断后者的价值和可靠程度，又是一项让很多开发者头痛的问题。对于概念论证是如此；对于潜在的产品或者特性验证同样如此——在收集你的专有数据之前，决定该验证需要何种数据集。
有经验的开发者都知道，机器学习系统在样本数据集上展示出的优异性能，并不能保证其实际效果。许多 AI 从业人员似乎已经忘记了，数据采集和标记才是开发 AI 解决方案最难的一环。标准的数据集，可被用作验证集，或作为开发更偏向私人订制方案的起始点。
本周，Vai Technologies 的创始人、前斯坦福 SLAC 实验室 CNN 算法架构师 Luke de Oliveira，和其他几名机器学习专家谈到了这个问题。雷锋网了解到，他们最后决定做一张表单，把 AI 领域含金量最高的开源数据库罗列出来，与大家分享。
计算机视觉
MNIST
标签：学术基准 经典 较旧
合理性测试（sanity check）最常用的数据库。规格为 25x25、中心的、B&W 手写数字。用 MNIST 测试非常容易，但不要因为你的模型在 MNIST 运行良好，就认为它事实上可用。
地址：https://pjreddie.com/projects/mnist-in-csv/
CIFAR 10 & CIFAR 100
标签：经典 较旧
32x32 彩色图像。虽然用得人比以前少了很多，但仍然能用它做有趣的合理性测试。
地址：https://www.cs.toronto.edu/~kriz/cifar.html
ImageNet
标签：实用 学术基准 经典
这个用不着介绍，新算法的首选图像数据集。Luke de Oliveira 表示，许多图像 API 公司从 REST 交互界面搞来的标记，与 ImageNet 1000 目录中的 WordNet 层级很接近，让人怀疑。
地址：http://image-net.org/
LSUN
标签：无
场景理解，许多其它附加任务（比如房间布局预估，显着性预测 “saliency prediction”），以及与之关联的竞赛。
地址：http://lsun.cs.princeton.edu/2016/
PASCAL VOC
标签：学术基准
一般性的图像分割和分类。对于创建现实世界中的图像注解并不是十分有用，但作为基准很不错。
地址：http://host.robots.ox.ac.uk/pascal/VOC/
SVHN
标签：学术基准
谷歌街景视图中的住宅号。可以把它当做野生的递归（ recurrent） MNIST。
地址：http://ufldl.stanford.edu/housenumbers/
MS COCO
标签：无
一般性的图像理解/说明，有相关竞赛。
地址：http://mscoco.org/
Visual Genome
标签：实用
非常细致的视觉知识库，对超过十万张图像有深度注解。
地址：http://visualgenome.org/
Labeled Faces in the Wild
标签：实用 学术基准 经典 较旧
修剪过的面部区域（使用 Viola-Jones），用一个 name identifier 做过标记。其中每一个展示的人在数据集中有两个图像，这是作为他的子集。开发者经常用它来训练面部匹配系统。
地址：http://vis-www.cs.umass.edu/lfw/
自然语言
Text Classification Datasets
标签：实用 学术基准
来自论文 Zhang et al., 2015。这是有八个文字分类数据集组成的大型数据库。对于新的文字分类基准，它是最常用的。样本大小为 120K 到 3.6M，包括了从二元到 14 阶的问题。来自 DBPedia, Amazon, Yelp, Yahoo!，搜狗和 AG 的数据集。
地址：https://drive.google.com/drive/u/0/folders/0Bz8a_Dbh9Qhbfll6bVpmNUtUcFdjYmF2SEpmZUZUcVNiMUw1TWN6RDV3a0JHT3kxLVhVR2M
WikiText
标签：实用 学术基准
源自高品质维基百科文章的大型语言建模语料库。Salesforce MetaMind 维护。
地址：http://metamind.io/research/the-wikitext-long-term-dependency-language-modeling-dataset/
Question Pairs
标签：实用
Quora 发布的第一个数据集，包含副本/语义近似值标记。
地址：https://data.quora.com/First-Quora-Dataset-Release-Question-Pairs
SQuAD
标签：实用 学术基准
斯坦福的问答社区数据集——适用范围较广的问题回答和阅读理解数据集。每一个回答都被作为一个 span，或者一段文本。
地址：https://rajpurkar.github.io/SQuAD-explorer/
CMU Q/A Dataset
标签：无
人工创建的仿真陈述问题/回答组合，还有维基百科文章的难度评分。
地址：http://www.cs.cmu.edu/~ark/QA-data/
Maluuba Datasets
标签：实用
为 NLP 研究人工创建的复杂数据集。
地址：https://datasets.maluuba.com/
Billion Words
标签：实用 学术基准
大型、通用型建模数据集。时常用来训练散布音（distributed）的词语表达，比如 word2vec 或 GloVe。
地址：http://www.statmt.org/lm-benchmark/
Common Crawl
标签：实用 学术基准
PB（拍字节）级别的网络爬虫。最经常被用来学习词语嵌入。可从 Amazon S3 免费获取。对于 WWW 万维网的信息采集，是一个比较有用的网络数据集。
地址：http://commoncrawl.org/the-data/
bAbi
标签：学术基准 经典
Facebook AI Research (FAIR) 推出的合成阅读理解和问题回答数据集。
地址：https://research.fb.com/projects/babi/
The Children's Book Test
标签：学术基准
Project Gutenberg（一项正版数字图书免费分享工程）儿童图书里提取的成对数据（问题加情境，回答）基准。对问答、阅读理解、仿真陈述（factoid）查询比较有用。
地址：https://research.fb.com/projects/babi/
Stanford Sentiment Treebank
标签：学术基准 经典 较旧
标准的情绪数据集，对每一句话每一个节点的语法树，都有细致的情感注解。
地址：http://nlp.stanford.edu/sentiment/code.html
20 Newsgroups
标签：经典 较旧
一个较经典的文本分类数据集。通常作为纯粹分类或者对 IR / indexing 算法验证的基准，在这方面比较有用。
地址：http://qwone.com/~jason/20Newsgroups/
Reuters
标签：经典 较旧
较老的、基于纯粹分类的数据集。文本来自于路透社新闻专线。常被用于教程之中。
地址：https://archive.ics.uci.edu/ml/datasets/Reuters-21578+Text+Categorization+Collection
IMDB
标签：经典 较旧
较老的、相对比较小的数据集。用于情绪分类。但在文学基准方面逐渐失宠，让位于更大的数据集。
地址：http://ai.stanford.edu/~amaas/data/sentiment/
UCI’s Spambase
标签：经典 较旧
较老的、经典垃圾邮件数据集，源自于 UCI Machine Learning Repository。由于数据集的管理细节，在学习私人订制垃圾信息过滤方面，这会是一个有趣的基准。
地址：https://archive.ics.uci.edu/ml/datasets/Spambase
语音
大多数语音识别数据库都是专有的——这些数据对其所有公司而言有巨大价值。绝大部分该领域的公共数据集已经很老了。
2000 HUB5 English
标签：学术基准 较旧
只包含英语的语音数据。最近一次被使用是百度的深度语音论文。
地址：https://catalog.ldc.upenn.edu/LDC2002T43
LibriSpeech
标签：学术基准
有声图书数据集，包含文字和语音。接近 500 个小时的清楚语音，来自于多名朗读者和多个有声读物，根据图书章节来组织。
地址：http://www.openslr.org/12/
VoxForge
标签：实用 学术基准
带口音英语的清晰语音数据集。如果你需要有强大的不同口音、语调识别能力，会比较有用。
地址：http://www.voxforge.org/
TIMIT
标签：学术基准 经典
只含英语的语音识别数据集。
地址：https://catalog.ldc.upenn.edu/LDC93S1
CHIME
标签：实用
含大量噪音的语音识别挑战杯数据集。它包含真实、模拟和清晰的录音：真实，是因为该数据集包含四个说话对象在四个不同吵闹环境下接近 9000 段的录音；模拟，是通过把多个环境与语音结合来生成；清晰，是指没有噪音的清楚录音。
地址：http://spandh.dcs.shef.ac.uk/chime_challenge/data.html
TED-LIUM
标签：无
TED 演讲的音频转录。包含 1495 场 TED 演讲，以及它们的完整字幕文本。
地址：http://www-lium.univ-lemans.fr/en/content/ted-lium-corpus
推荐和排名系统
Netflix Challenge
标签：经典 较旧
第一个 Kaggle 模式的数据挑战。由于盗版问题只能获得非官方版本。
地址：http://www.netflixprize.com/
MovieLens
标签：实用 学术基准 经典
不同大小的电影点评数据——一般作为协同过滤的基准。
地址：https://grouplens.org/datasets/movielens/
Million Song Dataset
标签：无
Kaggle 上的大型、富含原数据的开源数据集。对于试验混合推荐系统有价值。
地址：https://www.kaggle.com/c/msdchallenge
Last.fm
标签：实用
音乐推荐数据集，并关联相关社交网络和其他元数据。对混合系统有用处。
地址：http://grouplens.org/datasets/hetrec-2011/
网络和图
Amazon Co-Purchasing 和 Amazon Reviews
标签：学术基准
从亚马逊“买了这个的用户还买了XXX”功能抓取的数据，还有相关商品的评价数据。对于试验网络中的推荐系统有价值。
地址：http://snap.stanford.edu/data/#amazon
http://snap.stanford.edu/data/amazon-meta.html
Friendster 社交网络数据集
标签：无
在成为游戏网站之前，Friendster 发布了 103,750,348 名用户朋友名单的匿名数据。
地址：https://archive.org/details/friendster-dataset-201107
地理空间数据
OpenStreetMap
标签：实用
整个地球的矢量数据，处于免费协议下。它的旧版本包含美国人口统计部门的 TIGER 数据。
地址：http://wiki.openstreetmap.org/wiki/Planet.osm
Landsat8
标签：实用
整个地球表面的卫星拍照，每隔几周更新一次。
地址：https://landsat.usgs.gov/landsat-8
NEXRAD
标签：实用
多普勒天气雷达对美国大气情况的扫描。
地址：https://www.ncdc.noaa.gov/data-access/radar-data/nexrad
后话
人们经常认为，能在一个数据集上解决问题，就等同于有了一个能用的产品。开发者可以使用这些数据集作为验证集，或用作概念论证；但别忘了测试，或创建模拟产品运行的原型机。获取更新、更真实的数据来改善模型非常关键。雷锋网了解到，成功的数据驱动型公司，往往擅长收集新的专有数据，以及改善产品性能增强竞争优势。而这往往是竞争对手难以直接 copy 的。
原网站：http://www.toutiao.com/i6390620919512957441/