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