電子產(chǎn)業(yè)一站式賦能平臺(tái)

PCB聯(lián)盟網(wǎng)

搜索
查看: 33|回復(fù): 0
收起左側(cè)

硅基光電子技術(shù)在深度學(xué)習(xí)加速中的應(yīng)用:CrossLight介紹

[復(fù)制鏈接]

465

主題

465

帖子

3514

積分

四級(jí)會(huì)員

Rank: 4

積分
3514
跳轉(zhuǎn)到指定樓層
樓主
發(fā)表于 2024-9-4 08:00:00 | 只看該作者 |只看大圖 回帖獎(jiǎng)勵(lì) |倒序?yàn)g覽 |閱讀模式
引言隨著深度神經(jīng)網(wǎng)絡(luò)(DNNs)在各種應(yīng)用中變得越來(lái)越復(fù)雜和普遍,對(duì)高效硬件加速器的需求比以往任何時(shí)候都更為迫切。在后摩爾定律時(shí)代,傳統(tǒng)電子加速器面臨著基本限制,在帶寬和能效方面造成瓶頸。硅基光電子技術(shù)應(yīng)運(yùn)而生,可為深度學(xué)習(xí)加速提供節(jié)能、超高帶寬和低延遲的解決方案。- u& x' z  M$ c% M
本文介紹CrossLight,新型硅基光電子神經(jīng)網(wǎng)絡(luò)加速器,通過(guò)跨層設(shè)計(jì)方法解決光計(jì)算中的關(guān)鍵挑戰(zhàn)。將探討光計(jì)算的基礎(chǔ)知識(shí)、CrossLight的架構(gòu)以及與最先進(jìn)加速器的性能比較。% a; u" x, X) m$ q, k- q

' q4 u& R8 V, W光計(jì)算基礎(chǔ)5 d' g- b( s% {- V
在深入了解CrossLight之前,讓我們先了解深度學(xué)習(xí)光計(jì)算的基礎(chǔ)知識(shí)。光加速器通常使用廣播和權(quán)重(B&W)配置來(lái)執(zhí)行矩陣-矢量乘法,這對(duì)DNN中的卷積(CONV)和全連接(FC)層都是必不可少的。8 @) ?# `- x2 }+ Y
& b" {3 ^6 d5 |& z% W  @' H# g
圖1顯示了基于非相干廣播和權(quán)重(B&W)的光電子神經(jīng)元配置。3 s1 J$ q* [" I9 r
在此配置中,輸入值通過(guò)調(diào)制器印刻在不同波長(zhǎng)的光上。然后,這些波長(zhǎng)被合并并分成多個(gè)分支,每個(gè)分支由微環(huán)諧振器(MRs)加權(quán)。加權(quán)信號(hào)通過(guò)光電探測(cè)器求和,完成矩陣-矢量乘法運(yùn)算。
/ _7 V1 }4 I7 K/ ^( A* |/ O這種設(shè)置中的關(guān)鍵組件是微環(huán)諧振器(MR)。MR可以調(diào)諧以改變特定波長(zhǎng)的能量,有效地在光域中實(shí)現(xiàn)乘法運(yùn)算。
  c& O0 {8 `' X' Z) aCrossLight架構(gòu)0 q% w! ], h, H# ^
CrossLight采用跨層方法優(yōu)化光加速,解決設(shè)備、線路和架構(gòu)層面的挑戰(zhàn)。/ S) O# r: N3 I: i8 j8 a4 v
2 b! E7 r& S- C! \  o
圖2展示了CrossLight非相干硅基光電子神經(jīng)網(wǎng)絡(luò)加速器的高級(jí)概述。+ ]6 b2 b- R3 C( H
設(shè)備級(jí)優(yōu)化在設(shè)備級(jí)別,CrossLight引入了優(yōu)化的MR設(shè)計(jì),對(duì)制造工藝變化(FPVs)更具彈性。通過(guò)全面的設(shè)計(jì)空間探索,研究人員發(fā)現(xiàn),使用400納米的輸入波導(dǎo)寬度和800納米的環(huán)形波導(dǎo)寬度可以將由FPV引起的不期望的諧振波長(zhǎng)偏移減少70%。/ r. t) Q2 g% }! e
線路級(jí)優(yōu)化為解決緊密排列的MR之間的熱串?dāng)_問(wèn)題,CrossLight采用了結(jié)合熱光(TO)和電光(EO)調(diào)諧的混合調(diào)諧方法。與傳統(tǒng)的僅TO調(diào)諧方法相比,這種方法可以實(shí)現(xiàn)更快的操作速度和更低的功耗。( D$ `# t$ |, A4 S. _0 B
此外,CrossLight采用了稱為熱特征分解(TED)的方法,可以集體調(diào)諧MR組中的所有MR,有效地以較低的功耗消除熱串?dāng)_效應(yīng)。' `' Y* X7 d  c, X2 Y! C4 U) v
) i( Q$ J3 ]8 i) d
圖3顯示了10個(gè)制造的MR塊中相鄰MR對(duì)之間距離可變時(shí)的相位串?dāng)_比和調(diào)諧功耗。$ z6 n* r( j% O: H( _" U" ]
架構(gòu)級(jí)優(yōu)化CrossLight為CONV和FC層加速引入了單獨(dú)的矢量點(diǎn)積(VDP)單元,認(rèn)識(shí)到這些層的不同計(jì)算需求。這種分離允許更高效地處理這兩種類型的層。4 H8 u  X8 }& _  l4 ^
該架構(gòu)還在VDP單元內(nèi)實(shí)現(xiàn)了波長(zhǎng)重用策略,減少了所需的激光器總數(shù),從而降低了功耗。通過(guò)將較大的矢量分解為較小的矢量,并在VDP單元內(nèi)的多個(gè)分支上執(zhí)行并行計(jì)算,CrossLight在并行性和激光器功率需求之間實(shí)現(xiàn)了平衡。  u* D! t, m, [1 h7 ~  I4 C
性能分析為評(píng)估CrossLight的性能,研究人員使用四個(gè)不同復(fù)雜度的DNN模型進(jìn)行了廣泛的模擬。5 ]& X$ L& v6 |8 F
分辨率分析CrossLight的一個(gè)主要優(yōu)勢(shì)是能夠?qū)崿F(xiàn)高分辨率計(jì)算。雖然一些光加速器限制在2-4位分辨率,但CrossLight可以為其MR組實(shí)現(xiàn)高達(dá)16位的分辨率。
* x: I9 N  `* N; b5 C 4 \1 X" P+ |0 i6 |& _- `/ N. y
圖4演示了四個(gè)DNN模型在權(quán)重和激活的量化(分辨率)范圍從1位到16位時(shí)的推理準(zhǔn)確性。
; K* g" @- Q- o! S3 A' T" p這種高分辨率對(duì)于維持模型準(zhǔn)確性很重要,特別是對(duì)于在具有挑戰(zhàn)性的數(shù)據(jù)集上訓(xùn)練的復(fù)雜模型。/ b7 ?" H8 T& u6 {+ B: D
敏感性分析研究人員進(jìn)行了敏感性分析,以確定CrossLight的最佳配置,改變CONV和FC層加速器的VDP單元的數(shù)量和復(fù)雜度。+ t' F8 v5 S$ W1 X

