在AI訓(xùn)練市場(chǎng)，不乏躍躍欲試想挑戰(zhàn)英偉達(dá)霸主地位的廠商。不過(guò)，英偉達(dá)GPU仍是當(dāng)前全球大規(guī)模商用部署的頭號(hào)玩家。其次，Google的TPU通過(guò)內(nèi)部應(yīng)用及TensorFlow占據(jù)第二大生態(tài)規(guī)模。

要知道，一顆AI芯片從開(kāi)發(fā)定義到落地部署，中間存在著巨大的鴻溝，特別是算法越來(lái)越復(fù)雜、模型越來(lái)越大，AI芯片面臨著算力的嚴(yán)峻考驗(yàn)，最終要在數(shù)據(jù)中心批量部署，能夠成功的廠商鳳毛麟角。

不過(guò)，來(lái)自于底層的顛覆性創(chuàng)新正在悄然改變著格局。Graphcore，這家成立于2016年、來(lái)自于英國(guó)的AI芯片公司，通過(guò)創(chuàng)新的IPU處理器技術(shù)，已經(jīng)開(kāi)始在全球數(shù)據(jù)中心批量應(yīng)用，躋身于該市場(chǎng)第三梯隊(duì)。

5月27日，在Intelligent Health峰會(huì)上，微軟機(jī)器學(xué)習(xí)科學(xué)家Sujeeth Bharadwaj分享了在攻克新冠病毒時(shí)的一項(xiàng)研究，在訓(xùn)練CXR（胸部X射線檢查）模型時(shí)，用Graphcore IPU處理器和英偉達(dá) V100同時(shí)運(yùn)行微軟COVID-19影像分析算法SONIC，最終的結(jié)果可能令所有人大跌眼鏡：IPU在30分鐘內(nèi)完成了V100需5個(gè)小時(shí)的訓(xùn)練工作量！

這家年輕的公司，由此再次引起了業(yè)界的關(guān)注。日前，Graphcore面對(duì)<電子發(fā)燒友>等行業(yè)媒體，首次在中國(guó)市場(chǎng)全面揭開(kāi)了其創(chuàng)新背后的核心技術(shù)及最新業(yè)務(wù)進(jìn)展，以及在中國(guó)市場(chǎng)的布局等。

為什么傳統(tǒng)的處理器架構(gòu)需要被顛覆？

Graphcore高級(jí)副總裁兼中國(guó)區(qū)總經(jīng)理盧濤（Jason Lu）介紹稱，AI時(shí)代的機(jī)器智能代表的是全新的計(jì)算負(fù)載，不同于傳統(tǒng)計(jì)算的特點(diǎn)有：它是非常大規(guī)模的并行計(jì)算；數(shù)據(jù)結(jié)構(gòu)非常稀疏；相較于傳統(tǒng)的科學(xué)計(jì)算或高性能計(jì)算（HPC），AI/機(jī)器智能是低精度計(jì)算；另外在訓(xùn)練、推理過(guò)程中的數(shù)據(jù)參數(shù)復(fù)用、靜態(tài)圖結(jié)構(gòu)等，都是AI應(yīng)用全新計(jì)算負(fù)載的典型代表。

整個(gè)AI算法模型的演變，基本上從2016年1月份的ResNet50的2500萬(wàn)個(gè)參數(shù)，發(fā)展到2018年10月份BERT-Large的3.3億個(gè)參數(shù)，而到了2019年發(fā)展到GPT2的15.5億個(gè)參數(shù)，增長(zhǎng)幅度非常大。甚至，現(xiàn)在一些領(lǐng)先的科研機(jī)構(gòu)和AI研究者在探索更大的算法模型，能夠訓(xùn)練更復(fù)雜的算法，來(lái)提高精度。密集計(jì)算并不是可持續(xù)的方法，譬如算法模型參數(shù)要從15.5億規(guī)模擴(kuò)展到一萬(wàn)億，這種指數(shù)級(jí)的增長(zhǎng)，需要成倍的算力提升。Graphcore認(rèn)為，傳統(tǒng)處理器無(wú)法很好地應(yīng)對(duì)這些變化，因此市場(chǎng)需要一種顛覆式的創(chuàng)新架構(gòu)。
?
傳統(tǒng)的處理器架構(gòu)，如CPU是針對(duì)應(yīng)用和網(wǎng)絡(luò)進(jìn)行設(shè)計(jì)的標(biāo)量處理器，GPU是以向量處理為核心的、針對(duì)圖形和高性能計(jì)算的處理器。而AI是全新的應(yīng)用架構(gòu)，底層是以計(jì)算圖作為表征的，且從整個(gè)AI發(fā)展方向來(lái)看，大規(guī)模、稀疏化的數(shù)據(jù)會(huì)越來(lái)越多，因此，Graphcore針對(duì)這些發(fā)展趨勢(shì)設(shè)計(jì)了一種全新的處理器架構(gòu)。

