继小米在 6 月宣布自研的移动端深度学习框架 Mobile AI Compute Engine(MACE:https://github.com/xiaomi/mace)开源以来,小米近日又宣布开源行动端神经网络框架基准测试计划 MobileAIBench(https://github.com/xiaomi/mobile-ai-bench)。
据了解,MobileAIBench 旨在提供一系统性对比,为软硬件的选择给予直觉定量的指导,目的是建立统一的软硬件综合评测框架,能全方位评测不同硬件、计算单元、ABI 及神经网络计算框架。
目前大多数开发者面临同样困境,即如何选择满足应用计算需求同时具高性价比的硬件,以及如何选取硬件适合的神经网络计算框架。除此之外,开发者还需权衡模型量化压缩及模型的精度损失,对应用或算法开发者而言,如何做出合适的选择,往往需要多方面尝试,耗时耗力。用户对智慧性、低延迟和隐私保护的诉求越来越高的当下,行动装置的离线神经网络应用越来越普遍,而 MobileAIBench 或许能解决这个问题。
这次开源的 MobileAIBench 计划在系统支援方面,目前仅支援 Android,但比较容易推广所有 POSIX 相容系统。现阶段支援 MACE、SNPE、ncnn、TensorFlow Lite 等框架评测,后续还会加入模型量化压缩及模型精准度的对比。用户可根据计划文件,自行添加一个框架或一个新模型。
据悉,该计划会每日构建,并测试几款手机,测试数据可透过计划页的 CI Pipeline 页面取得(https://gitlab.com/llhe/mobile-ai-bench/pipelines)。
据了解,小米一直倡导拥抱开源,积极参与多个国际重大的开源计划,比如 Hadoop、HBase、Spark、TensorFlow 等。值得一提的是,HBase 计划中,培养出两位 PMC 成员和 7 位 Committer,成为社群的核心力量。在此之前,小米已推出一系列开源计划:
- Linden(分布式即时搜寻系统)
- Open-Falcon(网络企业级监控系统)
- Pegasus(分布式 KV 储存系统)
- 自研行动端深度学习框架 MACE 等
小米人工智能与云端平台副总裁崔宝秋博士之前表示:拥抱开源是小米工程文化的重要组成部分,从手机的行动操作系统,到云端计算、大数据、人工智能,开源在小米扮演极其重要的角色,小米欢迎更多开发者一起参与。
(本文由 雷锋网 授权转载;首图来源:达志影像)