英偉達打破實時對話AI的訓練和推理用時記錄
據國外媒體報導,Nvidia用於開發和運行可理解和響應請求的對話式AI的GPU強化平台,已經達成了一些重要的里程碑,並打破了一些記錄。這對任何基於其技術進行開發的人來說意義重大——當中包括大大小小的公司,因為英偉達將很多用於實現這些進步的代碼開源。那些代碼用PyTorch編寫,易於運行。
英偉達今天宣布的最大成果包括:BERT訓練用時打破記錄,進入1小時關口。BERT是世界上最先進的人工智能語言模型之一,也被廣泛認為是自然語言處理(NLP)標杆的先進模型。
Nvidia的AI平台僅僅需要53分鐘就能夠完成模型訓練,經過訓練的模型在短短2毫秒多一點的時間裡(10毫秒在業內被認為是高水平線)就能夠成功地做出推理(即利用通過訓練學習到的能力得出結果)——又一個記錄。
(Nvidia DGX SuperPOD,圖自:英偉達開發者博客)
英偉達的這些突破不僅僅是吹噓的資本——這些進步可為任何使用NLP對話式人工智能和GPU硬件的人提供切實的好處。
Nvidia在它的其中一個SuperPOD系統上打破了訓練時間記錄,該SuperPOD系統由運行1472個V100 GPU的92個Nvidia DGX-2H系統組成;在運行Nvidia TensorRT的Nvidia T4 GPU上完成推理過程。
Nvidia TensorRT在性能上超出高度優化的CPU許多個量級。不過,該公司將公開BERT訓練代碼和經過TensorRT優化的BERT樣本,讓所有人都可以通過GitHub利用。
(via Neowin)
除了這些里程碑以外,英偉達的研究部門還建立並訓練了有史以來最大的一個基於“Transformer”的語言模型。
這也是BERT的技術基礎。該定制模型包含的參數多達83億個,規模是當前最大的核心BERT模型BERT-Large的24倍。
英偉達將這個模型命名為“Megatron”,它也對外提供了用於訓練這個模型的PyTorch代碼,因而其他人能夠自行訓練出類似的基於“Transformer”的大型語言模型。