|
|

& y5 b0 O$ K+ K, R2 _【新智元导读】Google Coral Edge TPU和NVIDIA Jetson Nano大比拼!本文从分别对两款最新推出的EdgeAI芯片做了对比,分析了二者各自的优劣势。( h# G( D0 @& ~% g/ ?* b4 a+ T
+ `" x8 A; J3 ]9 K, _" y3 _/ ^% X
边缘智能被称作是人工智能的最后一公里。
3 f! p! G% \) H2 J$ C
. t* y6 H* R# vGoogle刚刚在3月份推出了Coral Edge TPU,是一款售价不到1000元人民币的开发板(Coral Dev Board),由Edge TPU模块和 Baseboard 组成。参数如下:% h' S5 F- U( D$ |' Y# u# @+ L7 V

1 t- }0 z0 w3 X% }* z: H. m* [/ ?2 w
6 x3 L! j4 n% T3 w0 J0 O3 u英伟达同样在上个月发布了最新的NVIDIA Jetson Nano,Jetson Nano是一款类似于树莓派的嵌入式电脑设备,其搭载了四核Cortex-A57处理器,GPU则是拥有128个NVIDIA CUDA核心的NVIDIA麦克斯韦架构显卡,内存4GB LPDDR4,存储则为16GB eMMC 5.1,支持4K 60Hz视频解码。. o. y) P+ h C
目前位置并没有太多关于这两款产品的评测报告。今天新智元为大家带来一篇由网友Sam Sterckval对两款产品的评测,除此以外他还测试了i7-7700K + GTX1080(2560CUDA),Raspberry Pi 3B +,以及一个2014年的MacBook pro包含一个i7-4870HQ(没有支持CUDA的内核)。
$ T& _0 d0 z" D- \: ySam使用MobileNetV2作为分类器,在imagenet数据集上进行预训练,直接从Keras使用这个模型,后端则使用TensorFlow。使用GPU的浮点权重,以及CPU和Coral Edge TPU的8bit量化tflite版本。8 u- f& z5 T7 z2 t5 Z3 E) l
首先,加载模型以及一张喜鹊图像。先执行1个预测作为预热,Sam发现第一个预测总是比随后的预测更能说明问题。然后Sleep 1秒,确保所有的线程的活动都终止,然后对同一图像进行250次分类。
. C3 |- F) I7 R5 K6 X) n , f% r7 z* `1 z8 x
对所有分类使用相同的图像,能够确保在整个测试过程中保持接近的数据总线。
/ r6 m! S- q. v! U/ z! v% ^
0 @( M9 }+ n2 ]+ o# L8 G对比结果 先来看最终的结果:7 c' U) i* t% g% L% H s
8 ?' D5 N) }# ]! P- N4 A
线性刻度,FPS 2 j+ t3 @7 K- g
对数刻度,FPS) L) X3 j I" c6 U- ^
$ L. X2 d" r6 g. j
4 {: E. r t9 v; I9 i% f9 C" _. |/ P线性刻度,推理时间(250x)
4 W2 F+ Z6 a& V0 t
0 q: h& t; u2 l1 e# sSam发现使用CPU的量化tflite模型得分是不同的,但似乎它总是返回与其它产品相同的预测结果,他怀疑模型有点奇怪,但能确保它不会影响性能。! \! h( N4 s( _+ G* c
对比分析 第一个柱状图中我们可以看到有3个比较突出的数据,其中两个2个是由Google Coral Edge TPU USB加速器实现的,第3个是由英特尔i7-7700K辅助NVIDIA GTX1080实现。
9 J; i0 R! B+ c0 i P我们再仔细对比一下就会发现,GTX1080实际上完全无法跟Google的Coral对飚。要知道GTX1080的最大功率为180W,而Coral Edge TPU只有2.5W。
0 E% V1 i9 z7 E+ j7 ZNVIDIA Jetson Nano的得分并不高。虽然它有一个支持CUDA的GPU,但实际上并没比那台2014年MBP的i7-4870HQ快太多,但毕竟还是比这款四核,超线程的CPU要快。
; P% k. r; ^5 \' c; S) C+ h然而相比i7 50W的能耗,Jetson Nano平均能耗始终保持在12.5W,也就是说功耗降低75%,性能提升了10%。
4 d6 L9 p& o4 g* qNVIDIA Jetson Nano
7 Q- ?+ h# d2 b8 A$ E$ f3 r尽管Jetson Nano并没有在MobileNetV2分类器中表现出令人印象深刻的FPS率,但它的优势非常明显:
7 q8 ]8 x( x7 a8 e( X" P$ I它很便宜,能耗低,更重要的是,它运行TensorFlow-gpu或任何其他ML平台的操作,和我们平时使用的其他设备一样。只要我们的脚本没有深入到CPU体系结构中,就可以运行与i7 + CUDA GPU完全相同的脚本,也可以进行训练!Sam强烈希望NVIDIA应该使用TensorFlow预加载L4T。
7 B7 X- s j& y6 [3 x ; u D5 D1 o* S" l4 T
Google Coral Edge TPU6 a" k" i1 {+ i( h
Sam毫不掩饰的表达了他对Google Coral Edge TPU的精心设计以及高效率的喜爱。下图我们可以对比Edge TPU有多小。
" @6 v# {2 H* Z- G, D 6 Y" Y; }- [1 a; e
Penny for scale,来源:谷歌( O& H% Y p* L' p; m
: l7 _- N) P; S; J9 vEdge TPU就是所谓的“ASIC”(专用集成电路),这意味着它具有FET等小型电子部件,以及能够直接在硅层上烧制,这样它就可以加快在特定场景下的推力速度。但Edge TPU无法执行反向传播。/ f5 V& x- @9 H/ j7 R7 Q) s+ E
; p k7 g! Y2 f
Google Coral Edge TPU USB加速器
4 y; u# |& T* J) z4 u7 f; F+ A: G5 w1 r, v ^$ _& `
下图显示了Edge TPU的基本原理。, S7 `# Y* U3 M Z* {) b
; r0 {( \. @+ A% [2 Z' [. z
像MobileNetV2这样的网络主要由后面带有激活层的卷积组成。公式如下:
" s y9 t# g# p+ O! D7 Q4 E V ' x! U. c/ }7 G' \4 V X* E9 A
卷积" f n$ H, Y g: v
2 \. W2 [3 [; x1 ^" ?" u
这意味着将图像的每个元素(像素)与内核的每个像素相乘,然后将这些结果相加,以创建新的“图像”(特征图)。这正是Edge TPU的主要工作。将所有内容同时相乘,然后以疯狂的速度添加所有内容。这背后没有CPU,只要你将数据泵入左边的缓冲区就可以了。
! ?. t/ M6 P8 s4 s6 N4 [$ b. v我们看到Coral在性能/瓦特的对比中,差异如此大的原因,它是一堆电子设备,旨在完成所需的按位操作,基本上没有任何开销。
h' r' J' d, g7 ~5 E总结 为什么GPU没有8位模型?
: Y8 h, ^8 V! N& U M1 E, i2 K( L8 uGPU本质上被设计为细粒度并行浮点计算器。而Edge TPU设计用于执行8位操作,并且CPU具有比完全位宽浮点数更快的8位内容更快的方法,因为它们在很多情况下必须处理这个问题。
/ `8 D O0 e9 T3 X# l L* W0 G- S为何选择MobileNetV2?* h0 N- J; M( V7 \& O
主要原因是,MobileNetV2是谷歌为Edge TPU提供的预编译模型之一。
, J# @+ ?/ ~$ T$ W. v, R* XEdge TPU还有哪些其他产品?
3 m; ?# O, N2 c6 Z5 H0 W# B4 U& @它曾经是不同版本的MobileNet和Inception,截至上周末,谷歌推出了一个更新,允许我们编译自定义TensorFlow Lite模型。但仅限于TensorFlow Lite模型。而反观Jetson Nano就没有这方面的限制。
" k: e9 @4 K& Z" QRaspberry Pi + Coral与其他人相比
: _( I/ C9 V5 G: k6 k' k为什么连接到Raspberry Pi时Coral看起来要慢得多?因为Raspberry Pi只有USB 2.0端口。7 U j- C1 @. x$ _; \, t! L* Q& g
i7-7700K在Coral和Jetson Nano上的速度都会更快一些,但仍然无法和后两者比肩。因此推测瓶颈是数据速率,不是Edge TPU。% K: s) Q5 K3 y5 P
: } W) |, b$ S. f' [
- W" e+ i$ [! ]5 Z+ j+ ]# y c
【加入社群】, N& ?+ D- f+ K6 n6 P6 a: V
- T2 R' X% o# J3 j/ n
1 O, r* E j" |( D# Q9 e+ s( S' m
来源:http://www.yidianzixun.com/article/0Lt3VMHK
- i x8 ]. j: G" T免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作! |
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有账号?立即注册
×
|