! q4 ~! X# x2 m8 e5 |圖5是散點(diǎn)圖,顯示了各種CrossLight配置的平均每秒幀數(shù)(FPS)與平均每比特能耗(EPB)與面積的關(guān)系。
/ n4 Z+ O$ n* y) \0 z9 V最佳配置是基于最高的FPS/EPB比率選擇的,平衡了性能和能效。, x/ y9 t2 B9 f( J8 a3 U
與最先進(jìn)加速器的比較CrossLight與兩個(gè)著名的光加速器(DEAP-CNN和Holylight)以及幾個(gè)電子加速器(包括GPU和CPU)進(jìn)行了比較。
# I5 z2 G, l  p) g 6 x* G2 @6 V; T2 u
圖6比較了CrossLight各變體與光電子和電子加速器平臺(tái)的功耗。7 X1 K8 Y; C: \9 o" s1 @8 E- `
結(jié)果顯示,CrossLight,特別是在優(yōu)化配置(Cross_opt_TED)中,實(shí)現(xiàn)了比其他光加速器和傳統(tǒng)CPU/GPU平臺(tái)更低的功耗,盡管功耗仍高于一些專用電子加速器。  `  {9 g6 m" q1 I7 O7 T

! X( p* M1 u0 H
  @/ Q" L" ]+ K& O9 |圖7比較了光電子DNN加速器的每比特能耗(EPB)值。4 j4 }7 H: w; a" L% y
在能效方面,CrossLight顯著優(yōu)于其他光加速器,平均比DEAP-CNN和Holylight分別低1544倍和9.5倍的EPB。9 R7 u* Y  {+ q. _% k
CrossLight的性能優(yōu)勢(shì)源于全面考慮了光系統(tǒng)中的各種損耗和串?dāng)_,以及在設(shè)備、線路和架構(gòu)層面采用新方法來(lái)減輕影響。6 L6 c5 t5 V: ~) k/ L+ g. Z+ k
結(jié)論CrossLight展示了光電子神經(jīng)網(wǎng)絡(luò)加速器中跨層優(yōu)化的潛力。通過(guò)解決硬件棧多個(gè)層面的挑戰(zhàn),與最先進(jìn)的光電子和電子加速器相比,在能效和每瓦性能方面實(shí)現(xiàn)了顯著改進(jìn)。) J# ^5 B# k3 [: K, _7 N. P
隨著硅基光電子制造工藝的不斷成熟,我們可以期待設(shè)備調(diào)諧成本、損耗和激光器功率開銷進(jìn)一步降低。這一趨勢(shì)可能會(huì)加強(qiáng)光域加速器在深度學(xué)習(xí)推理任務(wù)中的地位。2 v9 p+ {; l; l: l
CrossLight的成功突出了在設(shè)計(jì)下一代硬件加速器時(shí)采用全面、跨層方法的重要性。隨著我們推動(dòng)人工智能和機(jī)器學(xué)習(xí)的邊界,這種創(chuàng)新架構(gòu)將在實(shí)現(xiàn)更高效和強(qiáng)大的計(jì)算系統(tǒng)方面發(fā)揮關(guān)鍵作用。
5 n& g/ b  F' S! u: R參考文獻(xiàn)[1]M. Nikdast, S. Pasricha, G. Nicolescu, and A. Seyedi, Eds., Silicon Photonics for High-Performance Computing and Beyond, 1st ed. Boca Raton, FL, USA: CRC Press, 2021./ N) t, ~  ^: X$ i/ g
- END -
9 F2 E; Q& [1 P. N0 ]
5 ~3 i/ D7 k. C$ o" x軟件申請(qǐng)我們歡迎化合物/硅基光電子芯片的研究人員和工程師申請(qǐng)?bào)w驗(yàn)免費(fèi)版PIC Studio軟件。無(wú)論是研究還是商業(yè)應(yīng)用,PIC Studio都可提升您的工作效能。
; h( G& z2 l" a! C5 @( q7 E, Z點(diǎn)擊左下角"閱讀原文"馬上申請(qǐng)
# F( h4 N, C8 X9 c: A3 @. a: Z* m; `) o- q0 |& J1 }
歡迎轉(zhuǎn)載% [& c4 C( x3 n

& @* X/ s- m# _9 z9 f  H1 P轉(zhuǎn)載請(qǐng)注明出處,請(qǐng)勿修改內(nèi)容和刪除作者信息!6 O0 `, N/ o5 {( U# Z
: r3 e+ }# W2 H/ }4 ~
- i+ \' u: t" E* a( \
7 N; b, h5 N3 x$ ?. W

8 X, h+ o3 i- }* C" i# R0 q; [1 ^. _& E
關(guān)注我們8 u/ p1 e1 e/ M( i
( r$ D$ w- m" W5 m! d

1 `0 d: C8 L. P: t9 @% g
3 p( G  ^  `, [% J; D" t

. T( S1 Z9 o+ E0 S: \: C 4 ?/ w4 T) m4 S0 ?: u5 G0 [. w

0 c6 Z+ @4 {% E+ _ 3 S2 G; O/ J$ P* I  o
                      . G5 P# A" w; _$ `

6 f) j0 R  i' g$ [( ]
5 B+ ?5 E8 E1 s; d& F* g0 w  c# y* b

8 [. }& R5 k' {關(guān)于我們:
5 A$ W3 h8 o5 C: A3 j深圳逍遙科技有限公司(Latitude Design Automation Inc.)是一家專注于半導(dǎo)體芯片設(shè)計(jì)自動(dòng)化(EDA)的高科技軟件公司。我們自主開發(fā)特色工藝芯片設(shè)計(jì)和仿真軟件,提供成熟的設(shè)計(jì)解決方案如PIC Studio、MEMS Studio和Meta Studio,分別針對(duì)光電芯片、微機(jī)電系統(tǒng)、超透鏡的設(shè)計(jì)與仿真。我們提供特色工藝的半導(dǎo)體芯片集成電路版圖、IP和PDK工程服務(wù),廣泛服務(wù)于光通訊、光計(jì)算、光量子通信和微納光子器件領(lǐng)域的頭部客戶。逍遙科技與國(guó)內(nèi)外晶圓代工廠及硅光/MEMS中試線合作,推動(dòng)特色工藝半導(dǎo)體產(chǎn)業(yè)鏈發(fā)展,致力于為客戶提供前沿技術(shù)與服務(wù)。
/ \1 @% e6 ~% n( j: U: n2 f1 Q- |9 q6 ^* c- ~6 N
http://www.latitudeda.com/& ^  L8 ^8 @5 F/ ]: T9 S1 h8 C) Z
(點(diǎn)擊上方名片關(guān)注我們,發(fā)現(xiàn)更多精彩內(nèi)容)

發(fā)表回復(fù)

本版積分規(guī)則

關(guān)閉

站長(zhǎng)推薦上一條 /1 下一條


聯(lián)系客服 關(guān)注微信 下載APP 返回頂部 返回列表