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