小米再開源!這次是移動(dòng)端神經(jīng)網(wǎng)絡(luò)框架基準(zhǔn)測(cè)試項(xiàng)目MobileAIBench
繼小米在 6 月宣布自研的移動(dòng)端深度學(xué)習(xí)框架 Mobile AI Compute Engine (MACE:https://github.com/xiaomi/mace)開源以來(lái),小米近日又宣布開源移動(dòng)端神經(jīng)網(wǎng)絡(luò)框架基準(zhǔn)測(cè)試項(xiàng)目 MobileAIBench(https://github.com/xiaomi/mobile-ai-bench)。
據(jù)雷鋒網(wǎng)(公眾號(hào):雷鋒網(wǎng))了解,MobileAIBench 旨在給開發(fā)這提供一個(gè)系統(tǒng)性的對(duì)比,為軟硬件的選擇提供一個(gè)直觀定量的指導(dǎo),其目的是建立一個(gè)統(tǒng)一的軟硬件綜合評(píng)測(cè)框架,能夠?qū)Σ煌挠布?、?jì)算單元、ABI 以及神經(jīng)網(wǎng)絡(luò)計(jì)算框架進(jìn)行全方位的評(píng)測(cè)。
眼下大多數(shù)開發(fā)者面臨著同樣一個(gè)困境,即如何選擇滿足應(yīng)用計(jì)算需求同時(shí)具有高性價(jià)比的硬件,以及如何選取硬件適合的神經(jīng)網(wǎng)絡(luò)計(jì)算框架。除此之外,開發(fā)者還需要權(quán)衡模型量化壓縮以及模型的精度損失,對(duì)于應(yīng)用或者算法開發(fā)者而言,如何做出合適的選擇,往往需要進(jìn)行多方位的嘗試,耗時(shí)耗力。在用戶對(duì)智能性、低延遲和隱私保護(hù)的訴求變得越來(lái)越高的當(dāng)下,移動(dòng)設(shè)備上的離線神經(jīng)網(wǎng)絡(luò)應(yīng)用變得越來(lái)越普遍。而 MobileAIBench 或許能解決這個(gè)問題。
這次開源的 MobileAIBench 項(xiàng)目在系統(tǒng)支持上,目前僅支持安卓,但比較容易推廣所有 POSIX 兼容系統(tǒng)。現(xiàn)階段支持 MACE、SNPE、ncnn、TensorFlow Lite 等框架的評(píng)測(cè),后續(xù)還將加入對(duì)模型量化壓縮以及模型精度的對(duì)比。用戶可以根據(jù)項(xiàng)目文檔,自行添加一個(gè)框架或者一個(gè)新模型。
據(jù)悉,該項(xiàng)目會(huì)進(jìn)行每日構(gòu)建,并對(duì)幾款手機(jī)進(jìn)行測(cè)試,測(cè)試數(shù)據(jù)可以通過(guò)項(xiàng)目頁(yè)的 CI Pipeline 頁(yè)面獲取 (https://gitlab.com/llhe/mobile-ai-bench/pipelines) 。
雷鋒網(wǎng)了解到,小米一直在倡導(dǎo)擁抱開源,積極參與多個(gè)國(guó)際重大的開源項(xiàng)目,比如 Hadoop,HBase,Spark,TensorFlow 等。值得一提的是,在 HBase 項(xiàng)目中,培養(yǎng)出了兩位 PMC 成員和七位 Committer,成為社區(qū)中的核心力量。在此之前,小米已經(jīng)推出了一系列開源項(xiàng)目:
Linden(分布式實(shí)時(shí)搜索系統(tǒng))
Open-Falcon(互聯(lián)網(wǎng)企業(yè)級(jí)監(jiān)控系統(tǒng))
Pegasus(分布式 KV 存儲(chǔ)系統(tǒng))
自研移動(dòng)端深度學(xué)習(xí)框架 MACE 等
小米人工智能與云平臺(tái)副總裁崔寶秋博士此前表示:擁抱開源是小米工程文化的重要組成部分,從手機(jī)的移動(dòng)操作系統(tǒng),到云計(jì)算、大數(shù)據(jù) 、人工智能,開源都在小米扮演了極其重要的角色。小米歡迎更多的開發(fā)者參與進(jìn)來(lái)。