全世界最復(fù)雜的擁有236億個(gè)晶體管的芯片處理器

目前為止，機(jī)器學(xué)習(xí)的算力來(lái)源主要還是傳統(tǒng)的處理器，它們的算力提升也非常快。不過(guò)，峰值算力和有效算力是兩回事，這其中，內(nèi)存帶寬成為掣肘。

當(dāng)處理器算力提高了10倍，內(nèi)存如何相應(yīng)提高10倍的性能呢?盧濤介紹，如果用傳統(tǒng)的DDR4、DDR5、HBM、HBM1、HBM2、HBM3等內(nèi)存，基本上每一代能有30%或40%的提升，因此，這對(duì)傳統(tǒng)架構(gòu)是一個(gè)非常大的挑戰(zhàn)。

相較于傳統(tǒng)CPU、GPU，IPU采用了大規(guī)模并行MIMD（多指令多數(shù)據(jù)）處理器核，通過(guò)緊密耦合的大型本地分布式SRAM，在片內(nèi)能夠做到300MB SRAM。相對(duì)CPU的DDR2子系統(tǒng)或GPU的GDDR、HBM來(lái)說(shuō)，IPU能夠?qū)崿F(xiàn)10到320倍的性能提升。這樣帶來(lái)的好處是，能夠?qū)⒛Ｐ秃蛿?shù)據(jù)放在片內(nèi)處理，從時(shí)延的角度來(lái)看，與訪問(wèn)外存相比較，時(shí)延僅為1%。

通過(guò)采用大規(guī)模分布式的片上SRAM架構(gòu)，IPU處理器將所有memory都放在片上，解決了當(dāng)前機(jī)器學(xué)習(xí)中大量出現(xiàn)的內(nèi)存帶寬所造成的瓶頸。

目前，基于這一創(chuàng)新架構(gòu)的IPU處理器GC2已量產(chǎn)，采用TSMC 16nm工藝，號(hào)稱是世界上最復(fù)雜的擁有236億個(gè)晶體管的芯片處理器。

GC2片內(nèi)有1216個(gè)IPU-Tiles，每個(gè)Tile有獨(dú)立的IPU核心作為計(jì)算以及In-Processor-Memory（處理器之內(nèi)的內(nèi)存），總共有7296個(gè)線程，能夠支持7296個(gè)程序并行運(yùn)行。對(duì)整片來(lái)說(shuō)，In-Processor-Memory總共是300MB，PCIe是16個(gè)PCIe Gen 4。

而在各個(gè)核心之間，Graphcore通過(guò)BSP同步協(xié)議，能夠支持同一個(gè)IPU處理器內(nèi)1216個(gè)核心之間的通信，以及跨不同的IPU之間進(jìn)行通信。另外，在IPU和IPU之間，擁有80個(gè)IPU-Links，總共有320GB/s的chip to chip的帶寬。正因如此，IPU處理器可以同時(shí)支持訓(xùn)練和推理。

