微软神经网络引擎 ONNX 获推 1.17 更新,支持浏览器训练模型

it之家 2 月 14 日消息,据微软官方新闻稿,微软日前为旗下神经网络推论引擎“onnx runtime”推出了 1.17 版本,主要加入了网页端训练功能,允许开发者在浏览器中训练模型。

据介绍,微软在 onnx runtime 1.8 版本中,首度为引擎推出了浏览器内推论功能,该功能利用 webgl 和 webgpu 进行推论,允许开发者调用浏览器跨平台部署机器学习模型。

而在最新的 onnx runtime 1.17 中,微软选择进一步强化浏览器在训练过程中起到的作用,允许开发者使用浏览器训练模型。

微软提到,相关基于浏览器的网页训练功能主要由 webassembly 和 javascript api 两部分组成,其中 webassembly 用于转换代码,javascript api 用于管理 webassembly 与缓冲区转换过程。

▲ 图源 微软官方新闻稿

it之家注意到,由于浏览器不能直接允许 c 语言代码,因此微软使用 webassembly 将代码转换为能够在浏览器中高效运行的.wasm 二进制文件,以便于开发者在浏览器中进行模型训练。

微软同时提到,目前基于浏览器的模型训练方式仅支持 cpu 单线程运算,开发者同时需要使用 onnx 工具生成必要的运行环境,之后才能进行训练。