旷视开源自研深度学习框架“天元” 助力行业打造AI基础设施

白杨2020-03-25 18:19

目前,旷视已在OpenI启智社区和GitHub上同步发布了天元Alpha版源代码。

3月25日下午,旷视科技对外宣布,将正式开源公司的核心深度学习框架——旷视天元(MegEngine),这也使旷视科技成为全球首个将底层框架开源的人工智能企业。

据悉,天元的前身是旷视内部自用的算法训练推理引擎MegEngine,由旷视3名实习生在2013年启动研发,并与2014年正式完成上线。该框架作为旷视科技的底层技术架构,支撑了公司过去6年间各项产品和业务的落地,同时也服务于旷视研究院1400余名AI开发者。

旷视科技联合创始人兼CTO唐文斌表示,旷视是全球最早用深度学习方法开展人工智能商业化探索的科技企业,但由于早期AI 开发工具的匮乏,旷视靠自主研发,从 0 到 1 地解决了算法生产的问题。

也正是有了这一经历,唐文斌深知,真正影响产业跑速的不是算法、不是软件和硬件,也不是解决方案,而是AI的基础设施。他表示,“AI的基础设施由两部分组成—— 芯片平台和生产力平台。其中,芯片平台承载AI的计算, 而AI算法的生产则是一个系统工程,需要能够协同优化数据、算法、算力的平台级产品。”

唐文斌,此前,旷视科技基于天元打造了自己的基础设施Brain++,现在也希望把它开放给更多的企业和开发者,促进整个AI产业的繁荣。

据天元开源项目负责人、旷视研究院高级技术总监田忠博介绍,传统深度学习研发中,产品从原型到生产部署往往需要分开设计和调用训练框架和推理框架两种框架,这就导致模型在训练和推理的转换的过程中出现不明原因的性能或精度损失,需要开发者手工进行优化,而算法在计算平台部署的时候出现各种问题也无法追溯。

天元框架在设计之初就避免了这样的问题,通过训练、推理一体化的机制省去模型转换的过程,可直接将训练后的模型进行推理,并保证跨设备的模型精度对齐,同时天元内置自动模型优化、简化流程,减少因手工操作的机会,降低出错概率。

除此之外,天元还配置了 Python 和 C++ 接口,支持动态图、静态图一键转换和混合编程,可使用高级编程语言进行图优化和图编译。

田忠博表示,目前,旷视已在人工智能开源开放平台OpenI启智社区和全球最大的开源社区 GitHub 上同步发布了天元 Alpha 版源代码,开发者也可以通过天元官方网站的在线深度学习工具免费调用算力、获取最新数据集和训练脚本,进行简单训练和试用。

对于天元框架开源后的发展规划,旷视透露,将在技术贡献者的帮助下,于6月份推出 Beta 版本。

(编辑:徐旭)