PaddlePaddle Models
PaddlePaddle provides a rich set of computational units to enable users to adopt a modular approach to solving various learning problems. In this Repo, we demonstrate how to use PaddlePaddle to solve common machine learning tasks, providing several different neural network model that anyone can easily learn and use.
PaddlePaddle 提供了丰富的计算单元,使得用户可以采用模块化的方法解决各种学习问题。在此Repo中,我们展示了如何用 PaddlePaddle 来解决常见的机器学习任务,提供若干种不同的易学易用的神经网络模型。PaddlePaddle用户现可申请AI Studio平台的免费Tesla V100算力资源,高效在线训练模型,点击申请。
目录
PaddleCV
图像分类
图像分类是根据图像的语义信息对不同类别图像进行区分,是计算机视觉中重要的基础问题,是物体检测、图像分割、物体跟踪、行为分析、人脸识别等其他高层视觉任务的基础,在许多领域都有着广泛的应用。如:安防领域的人脸识别和智能视频分析等,交通领域的交通场景识别,互联网领域基于内容的图像检索和相册自动归类,医学领域的图像识别等。
模型名称 | 模型简介 | 数据集 | 评估指标 top-1/top-5 accuracy(CV2) |
---|---|---|---|
AlexNet | 首次在CNN中成功的应用了ReLU、Dropout和LRN,并使用GPU进行运算加速 | ImageNet-2012验证集 | 56.72%/79.17% |
VGG | 在AlexNet的基础上使用3*3小卷积核,增加网络深度,具有很好的泛化能力 | ImageNet-2012验证集 | 72.56%/90.93% |
GoogleNet | 在不增加计算负载的前提下增加了网络的深度和宽度,性能更加优越 | ImageNet-2012验证集 | 70.70%/89.66% |
ResNet | Residual Network,引入了新的残差结构,解决了随着网络加深,准确率下降的问题 | ImageNet-2012验证集 | 80.93%/95.33% |
ResNet-D | 融合最新多种对ResNet改进策略,ResNet50_vd的top1准确率达到79.84% | ImageNet-2012验证集 | 79.84%/94.93% |
Inception-v4 | 将Inception模块与Residual Connection进行结合,通过ResNet的结构极大地加速训练并获得性能的提升 | ImageNet-2012验证集 | 80.77%/95.26% |
MobileNet v1 | 将传统的卷积结构改造成两层卷积结构的网络,在基本不影响准确率的前提下大大减少计算时间,更适合移动端和嵌入式视觉应用 | ImageNet-2012验证集 | 70.99%/89.68% |
MobileNet v2 | MobileNet结构的微调,直接在thinner的bottleneck层上进行skip learning连接以及对bottleneck layer不进行ReLu非线性处理可取得更好的结果 | ImageNet-2012验证集 | 72.15%/90.65% |
SE_ResNeXt | 在ResNeXt 基础、上加入了SE(Sequeeze-and-Excitation) 模块,提高了识别准确率,在ILSVRC 2017 的分类项目中取得了第一名 | ImageNet-2012验证集 | 81.40%/95.48% |
ShuffleNet v2 | ECCV2018,轻量级CNN网络,在速度和准确度之间做了很好地平衡。在同等复杂度下,比ShuffleNet和MobileNetv2更准确,更适合移动端以及无人车领域 | ImageNet-2012验证集 | 70.03%/89.17% |
目标检测
目标检测任务的目标是给定一张图像或是一个视频帧,让计算机找出其中所有目标的位置,并给出每个目标的具体类别。对于计算机而言,能够“看到”的是图像被编码之后的数字,但很难解图像或是视频帧中出现了人或是物体这样的高层语义概念,也就更加难以定位目标出现在图像中哪个区域。
模型名称 | 模型简介 | 数据集 | 评估指标 mAP |
---|---|---|---|
SSD | 很好的继承了MobileNet预测速度快,易于部署的特点,能够很好的在多种设备上完成图像目标检测任务 | VOC07 test | mAP = 73.32% |
Faster-RCNN | 创造性地采用卷积网络自行产生建议框,并且和目标检测网络共享卷积网络,建议框数目减少,质量提高 | MS-COCO | 基于ResNet 50 mAP(0.50:0.95) = 36.7% |
Mask-RCNN | 经典的两阶段框架,在Faster R-CNN模型基础上添加分割分支,得到掩码结果,实现了掩码和类别预测关系的解藕,可得到像素级别的检测结果。 | MS-COCO | 基于ResNet 50 Mask mAP(0.50:0.95) = 31.4% |
RetinaNet | 经典的一阶段框架,由ResNet主干网络、FPN结构、和两个分别用于回归物体位置和预测物体类别的子网络组成。在训练过程中使用Focal Loss,解决了传统一阶段检测器存在前景背景类别不平衡的问题,进一步提高了一阶段检测器的精度。 | MS-COCO | 基于ResNet mAP (500.50:0.95) = 36% |
YOLOv3 | 速度和精度均衡的目标检测网络,相比于原作者darknet中的YOLO v3实现,PaddlePaddle实现参考了论文Bag of Tricks for Image Classification with Convolutional Neural Networks 增加了mixup,label_smooth等处理,精度(mAP(0.5:0.95))相比于原作者提高了4.7个绝对百分点,在此基础上加入synchronize batch normalization, 最终精度相比原作者提高5.9个绝对百分点。 | MS-COCO | 基于DarkNet mAP(0.50:0.95)= 38.9% |
PyramidBox | PyramidBox 模型是百度自主研发的人脸检测模型,利用上下文信息解决困难人脸的检测问题,网络表达能力高,鲁棒性强。于18年3月份在WIDER Face数据集上取得第一名 | WIDER FACE | mAP (Easy/Medium/Hard set)= 96.0%/ 94.8%/ 88.8% |
图像分割
图像语义分割顾名思义是将图像像素按照表达的语义含义的不同进行分组/分割,图像语义是指对图像内容的理解,例如,能够描绘出什么物体在哪里做了什么事情等,分割是指对图片中的每个像素点进行标注,标注属于哪一类别。近年来用在无人车驾驶技术中分割街景来避让行人和车辆、医疗影像分析中辅助诊断等。
模型名称 | 模型简介 | 数据集 | 评估指标 |
---|---|---|---|
ICNet | 主要用于图像实时语义分割,能够兼顾速度和准确性,易于线上部署 | Cityscape | Mean IoU=67.0% |
DeepLab V3+ | 通过encoder-decoder进行多尺度信息的融合,同时保留了原来的空洞卷积和ASSP层, 其骨干网络使用了Xception模型,提高了语义分割的健壮性和运行速率 | Cityscape | Mean IoU=78.81% |
关键点检测
人体骨骼关键点检测,Pose Estimation,主要检测人体的一些关键点,如关节,五官等,通过关键点描述人体骨骼信息。人体骨骼关键点检测对于描述人体姿态,预测人体行为至关重要。是诸多计算机视觉任务的基础,例如动作分类,异常行为检测,以及自动驾驶等等。
模型名称 | 模型简介 | 数据集 | 评估指标 |
---|---|---|---|
Simple Baselines | coco2018关键点检测项目亚军方案,网络结构非常简单,效果达到state of the art | COCO val2017 | AP = 72.7% |
图像生成
图像生成是指根据输入向量,生成目标图像。这里的输入向量可以是随机的噪声或用户指定的条件向量。具体的应用场景有:手写体生成、人脸合成、风格迁移、图像修复等。
模型名称 | 模型简介 | 数据集 |
---|---|---|
CGAN | 条件生成对抗网络,一种带条件约束的GAN,使用额外信息对模型增加条件,可以指导数据生成过程 | Mnist |
DCGAN | 深度卷积生成对抗网络,将GAN和卷积网络结合起来,以解决GAN训练不稳定的问题 | Mnist |
Pix2Pix | 图像翻译,通过成对图片将某一类图片转换成另外一类图片,可用于风格迁移 | Cityscapes |
CycleGAN | 图像翻译,可以通过非成对的图片将某一类图片转换成另外一类图片,可用于风格迁移 | Cityscapes |
StarGAN | 多领域属性迁移,引入辅助分类帮助单个判别器判断多个属性,可用于人脸属性转换 | Celeba |
AttGAN | 利用分类损失和重构损失来保证改变特定的属性,可用于人脸特定属性转换 | Celeba |
STGAN | 人脸特定属性转换,只输入有变化的标签,引入GRU结构,更好的选择变化的属性 | Celeba |
场景文字识别
场景文字识别是在图像背景复杂、分辨率低下、字体多样、分布随意等情况下,将图像信息转化为文字序列的过程,可认为是一种特别的翻译过程:将图像输入翻译为自然语言输出。
模型名称 | 模型简介 | 数据集 | 评估指标 |
---|---|---|---|
CRNN-CTC | 使用CTC model识别图片中单行英文字符,用于端到端的文本行图片识别方法 | 单行不定长的英文字符串图片 | 错误率= 22.3% |
OCR Attention | 使用attention 识别图片中单行英文字符,用于端到端的自然场景文本识别, | 单行不定长的英文字符串图片 | 错误率 = 15.8% |
度量学习
度量学习也称作距离度量学习、相似度学习,通过学习对象之间的距离,度量学习能够用于分析对象时间的关联、比较关系,在实际问题中应用较为广泛,可应用于辅助分类、聚类问题,也广泛用于图像检索、人脸识别等领域。
模型名称 | 模型简介 | 数据集 | 评估指标 Recall@Rank-1(使用arcmargin训练) |
---|---|---|---|
ResNet50未微调 | 使用arcmargin loss训练的特征模型 | Stanford Online Product(SOP) | 78.11% |
ResNet50使用triplet微调 | 在arcmargin loss基础上,使用triplet loss微调的特征模型 | Stanford Online Product(SOP) | 79.21% |
ResNet50使用quadruplet微调 | 在arcmargin loss基础上,使用quadruplet loss微调的特征模型 | Stanford Online Product(SOP) | 79.59% |
ResNet50使用eml微调 | 在arcmargin loss基础上,使用eml loss微调的特征模型 | Stanford Online Product(SOP) | 80.11% |
ResNet50使用npairs微调 | 在arcmargin loss基础上,使用npairs loss微调的特征模型 | Stanford Online Product(SOP) | 79.81% |
视频分类和动作定位
视频分类和动作定位是视频理解任务的基础。视频数据包含语音、图像等多种信息,因此理解视频任务不仅需要处理语音和图像,还需要提取视频帧时间序列中的上下文信息。视频分类模型提供了提取全局时序特征的方法,主要方式有卷积神经网络(C3D,I3D,C2D等),神经网络和传统图像算法结合(VLAD等),循环神经网络等建模方法。视频动作定位模型需要同时识别视频动作的类别和起止时间点,通常采用类似于图像目标检测中的算法在时间维度上进行建模。
模型名称 | 模型简介 | 数据集 | 评估指标 |
---|---|---|---|
TSN | ECCV'16提出的基于2D-CNN经典解决方案 | Kinetics-400 | Top-1 = 67% |
Non-Local | 视频非局部关联建模模型 | Kinetics-400 | Top-1 = 74% |
stNet | AAAI'19提出的视频联合时空建模方法 | Kinetics-400 | Top-1 = 69% |
TSM | 基于时序移位的简单高效视频时空建模方法 | Kinetics-400 | Top-1 = 70% |
Attention LSTM | 常用模型,速度快精度高 | Youtube-8M | GAP = 86% |
Attention Cluster | CVPR'18提出的视频多模态特征注意力聚簇融合方法 | Youtube-8M | GAP = 84% |
NeXtVlad | 2nd-Youtube-8M最优单模型 | Youtube-8M | GAP = 87% |
C-TCN | 2018年ActivityNet夺冠方案 | ActivityNet1.3 | MAP=31% |
PaddleNLP
基础模型
词法分析
LAC(Lexical Analysis of Chinese)百度自主研发中文特色模型词法分析任务,**输入是一个字符串,而输出是句子中的词边界和词性、实体类别。
模型 | Precision | Recall | F1-score |
---|---|---|---|
Lexical Analysis | 88.0% | 88.7% | 88.4% |
BERT finetuned | 90.2% | 90.4% | 90.3% |
ERNIE finetuned | 92.0% | 92.0% | 92.0% |
语言模型
基于LSTM的语言模型任务,给定一个输入词序列(中文分词、英文tokenize),计算其PPL(语言模型困惑度,用户表示句子的流利程度)。
large config | train | valid | test |
---|---|---|---|
paddle | 37.221 | 82.358 | 78.137 |
tensorflow | 38.342 | 82.311 | 78.121 |
文本理解
情感分析
Senta(Sentiment Classification)百度AI开放平台中情感倾向分析模型、百度自主研发的中文特色模型,是目前最好的中文情感分析模型。
模型 | dev | test | 模型(finetune) | dev | test |
---|---|---|---|---|---|
BOW | 89.8% | 90.0% | BOW | 91.3% | 90.6% |
CNN | 90.6% | 89.9% | CNN | 92.4% | 91.8% |
LSTM | 90.0% | 91.0% | LSTM | 93.3% | 92.2% |
GRU | 90.0% | 89.8% | GRU | 93.3% | 93.2% |
BI-LSTM | 88.5% | 88.3% | BI-LSTM | 92.8% | 91.4% |
ERNIE | 95.1% | 95.4% | ERNIE | 95.4% | 95.5% |
ERNIE+BI-LSTM | 95.3% | 95.2% | ERNIE+BI-LSTM | 95.7% | 95.6% |
对话情绪识别
EmoTect(Emotion Detection)专注于识别智能对话场景中用户的情绪识别,并开源基于百度海量数据训练好的预训练模型。
模型 | 闲聊 | 客服 | 微博 |
---|---|---|---|
BOW | 90.2% | 87.6% | 74.2% |
LSTM | 91.4% | 90.1% | 73.8% |
Bi-LSTM | 91.2% | 89.9% | 73.6% |
CNN | 90.8% | 90.7% | 76.3% |
TextCNN | 91.1% | 91.0% | 76.8% |
BERT | 93.6% | 92.3% | 78.6% |
ERNIE | 94.4% | 94.0% | 80.6% |
阅读理解
MRC(Machine Reading Comprehension)机器阅读理解(MRC)是自然语言处理(NLP)中的关键任务之一,开源的DuReader升级了经典的阅读理解BiDAF模型,去掉了char级别的embedding,在预测层中使用了pointer network,并且参考了R-NET中的一些网络结构,效果上有了大幅提升
Model | Dev ROUGE-L | Test ROUGE-L |
---|---|---|
BiDAF (原始论文基线) | 39.29 | 45.90 |
本基线系统 | 47.68 | 54.66 |
语义模型
ERNIE
ERNIE(Enhanced Representation from kNowledge IntEgration)百度自研的语义表示模型,通过建模海量数据中的词、实体及实体关系,学习真实世界的语义知识。相较于 BERT 学习原始语言信号,ERNIE直接对先验语义知识单元进行建模,增强了模型语义表示能力。
数据集 |
XNLI |
LCQMC |
MSRA-NER |
ChnSentiCorp |
nlpcc-dbqa |
|||||||
评估 |
acc |
acc |
f1-score |
acc |
mrr |
f1-score |
||||||
dev |
test |
dev |
test |
dev |
test |
dev |
test |
dev |
test |
dev |
test |
|
BERT |
78.1 |
77.2 |
88.8 |
87 |
94.0 |
92.6 |
94.6 |
94.3 |
94.7 |
94.6 |
80.7 |
80.8 |
ERNIE |
79.9(+1.8) |
78.4(+1.2) |
89.7(+0.9) |
87.4(+0.4) |
95.0(+1.0) |
93.8(+1.2) |
95.2(+0.6) |
95.4(+1.1) |
95.0(+0.3) |
95.1(+0.5) |
82.3(+1.6) |
82.7(+1.9) |
BERT
BERT(Bidirectional Encoder Representation from Transformers)是一个迁移能力很强的通用语义表示模型, 以 Transformer 为网络基本组件,以双向 Masked Language Model和 Next Sentence Prediction 为训练目标,通过预训练得到通用语义表示,再结合简单的输出层,应用到下游的 NLP 任务,在多个任务上取得了 SOTA 的结果。
ELMo
ELMo(Embeddings from Language Models)是重要的通用语义表示模型之一,以双向 LSTM 为网路基本组件,以 Language Model 为训练目标,通过预训练得到通用的语义表示,将通用的语义表示作为 Feature 迁移到下游 NLP 任务中,会显著提升下游任务的模型性能。
SimNet
SimNet(Similarity Net)一个计算短文本相似度的框架,可以根据用户输入的两个文本,计算出相似度得分。
模型 | 百度知道 | ECOM | QQSIM | UNICOM | LCQMC |
---|---|---|---|---|---|
AUC | AUC | AUC | 正逆序比 | Accuracy | |
BOW_Pairwise | 0.6767 | 0.7329 | 0.7650 | 1.5630 | 0.7532 |
文本生成
机器翻译
MT(machine translation)机器翻译是利用计算机将一种自然语言(源语言)转换为另一种自然语言(目标语言)的过程,输入为源语言句子,输出为相应的目标语言的句子。
测试集 | newstest2014 | newstest2015 | newstest2016 |
---|---|---|---|
Base | 26.35 | 29.07 | 33.30 |
Big | 27.07 | 30.09 | 34.38 |
对话自动评估
对话自动评估(Auto Dialogue Evaluation)主要用于评估开放领域对话系统的回复质量,能够帮助企业或个人快速评估对话系统的回复质量,减少人工评估成本。
利用少量标注数据微调后,自动评估打分和人工打分spearman相关系数,如下表。
/ | seq2seq_naive | seq2seq_att | keywords | human |
---|---|---|---|---|
cor | 0.474 | 0.477 | 0.443 | 0.378 |
对话通用理解
DGU(Dialogue General Understanding)对话通用理解针对数据集开发了相关的模型训练过程,支持分类,多标签分类,序列标注等任务,用户可针对自己的数据集,进行相关的模型定制
ask_name | udc | udc | udc | atis_slot | dstc2 | atis_intent | swda | mrda |
---|---|---|---|---|---|---|---|---|
对话任务 | 匹配 | 匹配 | 匹配 | 槽位解析 | DST | 意图识别 | DA | DA |
任务类型 | 分类 | 分类 | 分类 | 序列标注 | 多标签分类 | 分类 | 分类 | 分类 |
任务名称 | udc | udc | udc | atis_slot | dstc2 | atis_intent | swda | mrda |
评估指标 | R1@10 | R2@10 | R5@10 | F1 | JOINT ACC | ACC | ACC | ACC |
SOTA | 76.70% | 87.40% | 96.90% | 96.89% | 74.50% | 98.32% | 81.30% | 91.70% |
DGU | 82.02% | 90.43% | 97.75% | 97.10% | 89.57% | 97.65% | 80.19% | 91.43% |
DAM
深度注意力机制模型(Deep Attention Maching)是开放领域多轮对话匹配模型。根据多轮对话历史和候选回复内容,排序出最合适的回复。
Ubuntu Corpus | Douban Conversation Corpus | |||||||||
---|---|---|---|---|---|---|---|---|---|---|
R2@1 | R10@1 | R10@2 | R10@5 | MAP | MRR | P@1 | R10@1 | R10@2 | R10@5 | |
DAM | 93.8% | 76.7% | 87.4% | 96.9% | 55.0% | 60.1% | 42.7% | 25.4% | 41.0% | 75.7% |
知识驱动对话
知识驱动对话的新对话任务其中机器基于构建的知识图与人交谈。它旨在测试机器进行类似人类对话的能力。
baseline system | F1/BLEU1/BLEU2 | DISTINCT1/DISTINCT2 |
---|---|---|
retrieval-based | 31.72/0.291/0.156 | 0.118/0.373 |
generation-based | 32.65/0.300/0.168 | 0.062/0.128 |
PaddleRec
个性化推荐,在当前的互联网服务中正在发挥越来越大的作用,目前大部分电子商务系统、社交网络,广告推荐,搜索引擎,都不同程度的使用了各种形式的个性化推荐技术,帮助用户快速找到他们想要的信息。
模型名称 | 模型简介 |
---|---|
TagSpace | 应用于工业级的标签推荐,具体应用场景有feed新闻标签推荐等 |
GRU4Rec | 首次将RNN(GRU)运用于session-based推荐,相比传统的KNN和矩阵分解,效果有明显的提升 |
SequenceSemanticRetrieval | 使用参考论文中的思想,使用多种时间粒度进行用户行为预测 |
DeepCTR | 只实现了DeepFM论文中介绍的模型的DNN部分,DeepFM会在其他例子中给出 |
Multiview-Simnet | 基于多元视图,将用户和项目的多个功能视图合并为一个统一模型 |
Word2Vec | skip-gram模式的word2vector模型 |
GraphNeuralNetwork | 基于会话的图神经网络模型的推荐系统,可以更好的挖掘item中丰富的转换特性以及生成准确的潜在的用户向量表示 |
DeepInterestNetwork | DIN通过一个兴趣激活模块(Activation Unit),用预估目标Candidate ADs的信息去激活用户的历史点击商品,以此提取用户与当前预估目标相关的兴趣。 |
License
This tutorial is contributed by PaddlePaddle and licensed under the Apache-2.0 license.
许可证书
此向导由PaddlePaddle贡献,受Apache-2.0 license许可认证。