端側(cè)AI推理,高效部署PyTorch模型:官方新工具開源,Meta已經(jīng)用上了
在 2023 年 PyTorch 大會上,一個深受大家關(guān)心的推理問題得到了很好的解決,會上宣布了一個用于在邊緣和移動設(shè)備上實(shí)現(xiàn) AI 推理的解決方案:ExecuTorch,并且還是開源的,而促成這一研究的,正是 Meta AI 與 PyTorch 基金會。
ExecuTorch 地址:https://github.com/pytorch/executorch
學(xué)習(xí)文檔:https://pytorch.org/executorch/stable/index.html
隨著 ExecuTorch 的開源,預(yù)示著 AI 應(yīng)用程序在設(shè)備上本地運(yùn)行、而需連接到服務(wù)器或云成為可能。我們可以將 ExecuTorch 理解成一個 PyTorch 平臺,其能提供基礎(chǔ)設(shè)施來運(yùn)行 PyTorch 程序,從 AR/VR 可穿戴設(shè)備到標(biāo)準(zhǔn)的 iOS 和 Android 設(shè)備的移動部署。
ExecuTorch 最大優(yōu)勢是可移植性,能夠在移動和嵌入式設(shè)備上運(yùn)行。不僅如此,ExecuTorch 還可以提高開發(fā)人員的工作效率。
據(jù)了解,Meta 已經(jīng)驗(yàn)證了這項(xiàng)技術(shù),并將其用于最新一代的雷朋智能眼鏡,而這款眼鏡也是 Meta 最近發(fā)布的 Quest 3 VR 頭顯的一部分。Meta 表示,作為開源 PyTorch 項(xiàng)目的一部分,他們旨在進(jìn)一步推動該技術(shù)的研究,從而邁入在設(shè)備上實(shí)現(xiàn) AI 推理的新時代。
Facebook 創(chuàng)始人、Meta 董事長兼首席執(zhí)行官扎克伯格表示:「作為開源 AI 工作的一部分,我們與 PyTorch 基金會及其行業(yè)合作伙伴一起開源了 ExecuTorch。這一變化預(yù)示著將 PyTorch 引入了手機(jī)和可穿戴設(shè)備等邊緣計(jì)算平臺。ExecuTorch 使 AI 模型能夠直接在設(shè)備上運(yùn)行,而無需連接到服務(wù)器。」
Meta 軟件工程師 Mergen Nachin 指出,「今天的 AI 模型正在從服務(wù)器擴(kuò)展到邊緣設(shè)備,如移動設(shè)備、AR、VR 和 AR 頭顯、可穿戴設(shè)備、嵌入式系統(tǒng)等。ExecuTorch 通過提供端到端的工作流來優(yōu)化本地程序,從而解決邊緣設(shè)備遇到的挑戰(zhàn)。」
ExecuTorch 關(guān)鍵組件
ExecuTorch 提供了緊湊的運(yùn)行時和輕量級操作注冊表,以覆蓋 PyTorch 模型生態(tài)系統(tǒng),以及在邊緣設(shè)備上執(zhí)行 PyTorch 程序的簡化路徑。此外,ExecuTorch 還附帶 SDK 和工具鏈,為 ML 開發(fā)人員提供了更好的用戶體驗(yàn)。
作為 PyTorch Edge 生態(tài)系統(tǒng)的一部分,ExecuTorch 可以有效地將 PyTorch 模型部署到邊緣設(shè)備。ExecuTorch 的優(yōu)點(diǎn)包括:
- 可移植性:與各種計(jì)算平臺兼容,從高端移動手機(jī)到高度受限的嵌入式系統(tǒng)和微控制器。
- 提高生產(chǎn)力:開發(fā)人員能夠使用相同的工具鏈和 SDK,從而提高生產(chǎn)力。
- 提高性能:由于輕量級運(yùn)行時和充分利用 CPU、NPU 和 DSP 等硬件功能,為最終用戶提供了無縫和高性能的體驗(yàn)。
由于 ExecuTorch 嚴(yán)重依賴 PyTorch 相關(guān)知識,因而,想要熟練掌握 ExecuTorch,還需提前補(bǔ)充相關(guān)知識。官方文檔已經(jīng)提供了入門級教程。例如,在構(gòu)建 ExecuTorch Android 演示應(yīng)用程序示例當(dāng)中,大家可以跟隨指導(dǎo)教程,從而熟悉如何使用 ExecuTorch。
最后,需要提醒大家的一點(diǎn)是,本次發(fā)布的 ExecuTorch 是一個預(yù)覽版本,在測試和評估中可以使用,但是不建議在生產(chǎn)環(huán)境中使用。PyTorch 團(tuán)隊(duì)歡迎來自社區(qū)的任何反饋、建議和錯誤報(bào)告,以幫助他們改進(jìn)技術(shù)。