從目前所公布的指標(biāo)來(lái)看，在自然語(yǔ)言處理、圖像分類、金融模型訓(xùn)練等方面，IPU在現(xiàn)有及下一代的模型上，性能均領(lǐng)先于GPU：在自然語(yǔ)言處理方面的速度能夠提升20%到50%；在圖像分類方面，能夠有6倍的吞吐量且時(shí)延更低；在金融模型方面，訓(xùn)練速度能夠提高26倍以上。目前，IPU在云上、在一些客戶的自建數(shù)據(jù)中心的服務(wù)器上已經(jīng)投產(chǎn)應(yīng)用。

而在場(chǎng)景應(yīng)用方面，IPU采用分組卷積的方式也體現(xiàn)出了獨(dú)有優(yōu)勢(shì)，特別是針對(duì)更為稀疏化的數(shù)據(jù)時(shí)。

盧濤分享了眾多AI創(chuàng)新者、算法科學(xué)家、AI應(yīng)用開(kāi)發(fā)者日常工作中遇到的一大問(wèn)題：當(dāng)算法模型在GPU上運(yùn)行速度非常慢的時(shí)候，通常被認(rèn)為是算法或軟件問(wèn)題。他指出，如果算法模型不是用稠密的卷積，而是用較為稀疏的卷積比如Fully depthwise做的，那么在GPU上運(yùn)行得慢的根本原因是GPU架構(gòu)不符合算法特點(diǎn)，因此采用IPU能夠提供更好的支持。

他解釋稱，Graphcore設(shè)計(jì)了一個(gè)分組卷積內(nèi)核的micro-benchmark，將組維度（group dimension）分成從1到512來(lái)比較，這里512就是應(yīng)用得較多的“Dense卷積網(wǎng)絡(luò)”，典型應(yīng)用如ResNet。此時(shí)，IPU GC2性能甚至比英偉達(dá)V100要高近一倍。隨著稠密程度降低、稀疏化程度增加，在組維度為1或32時(shí)，針對(duì)EfficientNet或MobileNet，IPU對(duì)比GPU展現(xiàn)出巨大優(yōu)勢(shì)，做到成倍的性能提升，同時(shí)時(shí)延大大降低。

創(chuàng)新架構(gòu)需要軟硬協(xié)同設(shè)計(jì)

IPU所采用的片上存儲(chǔ)架構(gòu)，確實(shí)是未來(lái)計(jì)算結(jié)構(gòu)的發(fā)展方向之一，但從芯片設(shè)計(jì)和應(yīng)用角度而言，這是一大挑戰(zhàn)。

片上存儲(chǔ)通常有兩種架構(gòu)，一是在片上規(guī)劃單塊大規(guī)模的存儲(chǔ)，這種方式通常會(huì)導(dǎo)致良品率極低。另一種架構(gòu)就是Graphcore這樣的分布式片上存儲(chǔ)架構(gòu)。但這又帶來(lái)了新的挑戰(zhàn)：如何把分布式存儲(chǔ)架構(gòu)有效利用起來(lái)？這對(duì)編譯器的要求非常高，可以說(shuō)是軟件、硬件協(xié)同設(shè)計(jì)的過(guò)程。要做出能夠真正落地的產(chǎn)品，最核心的挑戰(zhàn)就是軟硬件兩方面的專業(yè)知識(shí)和經(jīng)驗(yàn)。

為了提升芯片的可用性，以及便于用戶和開(kāi)發(fā)者更方便地在系統(tǒng)中進(jìn)行開(kāi)發(fā)、移植、優(yōu)化，Graphcore將產(chǎn)品擴(kuò)展到囊括龐大的部署軟件和基礎(chǔ)架構(gòu)套件，通過(guò)Poplar SDK給用戶提供更好的體驗(yàn)。而這通常是頭部廠商如英偉達(dá)在推進(jìn)GPU大規(guī)模應(yīng)用時(shí)才有的舉動(dòng)。

Poplar SDK是架構(gòu)在機(jī)器學(xué)習(xí)上的框架軟件（比如TensorFlow、ONNX、PyTorch和PaddlePaddle）和硬件之間的一個(gè)基于計(jì)算圖的整套工具鏈和庫(kù)。Poplar SDK支持容器化部署，能夠快速啟動(dòng)并運(yùn)行。在標(biāo)準(zhǔn)生態(tài)方面，Poplar SDK支持Docker、Kubernetes、以及微軟的Hyper-v等虛擬化技術(shù)和安全技術(shù)。在操作系統(tǒng)方面，Poplar SDK目前支持最主要的三個(gè)Linux發(fā)行版：ubuntu、RedHat Enterprise Linux、CentOS。

