小米再開源！這次是移動(dòng)端神經(jīng)網(wǎng)絡(luò)框架基準(zhǔn)測試項(xiàng)目MobileAIBench

繼小米在6月宣布自研的移動(dòng)端深度學(xué)習(xí)框架Mobile AI Compute Engine (MACE:https://github.com/xiaomi/mace)開源以來，小米近日又宣布開源移動(dòng)端神經(jīng)網(wǎng)絡(luò)框架基準(zhǔn)測試項(xiàng)目MobileAIBench（https://github.com/xiaomi/mobile-ai-bench）。

據(jù)雷鋒網(wǎng)了解，MobileAIBench旨在給開發(fā)這提供一個(gè)系統(tǒng)性的對(duì)比，為軟硬件的選擇提供一個(gè)直觀定量的指導(dǎo)，其目的是建立一個(gè)統(tǒng)一的軟硬件綜合評(píng)測框架，能夠?qū)Σ煌挠布⒂?jì)算單元、ABI以及神經(jīng)網(wǎng)絡(luò)計(jì)算框架進(jìn)行全方位的評(píng)測。

眼下大多數(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ù)的訴求變得越來越高的當(dāng)下，移動(dòng)設(shè)備上的離線神經(jīng)網(wǎng)絡(luò)應(yīng)用變得越來越普遍。而MobileAIBench或許能解決這個(gè)問題。

這次開源的MobileAIBench項(xiàng)目在系統(tǒng)支持上，目前僅支持安卓，但比較容易推廣所有 POSIX 兼容系統(tǒng)。現(xiàn)階段支持 MACE、SNPE、ncnn、TensorFlow Lite 等框架的評(píng)測，后續(xù)還將加入對(duì)模型量化壓縮以及模型精度的對(duì)比。用戶可以根據(jù)項(xiàng)目文檔，自行添加一個(gè)框架或者一個(gè)新模型。

據(jù)悉，該項(xiàng)目會(huì)進(jìn)行每日構(gòu)建，并對(duì)幾款手機(jī)進(jìn)行測試，測試數(shù)據(jù)可以通過項(xiàng)目頁的 CI Pipeline 頁面獲取 (https://gitlab.com/llhe/mobile-ai-bench/pipelines) 。

雷鋒網(wǎng)了解到，小米一直在倡導(dǎo)擁抱開源，積極參與多個(gè)國際重大的開源項(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)來。

雷峰網(wǎng)原創(chuàng)文章，未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。