谷歌正嘗試用深度學(xué)習(xí)技術(shù)開(kāi)發(fā)新一代計(jì)算機(jī)芯片
過(guò)去幾年中,谷歌開(kāi)發(fā)了一個(gè)AI硬件家族——Tensor Processing Unit(TPU芯片),用于在服務(wù)器計(jì)算機(jī)中處理AI。使用AI來(lái)設(shè)計(jì)芯片是一個(gè)良性循環(huán):AI讓芯片變得更好,經(jīng)過(guò)改良的芯片又能增強(qiáng)AI算法,依此類(lèi)推。
本文引用地址:http://m.ptau.cn/article/202002/410149.htm在主題演講中,Dean向參會(huì)者介紹了如何使用機(jī)器學(xué)習(xí)程序決定計(jì)算機(jī)芯片的電路布局,最終設(shè)計(jì)與人類(lèi)芯片設(shè)計(jì)人員相比,其敏銳度相當(dāng)甚至更高。
當(dāng)“布線”任務(wù)中,芯片設(shè)計(jì)人員通常會(huì)使用軟件來(lái)確定芯片中的電路布局,有點(diǎn)類(lèi)似于建筑物設(shè)計(jì)平面圖。為了找到滿足多個(gè)目標(biāo)的最佳布局,需要考慮很多因素,包括提供芯片性能,同時(shí)還要避免不必要的復(fù)雜性,否則可能增加芯片制造成本。這種平衡需要大量的人類(lèi)啟發(fā)式思維,以最佳方式進(jìn)行設(shè)計(jì)?,F(xiàn)在,人工智能算法也能夠以這種啟發(fā)式思維方式進(jìn)行設(shè)計(jì)。
Dean舉例說(shuō),深度學(xué)習(xí)神經(jīng)網(wǎng)絡(luò)只花了24個(gè)小時(shí)就解決了問(wèn)題,而人類(lèi)設(shè)計(jì)是需要6至8周,并且前者的解決方案更好。這減少了芯片總布線數(shù)量,從而提高了效率。
Dean向參會(huì)者表示,用于解決這個(gè)問(wèn)題的機(jī)器學(xué)習(xí)模型僅用24小時(shí)就提出了一種芯片設(shè)計(jì)方案,而人類(lèi)設(shè)計(jì)師用了8周時(shí)間才完成。
這個(gè)深度學(xué)習(xí)程序類(lèi)似于Google DeepMind部門(mén)為征服Go游戲開(kāi)發(fā)的AlphaZero程序,并且也是一種強(qiáng)化學(xué)習(xí)。為了實(shí)現(xiàn)目標(biāo),該程序嘗試了各種步驟來(lái)查看哪些步驟可以帶來(lái)更好的結(jié)果,只不過(guò)不是下棋,而是在芯片中設(shè)計(jì)最優(yōu)的電路布局。
與Go不同的是,這個(gè)解決方案的“空間”(布線數(shù)量)要大得多,而且如上所述必須滿足很多要求,不是僅僅贏得比賽這一個(gè)目標(biāo)。
Dean表示,這個(gè)內(nèi)部研究還處于了解深度學(xué)習(xí)技術(shù)的早期階段?!拔覀冋谧屛覀兊脑O(shè)計(jì)師進(jìn)行試驗(yàn),看看如何開(kāi)始在工作流程中使用該程序。另外我們正在試圖了解該程序的用處以及它在哪些方面可以改進(jìn)?!?/p>
谷歌進(jìn)軍AI設(shè)計(jì)領(lǐng)域正處于芯片生產(chǎn)復(fù)興之時(shí),旨在讓不同大小的專(zhuān)用芯片都可以更快地運(yùn)行機(jī)器學(xué)習(xí)。有機(jī)器學(xué)習(xí)科學(xué)家認(rèn)為,專(zhuān)用AI硬件可以催生更大型、更高效的機(jī)器學(xué)習(xí)軟件項(xiàng)目。
Dean說(shuō),即使谷歌擴(kuò)展了AI設(shè)計(jì)項(xiàng)目,仍然會(huì)有很多AI硬件初創(chuàng)公司(例如Cerebras Systems和Graphcore)給市場(chǎng)帶來(lái)多樣性,并迅速發(fā)展起來(lái)。并表示這種多樣性會(huì)很有意思。
“我不確定這些初創(chuàng)公司是否能夠在市場(chǎng)中生存下去,但這很有意思,因?yàn)槠渲泻芏喑鮿?chuàng)公司采取了截然不同的設(shè)計(jì)方式。有些加速的模型很小,可以用于片上SRAM?!边@意味著,機(jī)器學(xué)習(xí)模型可能非常小,不需要外置內(nèi)存。
“如果你的模型可用于SRAM,那么會(huì)變得非常高效,但如果不適合,那就不是你應(yīng)該選擇的芯片?!?/p>
谷歌稱(chēng),這個(gè)機(jī)器學(xué)習(xí)程序創(chuàng)造了很多連人類(lèi)設(shè)計(jì)師都沒(méi)有想到的新穎電路設(shè)計(jì)。
當(dāng)被問(wèn)及這些芯片是否會(huì)融入某些標(biāo)準(zhǔn)設(shè)計(jì)的時(shí)候,Dean暗示,多樣化是很有可能的,至少就目前來(lái)看?!拔掖_實(shí)認(rèn)為大家使用的方法會(huì)各有千秋,因?yàn)楫?dāng)前有關(guān)機(jī)器學(xué)習(xí)的研究呈現(xiàn)爆炸式增長(zhǎng),機(jī)器學(xué)習(xí)被用于解決各種各樣的問(wèn)題,當(dāng)有如此多選擇的時(shí)候,你肯定不想只盯著一個(gè)選擇,而是想要五個(gè)或者六個(gè)——不會(huì)是一千個(gè),但得有五個(gè)或六個(gè)不同的設(shè)計(jì)點(diǎn)。”
Dean補(bǔ)充說(shuō):“哪些設(shè)計(jì)方法會(huì)脫穎而出,這一點(diǎn)令人期待,不管是能夠解決很多問(wèn)題的通用方法,還是加速某個(gè)方面的專(zhuān)業(yè)化方法?!?/p>
談到谷歌除了TPU之外的舉措, Dean表示,谷歌正在嘗試越來(lái)越多的專(zhuān)用芯片。當(dāng)被問(wèn)及谷歌AI硬件是否可能延伸到其現(xiàn)有產(chǎn)品之外時(shí),Dean回答說(shuō):“哦,是的?!?/p>
“毫無(wú)疑問(wèn),機(jī)器學(xué)習(xí)正在被越來(lái)越廣泛地用于谷歌產(chǎn)品中,這些產(chǎn)品既包括基于數(shù)據(jù)中心的服務(wù),也有很多手機(jī)上的產(chǎn)品?!盌ean指出,Google Translate就是一個(gè)擺脫了復(fù)雜性的程序,該程序現(xiàn)在支持七十種不同的語(yǔ)言,即使是在飛行模式下也可以在手機(jī)上使用。
Dean指出,谷歌擴(kuò)充了面向AI的芯片家族。例如Edge TPU就涵蓋了“不同的設(shè)計(jì)點(diǎn)”,包括低功耗應(yīng)用,以及數(shù)據(jù)中心核心的高性能應(yīng)用。當(dāng)被問(wèn)及谷歌是否會(huì)進(jìn)一步擴(kuò)大多樣性的時(shí)候,Dean回答說(shuō):“我認(rèn)為會(huì)的?!?/p>
Dean表示,“即使在非數(shù)據(jù)中心領(lǐng)域,你會(huì)也看到不同高功率環(huán)境——例如自動(dòng)駕駛汽車(chē)——之間的區(qū)別,不一定是1瓦,有可能是50或100瓦。因此,針對(duì)這種環(huán)境和針對(duì)手機(jī)環(huán)境,你需要不同的方法。”同時(shí),還有農(nóng)業(yè)傳感器之類(lèi)的超低功耗應(yīng)用,這些應(yīng)用程序可以執(zhí)行一些AI處理,而無(wú)需將任何數(shù)據(jù)發(fā)送到云中。這種傳感器如果支持AI就可以評(píng)估是否收集了(例如攝像頭的)任何數(shù)據(jù),并將這些單獨(dú)的數(shù)據(jù)點(diǎn)發(fā)送回云端進(jìn)行分析。
評(píng)論