今年5月，Graphcore還推出了PopVision Graph Analyser分析工具，用戶可以通過(guò)這個(gè)可視化的圖形展示工具來(lái)分析軟件運(yùn)行情況、調(diào)試效率。

目前基于IPU的一些應(yīng)用已覆蓋到機(jī)器學(xué)習(xí)的各個(gè)應(yīng)用領(lǐng)域，包括自然語(yǔ)言處理、圖像/視頻處理、時(shí)序分析、推薦/排名及概率模型。一些應(yīng)用案例和模型已經(jīng)在TensorFlow、ONNX和Graphcore的PopART上可用，所有源代碼都可以在GitHub處下載。

新冠疫情下“小兵”立大功

當(dāng)前，全球都希望更高效地攻克新冠疫情中的難題。這時(shí)，AI在高清醫(yī)學(xué)影像領(lǐng)域就體現(xiàn)出了重要價(jià)值。

第一，疫情發(fā)展非常快，不斷有新的病例、影像和數(shù)據(jù)產(chǎn)生，這就要求現(xiàn)有的模型要不斷根據(jù)新的情況來(lái)提高精度。

第二，疫情造成全球醫(yī)療資源緊缺。放射影片往往需要富有經(jīng)驗(yàn)的醫(yī)生進(jìn)行判斷，而在資源緊張的情況下，AI工具可以幫助更多醫(yī)生獲得更專業(yè)的判斷力。

第三，全球都迫切需要攻克病毒的研究成果，如何提高研究效率至關(guān)重要。

Graphcore中國(guó)銷(xiāo)售總監(jiān)朱江，就本文開(kāi)頭所提到的微軟訓(xùn)練CXR（胸部X光射線檢查）的應(yīng)用案例，詳細(xì)介紹了IPU與英偉達(dá) V100的對(duì)比情況。

微軟專門(mén)開(kāi)發(fā)了SONIC CV模型進(jìn)行訓(xùn)練，IPU和GPU的訓(xùn)練結(jié)果對(duì)比如下圖：左邊是訓(xùn)練時(shí)間，IPU優(yōu)勢(shì)明顯。右邊紅色曲線代表訓(xùn)練時(shí)精度上升的情況，藍(lán)色曲線代表測(cè)試精度。可以看到測(cè)試精度和訓(xùn)練精度較為接近甚至吻合，這也說(shuō)明SONIC模型在泛化性能上更好，在針對(duì)未知的新數(shù)據(jù)方面，其處理能力比微軟傳統(tǒng)的EfficientNet模型更好。整體上，SONIC的模型通過(guò)30分鐘的訓(xùn)練達(dá)到94%的訓(xùn)練精度和測(cè)試精度，訓(xùn)練速度方面，IPU需要30分鐘，而GPU差不多需要5個(gè)小時(shí)。

通過(guò)這一訓(xùn)練，微軟認(rèn)為能夠訓(xùn)練到SOTA的精度的模型不一定是大模型，可以用小模型來(lái)達(dá)到這樣的精度要求。另外，IPU的MIMD架構(gòu)非常適用于以分組卷積為代表的新模型。

據(jù)了解，目前微軟已采用IPU來(lái)進(jìn)行計(jì)算機(jī)視覺(jué)中分類方面的訓(xùn)練，能達(dá)到一個(gè)數(shù)量級(jí)的速度提升。未來(lái)，微軟期望把IPU在CV領(lǐng)域的應(yīng)用擴(kuò)展到更多方面，包括監(jiān)測(cè)、分割以及配準(zhǔn)。

創(chuàng)新帶給Graphcore的底氣

迄今為止，Graphcore獲得了AI領(lǐng)域多位重量級(jí)人物的背書(shū)。

