11月1日，騰訊AI Lab在南京舉辦的騰訊全球合作伙伴論壇上宣布正式開源“PocketFlow”項目， 該項目是一個自動化深度學習模型壓縮與加速框架，整合多種模型壓縮與加速算法并利用強化學習自動搜索合適壓縮參數，解決傳統深度學習模型由于模型體積太大，計算資源消耗高而難以在移動設備上部署的痛點，同時極大程度的降低了模型壓縮的技術門檻，賦能移動端AI應用開發。

這是一款適用于各個專業能力層面開發者的模型壓縮框架，基于Tensorflow開發，集成了當前主流與AI Lab自研的多個模型壓縮與訓練算法，并采用超參數優化組件實現了全程自動化托管式的模型壓縮。開發者無需了解具體模型壓縮算法細節，即可快速地將AI技術部署到移動端產品上，實現用戶數據的本地高效處理。

目前該框架在騰訊內部已對多個移動端AI應用模型進行壓縮和加速，并取得了令人滿意的效果， 對應用整體的上線效果起到了非常重要的作用。

隨著移動互聯網的普及和AI算法的愈加成熟，移動端AI的需求和應用場景也愈加豐富，如智能美顏，瘦身，手勢識別，場景識別，游戲AI，視頻檢測，車載語音交互，智能家居，實時翻譯等。將模型部署在移動端代替服務器端可以實現實時交互，在無網絡環境的情況下也可正常運行，并且保護了數據的隱私和安全性，降低運維成本。而在移動端AI模型部署方面，小而快變成了模型推理最渴望的要求之一，因為小的模型可以使得整個應用的存儲空間減小；而提高模型的推理速度則可以使得整個模型應用場景的反應速度加快，同時也可以在低配設備上取得很好的效果。因此，模型的推理速度逐漸已經成為了各大AI應用市場競爭力上最重要的評測指標之一。然而往往訓練出來的模型推理速度或者尺寸大小不能滿足需求，大部分開發者對極致的模型壓縮和加速方法又感到困惑。

PocketFlow的出現就是為了解決這個令眾多AI應用開發者頭痛的問題。

?

框架開源內容

PocketFlow框架本次開源內容主要由兩部分組件構成，分別是模型壓縮/加速算法部分和超參數優化部分，具體結構如下圖所示：?

模型壓縮/加速算法部分包括多種深度學習模型壓縮和加速算法：

?

• 通道剪枝（channel pruning）： 在CNN網絡中，通過對特征圖中的通道維度進行剪枝，可以同時降低模型大小和計算復雜度，并且壓縮后的模型可以直接基于現有的深度學習框架進行部署。PocketFlow還支持通道剪枝的分組finetune/retrain功能，通過實驗我們發現此方法可以使原本壓縮后的模型精度有明顯的提升。

• 權重稀疏化（weight sparsification）：通過對網絡權重引入稀疏性約束，可以大幅度降低網絡權重中的非零元素個數；壓縮后模型的網絡權重可以以稀疏矩陣的形式進行存儲和傳輸，從而實現模型壓縮。

• 權重量化（weight quantization）：通過對網絡權重引入量化約束，可以降低用于表示每個網絡權重所需的比特數；我們同時提供了對于均勻和非均勻兩大類量化算法的支持，可以充分利用ARM和FPGA等設備的硬件優化，以提升移動端的計算效率，并為未來的神經網絡芯片設計提供軟件支持。

• 網絡蒸餾（network distillation）：對于上述各種模型壓縮組件，通過將未壓縮的原始模型的輸出作為額外的監督信息，指導壓縮后模型的訓練，在壓縮/加速倍數不變的前提下均可以獲得0.5%-2.0%不等的精度提升。

• 多GPU訓練（multi-GPU training）：深度學習模型訓練過程對計算資源要求較高，單個GPU難以在短時間內完成模型訓練，因此我們提供了對于多機多卡分布式訓練的全面支持，以加快使用者的開發流程。無論是基于ImageNet數據的Resnet-50圖像分類模型還是基于WMT14數據的Transformer機器翻譯模型，均可以在一個小時內訓練完畢。

