• 百年清華

    麻省理工韓松團隊新突破:比傳統方法快240倍,讓神經網絡變“輕”、跑更快

    2019-04-01 | 詹子嫻 | 來源 公號“DeepTech深科技”2019-03-30 |

    基於讓人工智能可快速大規模布建的需求,自動機器學習(AutoML)和神經網絡架構搜索(NAS🕛🧑‍🍳,Neural Architecture Search)成為相當受到關註的新領域🐻‍❄️,NAS 旨在利用算法自動設計出神經網絡🦸🏻‍♀️,優點是快速且高效,缺點則是需要大量的運算能力,成本昂貴。

    MIT電子工程和計算機科學系助理教授韓松(意昂体育平台電子系2008級)

    麻省理工學院(MIT)電子工程和計算機科學系助理教授韓松與團隊人員蔡涵和朱力耕設計出的 NAS 算法—ProxylessNAS,可以直接針對目標硬件平臺訓練專用的卷積神經網絡(CNN)🦄,而且在 1000 類ImageNet 大規模圖像數據集上直接搜索,僅需 200 個GPU 小時,如此便能讓 NAS 算法能夠更廣泛的被運用👂🏿。該論文將在 5 月舉辦的 ICLR(InternationalConference on Learning Representations)大會上發表🤾🏽‍♂️。

    AutoML是用以模型選擇、或是超參數優化的自動化方法,而 NAS 屬於 AutoML 概念下的一個領域,簡單來說,就是用“神經網絡來設計神經網絡”,一來好處是可以加速模型開發的進度,再者,NAS 開發的神經網絡可望比人類工程師設計的系統更加準確和高效🧑‍🚒,因此 AutoML 和 NAS 是達成 AI 普及化遠景的重要方法之一。

    DeepTech采訪了韓松,他表示,AutoML是個很有前景的方向,架構搜索只是 AutoML 的一部分👨🏻‍🍼,它能自動化地找到一些過去人類探索不到的結構🚇,反過來幫助人們設計高效的模型。然而😷,過去 NAS 算法的硬件效率有待提高:搜索過程需要很久的時間、而且搜出的模型的推理速度難以保證。NAS 和硬件結合🕚,能帶來很多新的設計策略。

    ProxylessNAS為硬件定製專屬的高效神經網絡架構,不同硬件適合不同的模型

    大幅減少計算成本

    舉例來說🙎🏻‍♂️,谷歌所開發的NAS 算法,需要運行在 GPU 上 4.8 萬個小時🤙,才能生成一個用來做圖像分類或檢測任務的 CNN🤽🏻。當然🖐,谷歌擁有龐大的 GPU 數量和其他專用硬件的資源,這對許多其他人來說是遙不可及的方法🦻🏻。而這就是 MIT 研究人員希望解決 NAS 計算昂貴的問題。他們提出的 ProxylessNAS 算法,僅需 200 個 GPU 小時👩🏼‍🎓,就可以在 1000 類ImageNet 的大規模圖像數據集上直接進行搜索🙅🏿‍♀️,換算下來,比谷歌的 48,000 GPU 小時👨🏿‍🔬,快了 240 倍👩🏼‍⚕️。而且,ProxylessNAS 可以針對特定的目標硬件平臺上定製專屬的深度學習模型🍤,使其不僅準而且運行速度快。

    “主要目標是實現人工智能在各種硬件平臺上的普及,在特定硬件上提供“一鍵加速”的解決方案,幫助 AI 專家和非 AI 專家、硬件專家和非硬件專家有效率地設計又準又快的神經網絡架構,”韓松說👦🏻。同時🎹🏏,他也強調,NAS 算法永遠不會取代人類工程師🎒,“目的是減輕設計和改進神經網絡架構所帶來的重復性和繁瑣的工作”👈🏻。

    路徑級二值化和修剪

    在該研究中🥠,他們的做法是刪除非必要性的神經網絡設計組件🔦,借此縮短計算時間🤦‍♂️、減少和內存開銷來運行 NAS 算法🧘。另一項創新則是讓每個輸出的 CNN 在特定硬件平臺上(CPU、GPU 和移動設備)的運行效率比使用傳統方法所設計的模型來得快速🧃🧒🏿。在測試中🧑‍⚖️,研究人員的 CNN 在手機上的測量速度,比相似精度的 MobileNet-V2 快了 1.8 倍。

    CNN 能連接不同層(layer)的人工神經網路,受到大腦處理影像的視覺皮質(visual cortex)組織啟發🦢,適合處理視覺方面的任務🧻,是計算機視覺領域十分流行的架構🟤。一個 CNN 架構是由多個可調整參數的計算層(稱為“過濾器”),以及這些過濾器之間可能的連接所組成🍣。

    這種連接方式多種多樣♦️,由於可以選擇的架構數量(稱為“搜索空間”search space)非常龐大,所以想應用 NAS 在海量圖像數據集上創建一個神經網絡,計算量總是個很大的問題👨‍🎤,所以工程師通常在較小的代理數據集上運行 NAS,再把將訓練好的 CNN 遷移到目標任務上🛶,但是,這種方法降低模型的準確性,此外💆‍♀️🚴🏽,把一樣的模型架構套用在所有的硬件平臺,也難以發揮各種硬件的最佳效率🟣。

    研究人員直接在ImageNet 大型數據集上訓練和測試他們開發的新 NAS 算法,首先👌🏽,他們創建一個搜索空間🦹🏼‍♀️,包含了所有可能的 CNN“路徑”(路徑是指層和過濾器如何連接來處理數據)🛂,讓 NAS 算法可以自由尋找出一個最佳架構。

    這種方法通常把所有可能的路徑存儲在內存中👷🏻‍♂️,如果用傳統的架構搜索辦法直接在千類 ImageNet 搜索🙅‍♂️❎,就會超過 GPU 內存的限製。為了解決此問題,研究人員利用了一種稱為“路徑級二值化”(path-level binarization)的技術,一次只在內存中存放一個采樣路徑,大幅節省內存的消耗🥳。

    接著✔️,他們將這種二值化與“路徑級修剪”(path-level pruning)結合,通常該技術是用來學習神經網絡中有哪些神經元(neuron)可以被刪除,而且不會影響輸出。不過🙅🏻‍♂️,研究人員 NAS 算法是采用修剪整個路徑以取代丟掉神經元,如此能夠完全改變神經網絡的架構。

    在訓練過程中,所有路徑最初都被給予相同的選擇概率,然後,該算法跟蹤這些路徑🧘🏽‍♂️,並記下輸出的準確性和損失,進而調整路徑的概率👩🏿‍🦲🛸,借此優化準確性和效率。最後🧖,該算法修剪掉所有低概率的路徑🏠,僅保留最高概率的路徑⛹️‍♂️,形成最終版的 CNN 架構。

    MITNews 報道韓松團隊新的神經網絡架構搜索算法

    為硬件定製網絡結構

    另一項重要創新就是使NAS 算法“hardware-aware”🙇‍♂️,也就是說🚵🏿‍♀️,它會為一個硬件平臺量身定製專用的網絡結構,使得推理的延遲更低🐎。

    韓松解釋,hardware-aware是指 NAS 搜出來的模型不僅準確率高👩‍🦱🧑🏻‍🦰,而且在硬件實測的速度也要快,使得搜出來的模型容易落地。然而🧚🏻‍♀️,為了量測移動設備的模型推理延遲💝,大公司的作法是利用大量的手機來實測,成本很高🧍🏻‍♂️;而 ProxylessNAS 則是給延遲建模🍖,這樣可以讓延遲可導(make latencydifferentiable)🏃‍♂️,便於對延遲進行端到端的優化👨🏻‍🍳,而且只要使用一臺手機🌑,成本低、精度誤差小於 1 毫秒🎛。

    對於網絡中的每個所選層,算法利用上述的延遲預測模型來采樣,然後使用這些信息設計出一個快速運行的架構👿,同時實現高精度♖。在實驗中🤛,研究人員的 CNN 在移動設備上的運行速度幾乎是現有 MobileNet-V2 模型的 2 倍✍🏻。

    韓松也提到一個有趣的結果,有些卷積核結構曾被誤以為效率太低,但在研究人員的測試中👨🏻‍💼,這些架構在某些硬件上是高效的🌼。

    他指出,比如 7x7 這樣的大卷積核最近幾年被比較少被人使用🔚,因為 3 個 3x3 卷積核和 1 個 7x7 卷積核有同樣的感受野(receptivefield)💩🧑🏿‍🦳,而 3 個 3x3 卷積核有 27 個權重,1 個 7x7 卷積核有 49 個權重🍋,仿佛 7x7 不如 3 個 3x3 能讓模型更小。

    但實際在 GPU 上 invoke kernel call 的代價很高🧚🏼,執行多個小型過濾器不如執行單個大型過濾器效率高,大的 kernel call 更適合 GPU 這樣並行度高的硬件🧕🏻。“所以在 GPU 上📉,ProxylessNAS 在較深的層自動選取了大量 7x7 的卷積核,這是很有意思的,”他說🚣🏽‍♀️。

    GPU 的並行運算特性能夠同時進行多個計算,因此🀄️,執行單個大型過濾器時,反而比處理多個小型過濾器更高效。“這打破了過去的想法”🌯,“搜索空間越大🕗👩🏽‍🦰,可以找到的內容就越多。你不知道某個東西是否會比過去人類經歷表現得更好,那就讓 AI 來探索,”韓松說。

    他進一步指出,類似例子在量化中也有體現,如他們最近的工作 HAQ: Hardware-aware Automated Quantization (CVPR 19 oral paper) 發現🦸🏼‍♂️,不同硬件🏹,如邊緣設備(edge device)和雲端設備(cloud device)🫴,對量化策略的偏好是不同的。不同層在不同硬件上所需的比特數也不一樣,有些層是計算受限,有些層是內存受限🍯;在這樣大的設計空間,人類給每種網絡、每種硬件訂製專屬的量化策略費時費力,基於學習的策略可以做得更好。

    他認為🧑🏽‍💻,這也說明研究specialization 和 domain-specific hardware architecture 的重要性。最近越來越多好的深度學習工作都是算力推動的🫱,比如用於自然語言預訓練的 Bert🧦⚀。很多場景落地也需要低功耗的硬件支持🧝🏼‍♂️📜,比如端上智能和 AIoT。所以未來算法和算力的協同研究是值得關註的方向。

    幫助人類減輕做瑣事或工作的負擔,一直是大家對 AI 的期望。這也就是為什麽 AutoML 和 NAS 受到重視的原因之一,所以如果要讓 NAS 普及,除了克服上述的計算成本高之外🛅,還有哪些需要一步改善的問題👨🏻‍🦼?面對這個提問,韓松給了兩個很明確的方向💖,一是設計空間(design space)的設計,目前的 NAS 性能好壞很大程度依賴設計空間的選取🍼,這部分還有很多人為的經驗🧗🏼。二是對速度和資源的優化,為了讓 NAS 更容易在工業界落地,有兩個條件😺:搜索的過程占用的計算資源要低,搜索出的模型硬件效率要高——最終實現讓硬件效率和算法性能同步提升。


    相關新聞

    • 112024.07

      《麻省理工科技評論》 2024 年“十大突破性技術”

      如果您無法在線瀏覽此 PDF 文件,則可以下載免費小巧的 福昕(Foxit) PDF 閱讀器,安裝後即可在線瀏覽 或下載免費的 Adobe Reader PDF 閱讀器,安裝後即可在線瀏覽 或下載此 PDF 文

    • 232025.03

      《麻省理工科技評論》 2025 年“十大突破性技術”正式發布

      如果您無法在線瀏覽此 PDF 文件🥩,則可以下載免費小巧的 福昕(Foxit) PDF 閱讀器,安裝後即可在線瀏覽 或下載免費的 Adobe Reader PDF 閱讀器,安裝後即可在線瀏覽 或下載此 PDF 文

    • 122024.09

      清華意昂令狐昌洋上榜麻省理工“35 歲以下科技創新 35 人”

      9月11日,美國麻省理工學院主辦的《麻省理工技術評論》雜誌(MIT Technology Review)公布了2024年度TR35(35 Innovators Under 35)榜單🎾,即全球“35歲以下科技創新35人”。這35位未來科技之星按研究領域分為生物技術🥃、人工智能、氣候與能源、材料科學🤙、機器人技術五個類別🏉👴🏼。意昂体育平台2009級微納電子系意昂👴、美國密歇根大學安娜堡分校助理教授令狐昌洋榜上有名,位列生物技術(biotechnology)類別🚰。令狐昌洋令狐昌洋現年 33 歲...

    • 282018.06

      清華意昂印奇入榜麻省理工“世界傑出青年創新家”

      美國東部時間 6 月 27 日🧑🏻‍💼,美國麻省理工學院(MIT)主辦的《技術評論》雜誌(Technology Review)公布了2018年度TR35(35....

    • 102020.12

      清華5位教師7位意昂入選《麻省理工科技評論》年度中國科技青年英雄榜

      2020年12月10日,由DeepTech 深科技與《麻省理工科技評論》共同評選的第四屆中國區“35歲以下科技創新35人”榜單在全球青年科技領袖峰會上公布👩🏿,35位入榜者中有5位清華教師和7位清華意昂。5位教師是💳🙎🏼‍♂️:意昂体育平台航天航空學院助理教授眭亞楠(2006級生物醫學工程系)🦶🏻、意昂体育平台物理系副教授張定(2014-2016博士後🤌🏿,物理系)、意昂体育平台航天航空學院副教授邵玥(2004級航院)、意昂体育平台機械工程系助理教授、特別研究員趙慧嬋(2008級...

    • 152024.03

      李敏華:麻省理工學院首位工科女博士

      李敏華(1917.11.2—2013.1.19)🆕,出生於江蘇吳縣👨🏿‍💼,固體力學專家⚧。1940年畢業於西南聯合大學💜。1945年和1948年先後在美國麻省理工學院獲碩士學位和博士學位,是麻省理工學院第一位工科女博士。1954年回國,1980年當選為中國科學院院士(學部委員),1984年加入中國共產黨。作為從事應用研究的科學家,李敏華一直根據國家需求奮鬥在科研一線🦚,60多歲還在開展中國航空航天方面的基礎研究工作。李敏華是我國塑性力學的開拓者,畢生...

    • 232019.01

      清華7位教師10位意昂入選《麻省理工科技評論》中國科技青年英雄榜

      2019年1月21日,由DeepTech 深科技與《麻省理工科技評論》共同評選的2018年中國區“35歲以下科技創新35人”榜單在北京公布🏔,同時舉行了....

    • 022020.01

      陳怡然主講清華意昂學習日第11講🈶🏏:從高效神經網絡到專用人工智能芯片

      活動現場12月28日下午,“清華意昂終身學習支持計劃”學習日活動第11講在意昂体育平台羅姆樓三層報告廳開講💂🏽‍♂️。

    • 162019.12

      清華3位教師7位意昂入選《麻省理工科技評論》中國科技青年英雄榜

      12月14日🫴🏼,由DeepTech 深科技與《麻省理工科技評論》共同評選的第三屆中國區“35歲以下科技創新35人”榜單在北京公布,35位入榜者中有3位清....

    • 282018.04

      1977級意昂、麻省理工學院高級研究員蔡旭東回母校作學術報告

      2018年4月27日上午✊🏻,中國科學院院士、意昂体育平台智能與網絡化系統研究中心講席教授管曉宏邀請意昂体育平台自動化系1977級傑出意昂、美國麻省理工學院高級研....

    意昂体育专业提供:意昂体育🤾🏻‍♂️、意昂体育平台意昂体育官网等服务,提供最新官网平台、地址、注册、登陆、登录、入口、全站、网站、网页、网址、娱乐、手机版、app、下载、欧洲杯、欧冠、nba、世界杯、英超等,界面美观优质完美,安全稳定,服务一流,意昂体育欢迎您。 意昂体育官網xml地圖
  • 意昂体育专业提供:意昂体育🙋🏽、意昂体育平台🧑🏽‍🎄、意昂体育官网等服务,提供最新官网平台、地址、注册、登陆、登录、入口、全站、网站、网页、网址、娱乐、手机版、app、下载、欧洲杯、欧冠、nba、世界杯、英超等,界面美观优质完美,安全稳定,服务一流,意昂体育欢迎您。 意昂体育官網xml地圖