英國(guó)半導(dǎo)體之父、Arm聯(lián)合創(chuàng)始人Hermann爵士認(rèn)為:“在計(jì)算機(jī)歷史上只發(fā)生過(guò)三次革命，一次是70年代的CPU，第二次是90年代的GPU，而Graphcore就是第三次革命。”意指其率先提出了為AI計(jì)算而生的IPU。

AI教父Geoff Hinton教授在接受Wired采訪時(shí)，被問(wèn)到 “我們應(yīng)該如何構(gòu)建功能更像大腦的機(jī)器學(xué)習(xí)系統(tǒng)”時(shí)，從錢(qián)包中掏出了一個(gè)又大又亮的硅片，并回答說(shuō)：“我們需要轉(zhuǎn)向不同類型的計(jì)算機(jī)來(lái)處理新的機(jī)器學(xué)習(xí)系統(tǒng)。”他認(rèn)為Graphcore的IPU正在滿足這樣的系統(tǒng)需求。


迄今為止，Graphcore總?cè)谫Y超過(guò)4.5億美金，其中包括全球知名的金融投資者和戰(zhàn)略投資者。

不論是技術(shù)本身所帶來(lái)的創(chuàng)新地位，還是大佬的站臺(tái)或融資歷程，Graphcore的履歷都堪稱漂亮。

不過(guò)，作為一家初創(chuàng)企業(yè)，Graphcore直面的都是業(yè)界巨擘。強(qiáng)如英偉達(dá)，也已經(jīng)感受了種種威脅，正在加速創(chuàng)新。上個(gè)月，英偉達(dá)推出了基于Ampere架構(gòu)的NVIDIA A100，將AI訓(xùn)練和推理性能提高20倍，可以說(shuō)是英偉達(dá)GPU迄今為止最大的性能飛躍。

對(duì)于未來(lái)的競(jìng)爭(zhēng)，Graphcore方面信心滿滿。盧濤表示，雖然目前對(duì)比的都是與V100這樣的大量部署的旗艦級(jí)產(chǎn)品，但即使是第一代IPU產(chǎn)品也不會(huì)輸于A100，且下一代IPU處理器也將有重磅發(fā)布。

未來(lái)的推進(jìn)策略，Graphcore還是會(huì)在訓(xùn)練和推理兩方面并行，聚焦對(duì)高精度、低時(shí)延、高吞吐量要求更高的場(chǎng)景。另外還有一個(gè)趨勢(shì)是訓(xùn)練和推理混布的需求，例如視頻平臺(tái)、電商網(wǎng)站等希望通過(guò)算法同時(shí)進(jìn)行訓(xùn)練和推理，能夠根據(jù)用戶數(shù)據(jù)實(shí)時(shí)更新算法模型；未來(lái)的汽車(chē)應(yīng)用也是訓(xùn)練和推理混布的場(chǎng)景，都將有一定的增長(zhǎng)。

積極擁抱中國(guó)AI生態(tài)圈

在中國(guó)，Graphcore剛與兩大頭部客戶有了重大進(jìn)展。

一是阿里巴巴新的開(kāi)放式深度學(xué)習(xí)API ODLA（Open Deep Learning API）支持Graphcore IPU，某種程度上，這也反映了數(shù)據(jù)中心對(duì)IPU的計(jì)算需求正在增長(zhǎng)。

二是成為百度飛槳（PaddlePaddle）硬件生態(tài)圈共建計(jì)劃伙伴之一，這一合作使Graphcore進(jìn)入了中國(guó)深度學(xué)習(xí)開(kāi)源框架的生態(tài)系統(tǒng)中，觸及百萬(wàn)以上的AI開(kāi)發(fā)者。

盧濤表示，Graphcore正在積極擁抱中國(guó)的AI生態(tài)圈，中國(guó)市場(chǎng)未來(lái)有望占據(jù)其全球市場(chǎng)的40%甚至50%。

本文由電子發(fā)燒友網(wǎng)原創(chuàng)，未經(jīng)授權(quán)禁止轉(zhuǎn)載。如需轉(zhuǎn)載，請(qǐng)?zhí)砑游?a target="_blank">信號(hào)elecfans999.