英特爾、Habana Labs與Hugging Face推動(dòng)Transformer業(yè)務(wù)在訓(xùn)練和推理優(yōu)化及擴(kuò)展取得關(guān)鍵進(jìn)展

在過(guò)去的一年中，英特爾、Habana Labs和Hugging Face基于開源項(xiàng)目、集成開發(fā)者體驗(yàn)與科學(xué)研究，不斷提升人工智能應(yīng)用的效率并降低使用門檻，在創(chuàng)建和訓(xùn)練高質(zhì)量Transformer模型上取得了重大進(jìn)展。

Transformer模型為包括自然語(yǔ)言處理（NLP）、計(jì)算機(jī)視覺（CV）、語(yǔ)音等在內(nèi)廣泛的機(jī)器學(xué)習(xí)和深度學(xué)習(xí)任務(wù)提供先進(jìn)的性能。大規(guī)模訓(xùn)練這些深度學(xué)習(xí)模型需要龐大的算力，這個(gè)過(guò)程非常復(fù)雜，不僅需要大量時(shí)間，而且成本高昂。

通過(guò)英特爾顛覆者計(jì)劃（Intel Disruptor Program）與Hugging Face密切合作，能夠幫助用戶更廣泛地采用基于最新英特爾^?至強(qiáng)^?可擴(kuò)展處理器、Habana Gaudi^?以及Gaudi^?2處理器優(yōu)化的訓(xùn)練和推理解決方案。此次合作將英特爾AI工具包中先進(jìn)的深度學(xué)習(xí)創(chuàng)新技術(shù)引入Hugging Face的開源生態(tài)系統(tǒng)中，并以此推動(dòng)未來(lái)英特爾^?架構(gòu)的創(chuàng)新發(fā)展，在英特爾至強(qiáng)平臺(tái)上的分布式調(diào)優(yōu)、內(nèi)置優(yōu)化、配合Habana Gaudi加速訓(xùn)練，以及小樣本學(xué)習(xí)方面均取得了重大進(jìn)展。

當(dāng)使用單節(jié)點(diǎn)CPU進(jìn)行訓(xùn)練速度不佳時(shí)，數(shù)據(jù)科學(xué)家們就需要分布式訓(xùn)練。在分布式訓(xùn)練中，集群中的每臺(tái)服務(wù)器都保留一個(gè)模型副本，利用訓(xùn)練數(shù)據(jù)集的一部分進(jìn)行訓(xùn)練，并通過(guò)英特爾^?oneAPI集體通信庫(kù)（Collective Communications Library）在各節(jié)點(diǎn)之間交換結(jié)果，從而更快地收斂到最終模型。目前，Transformer可原生支持該功能，并使數(shù)據(jù)科學(xué)家們更容易地進(jìn)行分布式調(diào)優(yōu)。

例如，在英特爾至強(qiáng)可擴(kuò)展處理器的分布式集群上加速Transformer模型的PyTorch訓(xùn)練時(shí)，為在PyTorch中能夠有效利用英特爾^?高級(jí)矩陣擴(kuò)展（英特爾^?AMX）、AVX-512以及英特爾矢量神經(jīng)網(wǎng)絡(luò)指令（VNNI）等最新英特爾至強(qiáng)可擴(kuò)展處理器所支持的硬件性能，英特爾為PyTorch設(shè)計(jì)了英特爾擴(kuò)展，該軟件庫(kù)可為訓(xùn)練和推理提供開箱即用的加速功能。

此外，Hugging Face Transformer提供Trainer API，使用戶可以無(wú)需手動(dòng)編寫訓(xùn)練循環(huán)，更輕松地開始訓(xùn)練。該Trainer為超參數(shù)搜索提供API，目前可支持多個(gè)搜索后端，其中包括可托管的超參數(shù)優(yōu)化服務(wù)英特爾SigOpt。得益于此，數(shù)據(jù)科學(xué)家們可以更有效地訓(xùn)練并獲取最佳模型。

非凡的開發(fā)者體驗(yàn)

Optimum是Hugging Face創(chuàng)建的一個(gè)開源庫(kù)，用于在日益增長(zhǎng)的訓(xùn)練及推理設(shè)備中簡(jiǎn)化Transformer的加速。通過(guò)其內(nèi)置的優(yōu)化技術(shù)和現(xiàn)成的腳本，初學(xué)者可以輕松地上手使用Optimum，而專家則可以通過(guò)不斷調(diào)整以獲得最佳性能。

“Optimum Intel”是Transformer庫(kù)與英特爾所提供的不同工具和庫(kù)之間的接口，用于加速英特爾架構(gòu)上的端到端管線。該接口基于英特爾^?神經(jīng)壓縮器所開發(fā)，為包括量化、剪枝、知識(shí)提取等多項(xiàng)網(wǎng)絡(luò)壓縮技術(shù)提供跨多個(gè)深度學(xué)習(xí)框架的統(tǒng)一體驗(yàn)。此外，開發(fā)人員亦可使用Optimum Intel來(lái)進(jìn)行針對(duì)評(píng)估數(shù)據(jù)集的模型指標(biāo)對(duì)比，從而更加輕松地在Transformer模型上運(yùn)行訓(xùn)練后量化（PTQ）。

與此同時(shí)，Optimum Intel還提供了一個(gè)簡(jiǎn)單的接口來(lái)優(yōu)化Transformer模型，并將模型轉(zhuǎn)換為OpenVINO的中間層表示（IR），從而使用OpenVINO進(jìn)行推理。

利用Habana Gaudi加速訓(xùn)練

