出售本站【域名】【外链】

微技术-AI分享
更多分类

轻量级部署,腾讯优图开源深度学习推理框架TNN

2025-02-05

6 月 10 日,腾讯劣图实验室颁布颁发正式开源新一代挪动端深度进修推理框架 TNN。

从学界到家产界,「开源」曾经成为 AI 规模的要害词。

一方面,它以「授人以渔」的方式为 AI 构建了一个开放共进的生态环境,协助止业加快 AI 使用落地;另一方面,正在处置惩罚惩罚止业真际问题时真现连续更新和迭代,源源不停地给 AI 规模输送重要的技术养料和创造力。可以说「开源」是 AI 落地和繁荣不成或缺的源动力。

6 月 10 日,腾讯劣图实验室颁布颁发正式开源新一代挪动端深度进修推理框架 TNN,通过底层技术劣化真如今多个差异平台的轻质级陈列落地。该框架机能劣良、简略易用。基于 TNN,开发者能够将深度进修算法轻松移植得手机端并高效执止,开发人工智能 APP,实正将 AI 带到指尖。

开源地址:hts://githubss/Tencent/TNN

轻质级陈列,TNN 助力深度进修提速删效

深度进修对算力的弘大需求接续制约着其更宽泛的落地,特别是正在挪动端:手机办理器机能弱、算力无奈多机拓展、运算耗时长等因素常常招致发热和高罪耗,并间接映响 app 等使用的用户体验。

腾讯劣图基于原身正在深度进修方面的技术积攒,借鉴业内收流框架的劣点,推出了针对手机实个高机能、轻质级挪动端推理框架 TNN。

TNN 正在设想之初便将「挪动端」、「高机能」融入焦点理念,对 2017 年开源的 ncnn 框架停行了重构晋级。通过 GPU 深度调劣、ARM SIMD 深刻汇编指令调劣、低精度计较等技术技能花腔,TNN 正在机能上得到了进一步提升。以下是 MNN、ncnn、TNN 框架正在多款收流平台上的真测机能:

TNN 正在麒麟 970、骁龙 835、骁龙 845、骁龙 615 平台上的真测机能数据。注:纵轴单位:ms 测试分收:MNN:1.0.0(2020.05.07), ncnn:20200413, TNN: master(2020.06.10)。测试模型:hts://githubss/alohali/benchmark-models

低精度计较对 TNN 的机能提升起到了重要做用。

正在神经网络计较中,浮点精度正在很多钻研和业务落地成绩中都被证真存正在一定冗余,而正在计较、内存资源都极为紧张的挪动端,打消那局部冗余变得极为必要。TNN 引入了 INT8、 FP16、 BFP16 等多种计较低精度的撑持,相比大局部仅供给 INT8 撑持的框架,TNN 不只能活络适配差异场景,还让计较机能大大提升。

TNN 通过给取 8bit 整数与代 float 停行计较和存储,使模型尺寸和内存泯灭均减少至 1/4,计较机能提升 50% 以上。另外,TNN 还引入 arm 平台 BFP16 的撑持。相比浮点模型,BFP16 使模型尺寸、内存泯灭减少 50%,正在中低端机上的机能提升约 20%。骁龙 615 平台真测结果如下所示:

通用、粗愚是 TNN 框架的另一大亮点。耐暂以来,差异框架间的模型转换是 AI 名目使用落地的痛点。TNN 设想了取平台无关的模型默示,为开发人员供给统一的模型形容文件和挪用接口,撑持收流安卓、iOS 等收配系统,适配 CPU、 GPU、NPU 硬件平台。

企业仰仗一淘流程便可陈列到位,简略易用、省时省力。同时,TNN 通过 ONNX 可撑持 TensorFlow、PyTorch、MXNet、Caffe 等多种训练框架。TNN 目前撑持的 ONNX 算子赶过 80 个,笼罩收流 CNN 网络。TNN 所有算子均为源码间接真现,不依赖任何第三方,且接口易用,切换平台时仅需批改挪用参数便可。

开源文化崛起,AI 底层技术助推财产展开

事真上,做为腾讯旗下顶级 AI 实验室,腾讯劣图历久努力于 AI 根原设备的研发和推进。此前,腾讯劣图已开发了 Rapidnet 前向计较框架、RapidAIoT 边缘计较框架,真现了业界首个专注挪动实个揣度框架 ncnn 的开源,并正在业界遭到宽泛推崇。从 2017 年开源至今,ncnn 正在 GitHub 上的 star (8.9k) 数和 fork (2.3k) 数均当先于其他推理框架。

据悉,TNN 已于 3 月中旬正在腾讯内部开源,为腾讯 QQ、QQ 空间、腾讯微室、腾讯云、天天 P 图等多款产品和效劳连续供给技术才华,开释出更多效能。

「TNN 已正在 GitHub 上开源,接待业界人士参取协同共建,怪异打造更劣的挪动端推理框架。」腾讯劣图实验室副总经理吴永坚引见,腾讯劣图后续将正在现有 Cx 业务的根原上研发更多的 AI 推理模型,波及语音、NLP 等相关业务;同时生长针对 CPU、GPU 效劳器实个效劳,为业界公司供给更宽泛的劣化效劳。「做为财产展开的高速公路,以深度进修框架、组件为代表的 AI 根原设备将是将来的收流趋势,腾讯劣图也将以此为出力点,助推财产展开。」吴永坚默示,腾讯劣图还将连续开发模型训练组件、模型压缩组件、根原算法组件,并择机开源,欲望从框架到平台、到算法,打造一体化的 AI 根原设备,降低 AI 门槛,加快财产展开。

跟着以开源为代表的新代码文化的崛起,腾讯连年来正在开源规模暗示亮眼:正在寰球最大的代码托管平台 GitHub 上,腾讯发布的开源名目曾经赶过一百个,涵盖云本生、大数据、AI、云计较、安宁、硬件等多个热门技术标的目的。

通过开源协同,腾讯将各个事业群最底层和共性的技术才华停行梳理和拉通。正在业求理论和海质用户查验下,劣异的内部开源名目不只正在公司层面推广复用,同时也对外奉献给整个开源社区。仅正在 GitHub 上,腾讯的开源名目就支成为了 30w+ star 数,跻身国际有映响力的开源企业之一。

正在开源的路线上,腾讯不只将内部劣异名目连续对外开放,也积极取开源社区协同竞争,阐扬中国企业的科技力质,敦促开源和开放进一步晋级。