2018 年 9 月腾讯 AI Lab 开源 “Tencent ML-Images” 项目,该项目由多标签图像数据集 ML-Images,以及业内目前同类深度学习模型中精度最高的深度残差网络 ResNet-101 构成。 近日腾讯在 arxiv 上发布论文详细介绍了数据集的构成以及评价标准等内容,对 Tencent ML-Images 数据集的理解很有帮助。
腾讯 AI Lab 公布的图像数据集 ML-Images,包含了 1800 万图像和 1.1 万多种常见物体类别,在业内已公开的多标签图像数据集中规模最大,足以满足一般科研机构及中小企业的使用场景。
此外,腾讯 AI Lab 还提供基于 ML-Images 训练得到的深度残差网络 ResNet-101。该模型具有优异的视觉表示能力和泛化性能,在当前业内同类模型中精度最高,将为包括图像、视频等在内的视觉任务提供强大支撑,并助力图像分类、物体检测、物体跟踪、语义分割等技术水平的提升。
题目:Tencent ML-Images: A Large-Scale Multi-Label Image Database for Visual Representation Learning
作者:Baoyuan Wu, Weidong Chen, Yanbo Fan, Yong Zhang, Jinlong Hou, Junzhou Huang, Wei Liu, Tong Zhang
【摘要】
在现有的视觉表示学习任务中,深度卷积神经网络(CNN)通常是针对带有单个标签的图像进行训练的,例如 ImageNet。然而,单个标签无法描述一幅图像的所有重要内容,一些有用的视觉信息在训练过程中可能会被浪费。在这项工作中,我们建议对带有多个标签的图像进行训练,以提高训练后的 CNN 模型的视觉表示质量。
为此,我们构建了一个大规模的多标签图像数据库,其中包含 18000000 个图像和 11000 个类别,我们称之为 Tencent ML-Images。我们基于大规模分布式深度学习框架,即 TFplus,在 Tencent ML-Images 上高效训练 ResNet-101 多标签输出模型,共 60 个 epoch,耗时 90 小时。 通过 ImageNet 和 Caltech-256 上的单标签图像分类、PASCAL VOC 2007 上的对象检测、PASCAL VOC 2012 上的语义分割三个迁移学习任务,验证了 Tencent ML-Images checkpoint 的视觉表示质量良好。
腾讯 ML-Images 数据库,ResNet-101 的 checkpoint 以及所有训练代码已在https://github.com/Tencent/tencent-ml-images上发布。它有望推动研究领域和工业界的其他视觉任务的发展。
论文地址:
https://arxiv.org/abs/1901.01703
作者简介
第一作者是吴保元,现在是腾讯 AI Lab 的高级研究员,2014 年 8 月至 2016 年 11 月在 KAUST 进行博士后学习,与 Bernard Ghanem 教授一起工作。2014 年 6 月获得中国科学院自动化研究所模式识别国家重点实验室博士学位,导师为胡包钢教授。作者研究兴趣包括机器学习、计算机视觉和优化,包括图像标注、弱 / 无监督学习、结构化预测、概率图模型、视频处理和整数规划。
https://sites.google.com/site/baoyuanwu2015/home
其他作者包括樊艳波、张勇也都是博士毕业于中科院自动化研究所,师从胡包钢教授。
文章简介
这项工作在新建的多标签图像数据库(称为 Tencent ML-Images)上展示了大规模的视觉表示学习。文章从讨论以下两个问题开始。
为什么我们需要大规模的图像数据库?深度学习一直处于长期低谷,直到 2012 年,AlexNet 在 ILSVRC2012 挑战的单标签图像分类任务中取得了令人惊讶的成绩。深度神经网络的潜力是通过大规模的图像数据库释放出来的,即 ImageNet-ILSVRC2012 。此外,对于许多视觉任务,如目标检测和语义分割,获取训练数据的成本是非常高的。由于训练数据不足,需要在其他大型数据库上预先训练好的视觉呈现良好的 checkpoint 作为初始化,用于其他视觉任务 (如针对单标签图像分类的 ImageNet-ILSVRC2012)。
为什么我们需要多标签图像数据库?由于在大多数自然图像中存在多个对象,单个标注可能会遗漏一些有用的信息,从而误导 CNN 的训练。例如,同时包含牛和草的两个视觉上相似的图像可能分别被标注为牛和草。合理的方法是 “告诉”CNN 模型这两幅图像同时包含牛和草。
这项工作的主要贡献有四个方面:
建立了一个包含一千八百万张图像和一万一千个类别的多标签图像数据库,被称为 Tencent ML-Images,这是迄今为止最大的公开可用的多标签图像数据库。
利用大规模分布式深度学习框架,在 Tencent ML-Images 上有效地训练 ResNet-101 模型。此外,还设计了一种新的损失函数来缓解大规模多标签数据库中严重的类失衡问题。
我们通过迁移学习三种不同的视觉任务,证实了 Tencent ML-Images 和其预训练的检查点有着比较好的质量。
在 GitHub(https://github.com/Tencent/tencent-ml-images)上发布了 Tencent ML-Images 数据库,包含训练的 ResNet-101 检查点,以及从数据预处理,预训练,微调到图像分类和特征提取的完整代码。预计这将推动研究领域和工业界的其他视觉任务的发展。
附文章中部分结果
-
图像分类
+关注
关注
0文章
89浏览量
11897 -
数据集
+关注
关注
4文章
1200浏览量
24613
原文标题:论文终于公布!腾讯开源最大规模多标签图像数据集,刷新行业数据集基准
文章出处:【微信号:AI_era,微信公众号:新智元】欢迎添加关注!文章转载请注明出处。
发布评论请先 登录
相关推荐
评论