?

超參數優化（hyper-parameter optimization）部分可以通過強化學習或者AutoML，在整體壓縮率一定的情況下，搜索出每一層最合適的壓縮比例使得整體的精度最高。多數開發者對模型壓縮算法往往了解較少，調節壓縮算法參數需要長期的學習和實驗才能有所經驗，但超參數取值對最終結果往往有著巨大的影。PocketFlow的超參數優化部分正是幫助開發者解決了這一大痛點，并且通過實驗我們發現，其優化結果也要好于專業的模型壓縮工程師手工調參的結果。其結構如下圖：

?

PocketFlow性能

通過引入超參數優化組件，不僅避免了高門檻、繁瑣的人工調參工作，同時也使得PocketFlow在各個壓縮算法上全面超過了人工調參的效果。以圖像分類任務為例，在CIFAR-10和ImageNet等數據集上，PocketFlow對ResNet和MobileNet等多種CNN網絡結構進行有效的模型壓縮與加速。

?

在CIFAR-10數據集上，PocketFlow以ResNet-56作為基準模型進行通道剪枝，并加入了超參數優化和網絡蒸餾等訓練策略，實現了2.5倍加速下分類精度損失0.4%，3.3倍加速下精度損失0.7%，且顯著優于未壓縮的ResNet-44模型； 在ImageNet數據集上，PocketFlow可以對原本已經十分精簡的MobileNet模型繼續進行權重稀疏化，以更小的模型尺寸取得相似的分類精度；與Inception-V1、ResNet-18等模型相比，模型大小僅為后者的約20~40%，但分類精度基本一致（甚至更高）。

? ? ? ? ?? ? ? ? ?

相比于費時費力的人工調參，PocketFlow框架中的AutoML自動超參數優化組件僅需10余次迭代就能達到與人工調參類似的性能，在經過100次迭代后搜索得到的超參數組合可以降低約0.6%的精度損失；通過使用超參數優化組件自動地確定網絡中各層權重的量化比特數，PocketFlow在對用于ImageNet圖像分類任務的MobileNet-v1模型進行壓縮時，取得了一致性的性能提升；用PocketFlow平均量化比特數為8時，準確率不降反升，從量化前的70.89%提升到量化后的71.29%。

PocketFlow落地助力內部移動應用AI落地

在騰訊公司內部，PocketFlow框架正在為多項移動端業務提供模型壓縮與加速的技術支持。例如，在手機拍照APP中，人臉關鍵點定位模型是一個常用的預處理模塊，通過對臉部的百余個特征點（如眼角、鼻尖等）進行識別與定位，可以為后續的人臉識別、智能美顏等多個應用提供必要的特征數據。我們基于PocketFlow框架，對人臉關鍵點定位模型進行壓縮，在保持定位精度不變的同時，大幅度地降低了計算開銷，在本身已經十分精簡的網絡上取得了1.3 ~ 2倍不等的加速效果，壓縮后的模型已經在實際產品中得到部署。

?

在人體體態識別項目中，PocketFlow更是在滿足上線精度的要求下，使得模型推理速度有3倍以上的加速， 為項目的移動端落地起到了決定性的作用。

?

?

PocketFlow開源后展望

此次開源后，騰訊公司將繼續致力于開發更多的模型壓縮算法以及自動調參策略，支持更多的AI算法與應用場景，非常歡迎更多的外部開發者貢獻代碼，一起完善模型壓縮與加速的生態！

Github開源地址：

https://github.com/Tencent/PocketFlow

（點擊文末閱讀原文直接訪問）

請給?PocketFlow?一個?Star?!
?歡迎提出你的 issue 和 PR

?PocketFlow?國內鏡像地址：

https://git.code.tencent.com/Tencent_Open_Source/PocketFlow

騰訊工蜂源碼系統為開源開發者提供完整、最新的騰訊開源項目國內鏡像

總結

以上是生活随笔為你收集整理的腾讯 AI Lab 正式开源PocketFlow自动化深度学习模型压缩与加速框架的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。