轻松上手UAI-Train,拍拍贷人脸识别算法优化效率提高85.7%

“UAI-Train平台可让咱们方便地在短时内使用大量的GPU资源,用较低的成本训练海量的数据集,提升算法模型迭代优化的效率。” — 拍拍贷算法研究员 朱运git

UAI-Train是什么github

UAI-Train是面向AI训练任务的大规模分布式计算平台,基于P40、V100等GPU云主机集群,经过分布式扩展,最高可实现192TFlops的单精度计算能力。提供一站式训练任务托管服务,可自动化解决计算节点调度、训练环境准备、数据上传下载以及任务容灾等问题,并支持按需收费、成本可控,无需担忧资源浪费。在视频图像识别、天然语言处理、语音处理等领域均已有诸多实践。算法

拍拍贷接入UAI-Train的效果网络

经过使用分布式GPU训练平台,700W人脸数据的模型训练所需时长可从原先的一周缩短至一天,总体算法优化效率提高85.7%, 相应的迭代频率也提升数倍,为更深层次的模型结构试验提供了可能。同时UAI-Train平台备有大量GPU资源,拍拍贷的算法工程师能够同时探索多种算法模型结构,极大缩短初期算法结构探索的时间。最重要的是UAI-Train平台具有按需收费的特性,拍拍贷人脸识别算法的 GPU资源成本可由原先的上万元/月,降低至数千元/月, GPU资源的有效利用率也达到了100%。并发

表:UAI-Train与购买GPU资源的特性对比框架

关于拍拍贷机器学习

拍拍贷是一家行业领先的金融科技公司,同时也是一家很是注重技术驱动、强调自主研发的高科技公司。一直以来很是重视AI技术的探索和应用,涉及到计算机视觉、语音分析和建模、天然语言处理、复杂网络分析等针对特定非结构化数据的领域,并将迁移学习、主动学习、强化学习、多任务学习、在线学习、非监督半监督等各类机器学习算法应用至多种业务场景。尤为是人脸识别、OCR、不良中介识别和欺诈团伙挖掘、智能对话机器人、社交文本挖掘等项目,在实际的业务实践中取得了不俗的效果,极大地提高了风险反欺诈水平和运营效率。分布式

人脸识别函数

人脸识别是拍拍贷AI技术的一个重要研究方向,它经过算法识别人的脸部特征,从而能够作到实时地从图片或者视频流中检测和追踪特定的人。学习

目前拍拍贷自研人脸识别算法,在700W规模多年龄段、多姿态、多表情、多环境的人脸图片上进行训练。经过尝试不一样的网络结构,包含Inception-v三、优化后的resnet等,以及多种损失函数,例如triplet_loss、sphere、cosine、arc_loss等来优化人脸识别算法,从而提高1:1人脸认证、1:N人脸搜索、N:N人脸交叉比对、人脸聚类等场景的识别精度,并将此类技术应用于拍拍贷的风险监控、反欺诈等业务,并发挥了重要做用。

图:人脸识别业务场景

面临的问题

算法人员在优化人脸识别算法的过程当中发现**使用单台GPU机器迭代一次算法须要一周左右的时间,**效率太低影响研发进度,可是采购更多的GPU机器来探索不一样算法会致使资源成本线性增加;此外因为算法调优工做涉及诸多研究内容,例如算法效果分析、新算法调研、开发等,实际的资源使用率不高。

接触UAI-Train

在一次线下技术交流活动中,拍拍贷技术人员了解到UCloud提供一种面向人工智能算法训练的UAI-Train平台,并支持GPU资源的按需租售服务,同时该平台上还可执行多机多卡的分布式训练任务。

为了提高模型训练的效率,充分高效地利用更多的新数据来进一步提升其准确率,拍拍贷抉择后选择尝试UAI-Train平台。UCloud AI团队在GitHub上发布了适配UAI-Train平台的Insightface开发案例,用于协助拍拍贷的算法工程师很方便地将单机的人脸识别算法转化成支持分布式训练的人脸识别算法,并成功在UAI-Train平台上进行算法的快速优化。

Insightface是GitHub上一个基于MXNet框架的开源人脸识别项目(https://github.com/deepinsight/insightface)。UCloud基于insightface开发了一整套能支持分布式训练的人脸识别训练和在线推理的案例代码,并发布在GitHub上(https://github.com/ucloud/uai-sdk/tree/master/examples/mxnet/insightface),其中包括基于MXNet框架的代码及开发案例。拍拍贷的工程师基于该案例,结合自身人脸识别算法的实现和数据,一周时间内就完成了开发和调试, 并顺利在UAI-Train平台上逐步展开人脸识别算法的训练迭代工做。

图:人脸识别算法接入过程

在屡次算法优化迭代尝试后,拍拍贷经过利用高维向量表征人脸,余弦距离表达类似度,最终在开源测试集准确率表现为:lfw 99.8%, cfp_fp 97%, agedb_30 98.2% ,实际业务应用中的准确率高达 99% 以上,进一步提高了风险监管、反欺诈等业务的效率。

后续拍拍贷和UCloud计划在更多算法场景和应用场景展开深刻合做,更好地服务于金融领域客户。

对人工智能更多应用场景和解决方案感兴趣的,欢迎扫描下方二维码进群交流。

TIC 2019报名火热进行中,欢迎扫描下方二维码或点击阅读原文加入咱们,共同探讨企业上云的更多落地案例!

相关文章
相关标签/搜索