|
|

9 Q9 I7 j3 s( w" V6 O【新智元导读】Google Coral Edge TPU和NVIDIA Jetson Nano大比拼!本文从分别对两款最新推出的EdgeAI芯片做了对比,分析了二者各自的优劣势。
" T# ]: P& R2 j" ?/ q% F/ y- _: E
; \- S/ Y' H" `* b" l: A边缘智能被称作是人工智能的最后一公里。
& y% d5 T4 H9 Z% L5 Q3 e" [- u
' d U8 j' s* U" IGoogle刚刚在3月份推出了Coral Edge TPU,是一款售价不到1000元人民币的开发板(Coral Dev Board),由Edge TPU模块和 Baseboard 组成。参数如下:
: F( M; n- a4 [5 V5 c3 H
v3 F, l+ g) L) k- q
" }- l6 m! A+ ^4 `: {4 A英伟达同样在上个月发布了最新的NVIDIA Jetson Nano,Jetson Nano是一款类似于树莓派的嵌入式电脑设备,其搭载了四核Cortex-A57处理器,GPU则是拥有128个NVIDIA CUDA核心的NVIDIA麦克斯韦架构显卡,内存4GB LPDDR4,存储则为16GB eMMC 5.1,支持4K 60Hz视频解码。7 v' v" q% {- B* F* C% _8 ?9 F
目前位置并没有太多关于这两款产品的评测报告。今天新智元为大家带来一篇由网友Sam Sterckval对两款产品的评测,除此以外他还测试了i7-7700K + GTX1080(2560CUDA),Raspberry Pi 3B +,以及一个2014年的MacBook pro包含一个i7-4870HQ(没有支持CUDA的内核)。
; M+ {2 T4 P" G- S LSam使用MobileNetV2作为分类器,在imagenet数据集上进行预训练,直接从Keras使用这个模型,后端则使用TensorFlow。使用GPU的浮点权重,以及CPU和Coral Edge TPU的8bit量化tflite版本。* l* l% g) ]: _, G/ e+ l
首先,加载模型以及一张喜鹊图像。先执行1个预测作为预热,Sam发现第一个预测总是比随后的预测更能说明问题。然后Sleep 1秒,确保所有的线程的活动都终止,然后对同一图像进行250次分类。) Y+ S3 C+ E) \8 n

$ u w; C! M) ~8 P! m0 n6 S对所有分类使用相同的图像,能够确保在整个测试过程中保持接近的数据总线。% ?7 i. \; ]: A. H- \ b( w/ v3 Z
+ S4 `/ ~# z, R" M: ~. H6 |" x对比结果 先来看最终的结果:% ^: A. u& i; d

' P2 V% ^% o4 p1 [3 a1 T* Z 线性刻度,FPS ' U" w5 X3 J6 C* m) Q
对数刻度,FPS
% ^1 K4 V5 ^! |, k9 g
^+ E7 t& S# M
$ g- [1 A. X1 n6 F! g线性刻度,推理时间(250x)
9 A9 m0 o. Z+ p& T1 x9 T) W1 B k# Z4 Z) J K0 @
Sam发现使用CPU的量化tflite模型得分是不同的,但似乎它总是返回与其它产品相同的预测结果,他怀疑模型有点奇怪,但能确保它不会影响性能。, V: t; E4 Y; I1 a
对比分析 第一个柱状图中我们可以看到有3个比较突出的数据,其中两个2个是由Google Coral Edge TPU USB加速器实现的,第3个是由英特尔i7-7700K辅助NVIDIA GTX1080实现。# S* O% m: |5 N1 e
我们再仔细对比一下就会发现,GTX1080实际上完全无法跟Google的Coral对飚。要知道GTX1080的最大功率为180W,而Coral Edge TPU只有2.5W。1 V6 K7 J& \2 L9 h8 R/ C
NVIDIA Jetson Nano的得分并不高。虽然它有一个支持CUDA的GPU,但实际上并没比那台2014年MBP的i7-4870HQ快太多,但毕竟还是比这款四核,超线程的CPU要快。/ V; e# i- a! T/ v, Y8 g5 b
然而相比i7 50W的能耗,Jetson Nano平均能耗始终保持在12.5W,也就是说功耗降低75%,性能提升了10%。
4 D, T3 l4 G9 g* \' t; S9 N( iNVIDIA Jetson Nano* \- h6 R- _8 r+ m
尽管Jetson Nano并没有在MobileNetV2分类器中表现出令人印象深刻的FPS率,但它的优势非常明显:, e, k2 o# f/ I. P% }$ c! l$ T
它很便宜,能耗低,更重要的是,它运行TensorFlow-gpu或任何其他ML平台的操作,和我们平时使用的其他设备一样。只要我们的脚本没有深入到CPU体系结构中,就可以运行与i7 + CUDA GPU完全相同的脚本,也可以进行训练!Sam强烈希望NVIDIA应该使用TensorFlow预加载L4T。
" m1 M; D9 B- X- Y+ V1 P. T, @
4 Z! s0 \- U5 g( T/ T6 YGoogle Coral Edge TPU
+ D0 U0 _8 u/ v5 l) HSam毫不掩饰的表达了他对Google Coral Edge TPU的精心设计以及高效率的喜爱。下图我们可以对比Edge TPU有多小。5 g; R& ]/ c" [8 N% N
0 { z4 s( k) S& x; t
Penny for scale,来源:谷歌" I% h: ~# |/ N/ D5 ]5 N
# C& _8 z/ R& S7 h2 O/ h# l
Edge TPU就是所谓的“ASIC”(专用集成电路),这意味着它具有FET等小型电子部件,以及能够直接在硅层上烧制,这样它就可以加快在特定场景下的推力速度。但Edge TPU无法执行反向传播。
% Q6 S8 p7 G, s/ f: _( [/ U. { : o8 r9 a7 f0 m) F0 M! [2 g
Google Coral Edge TPU USB加速器
( F8 B; f) H& V
- F$ @$ K# K& n/ K8 n9 l下图显示了Edge TPU的基本原理。! S2 \8 r, J/ ]* Y
, B: x2 a. v% D2 {! Z2 P t5 Q: ~
像MobileNetV2这样的网络主要由后面带有激活层的卷积组成。公式如下:
/ N, T. O) g& I* j" } , G, y* p2 U! i
卷积
6 X% I1 u+ k8 a- t: {
. ]! Y! m( `3 e这意味着将图像的每个元素(像素)与内核的每个像素相乘,然后将这些结果相加,以创建新的“图像”(特征图)。这正是Edge TPU的主要工作。将所有内容同时相乘,然后以疯狂的速度添加所有内容。这背后没有CPU,只要你将数据泵入左边的缓冲区就可以了。$ o8 G% ~/ q% M% V3 ~2 G
我们看到Coral在性能/瓦特的对比中,差异如此大的原因,它是一堆电子设备,旨在完成所需的按位操作,基本上没有任何开销。 L! k( K; c' b& e$ n1 H
总结 为什么GPU没有8位模型?2 w7 f' e5 a( {* a# F
GPU本质上被设计为细粒度并行浮点计算器。而Edge TPU设计用于执行8位操作,并且CPU具有比完全位宽浮点数更快的8位内容更快的方法,因为它们在很多情况下必须处理这个问题。
$ f6 }& p+ j9 {6 m' `3 ?为何选择MobileNetV2?- d6 g4 F# r+ M. \# C7 r% Y
主要原因是,MobileNetV2是谷歌为Edge TPU提供的预编译模型之一。 V! R9 }; c+ t9 u- ^2 ~
Edge TPU还有哪些其他产品?
% U" [& H3 B( A# S' Q5 _1 \* i, O它曾经是不同版本的MobileNet和Inception,截至上周末,谷歌推出了一个更新,允许我们编译自定义TensorFlow Lite模型。但仅限于TensorFlow Lite模型。而反观Jetson Nano就没有这方面的限制。
$ F+ o- `+ G5 P! L$ d; WRaspberry Pi + Coral与其他人相比
, y+ x0 R: R' N; x为什么连接到Raspberry Pi时Coral看起来要慢得多?因为Raspberry Pi只有USB 2.0端口。
8 Z+ y1 \6 F5 q$ W' T0 ^7 Ai7-7700K在Coral和Jetson Nano上的速度都会更快一些,但仍然无法和后两者比肩。因此推测瓶颈是数据速率,不是Edge TPU。' c5 o1 q8 t5 y* A5 G& t( d
" [# c! _" l: a3 _

7 U2 t) l" n1 `( I/ I2 W7 J. n【加入社群】 t1 R/ Q- w! G+ b5 i
2 c- H" q6 T7 r; d5 T. ]* Y/ E4 m
' S9 k" r9 c% g4 ?; ?% k: r来源:http://www.yidianzixun.com/article/0Lt3VMHK$ {8 T( A$ S- F! ~, R7 `( L
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作! |
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有账号?立即注册
×
|