現(xiàn)階段，Habana Labs正攜手Hugging Face更簡(jiǎn)易、快速地訓(xùn)練大規(guī)模、高質(zhì)量的Transformer模型。得益于Habana的SynapseAI? 軟件套件與Hugging Face Optimum-Habana開源庫(kù)，數(shù)據(jù)科學(xué)家和機(jī)器學(xué)習(xí)工程師能夠通過(guò)在Habana Gaudi和Habana Gaudi2處理器上運(yùn)行幾行代碼，加速Transformer深度學(xué)習(xí)的訓(xùn)練。

Optimum-Habana庫(kù)支持各種計(jì)算機(jī)視覺、自然語(yǔ)言和多模態(tài)模型。其支持且經(jīng)過(guò)測(cè)試的模型架構(gòu)包括BERT、AlBERT、DistilBERT、RoBERTa、Vision Transformer、swin、T5、GPT2、wav2vec2和Stable Diffusion。Hugging Facehub上目前已有4萬(wàn)多個(gè)基于這些架構(gòu)的模型，而開發(fā)人員可以使用Optimum-Habana在Gaudi和Gaudi2上輕松地使用這些模型¹。

Habana Gaudi解決方案已經(jīng)用于亞馬遜EC2 DL1實(shí)例，采用該解決方案進(jìn)行訓(xùn)練的一個(gè)主要優(yōu)勢(shì)是性價(jià)比。Habana Gaudi的性價(jià)比與同類訓(xùn)練解決方案相比高40%，使客戶能用更少的成本進(jìn)行更多訓(xùn)練²，Gaudi2采用與第一代Gaudi相同的高效架構(gòu)，同樣提供了卓越的性價(jià)比。

Habana DeepSpeed也集成在Optimum-Habana庫(kù)中，讓人們?cè)谑褂?/span>DeepSpeed優(yōu)化的Gaudi設(shè)備時(shí)，能更易于大規(guī)模部署和訓(xùn)練大型語(yǔ)言模型。您可以通過(guò)Optimum Habana DeepSpeed使用指南了解更多信息。

最新版的Optimum-Habana在Hugging Facediffusers庫(kù)中內(nèi)置支持Stable Diffusion，使Hugging Face開發(fā)者能夠在Habana Gaudi上進(jìn)行極具性價(jià)比的圖像生成測(cè)試。

生產(chǎn)中的小樣本學(xué)習(xí)

英特爾研究院、Hugging Face和UKP Lab最近推出了SetFit，這是一種用于對(duì)Sentence Transformer進(jìn)行小樣本調(diào)優(yōu)的有效框架。使用預(yù)先訓(xùn)練的語(yǔ)言模型進(jìn)行小樣本學(xué)習(xí)，將有望解決數(shù)據(jù)科學(xué)家在現(xiàn)實(shí)中面臨的一大挑戰(zhàn)：處理那些幾乎沒有標(biāo)簽的數(shù)據(jù)。

當(dāng)前的小樣本調(diào)優(yōu)需要手工提示或描述器，將示例轉(zhuǎn)換為適合底層語(yǔ)言模型的格式。通過(guò)直接從少量有標(biāo)簽的文本示例中直接生成豐富的嵌入，SetFit可省去提示。

研究人員設(shè)計(jì)了SetFit，可用于Hugging Face Hub上的任何Sentence Transformer，即通過(guò)調(diào)優(yōu)多語(yǔ)言檢查點(diǎn)，可以將文本分類為多種語(yǔ)言。

SetFit不需要像T5或GPT-3這樣的大模型來(lái)實(shí)現(xiàn)高精度。與標(biāo)準(zhǔn)調(diào)優(yōu)相比，它顯著提高了采樣效率并能夠更好地耐受噪聲。例如，對(duì)于在一個(gè)示例情感數(shù)據(jù)集上每類只有八個(gè)有標(biāo)簽的例子，SetFit可以與在包含3000個(gè)例子的完整訓(xùn)練集上的RoBERTa Large調(diào)優(yōu)相媲美。Hugging Face發(fā)現(xiàn)，在零提示且體積縮小27倍的情況下，SetFit也取得了與T-Few 3B相當(dāng)?shù)男Ч?，從而?shí)現(xiàn)了兼具低成本和高效的訓(xùn)練³。

一直以來(lái)，英特爾致力于積極構(gòu)建生態(tài)系統(tǒng)并助力降低AI成本，包括開源項(xiàng)目、集成的開發(fā)者體驗(yàn)和科學(xué)研究等舉措。而工具和軟件恰恰能夠讓開發(fā)人員加快構(gòu)建應(yīng)用程序，并釋放處理器性能。英特爾旨在讓人們能夠更輕松地在任何地方構(gòu)建和部署AI，使數(shù)據(jù)科學(xué)家和機(jī)器學(xué)習(xí)從業(yè)者能夠采用最新的優(yōu)化技術(shù)。

注釋：

¹ 基于Hugging Face搜索查詢：https://huggingface.co/models?other=or:bert,stable-diffusion-diffusers,albert,wav2vec2,vit,swin,t5,distilbert,roberta,gpt2。這個(gè)查詢會(huì)過(guò)濾所有已經(jīng)測(cè)試、驗(yàn)證并在Optimum-Habana中得到支持的Hugging Face模型。
² 基于AWS計(jì)算的價(jià)格和性能，對(duì)比了基于GPU的類似AWS實(shí)例。更多信息參見亞馬遜相關(guān)新聞稿：https://press.aboutamazon.com/2021/10/aws-announces-general-availability-of-amazon-ec2-dl1-instances 
³有關(guān)性能的詳情參見完整論文：https://arxiv.org/abs/2209.11055 和博客文章：https://huggingface.co/blog/set