Microsoft 在 Model Builder 中新增句子相似度场景

Microsoft 最近发布了关于在 Model Builder 中新增句子相似度场景的信息。该场景允许用户训练自定义的句子相似度模型。随着这一场景的加入，用户不再需要安装 Model Builder GPU 扩展。此外，Microsoft 还透露了未来几个月在深度学习、LightGBM 算法和 AutoML 等领域的发展计划。

句子相似度 API 的预览版

几个月前，Microsoft 发布了句子相似度 API 的预览版，该 API 提供了一种使用自定义数据训练基于句子相似度的机器学习模型的方法。这是通过将 TorchSharp 的 NAS-BERT 实现与 ML.NET 集成来实现的。该模型与文本分类 API 使用的基本变压器模型相同。句子相似度 API 利用自定义数据对预训练模型进行微调。

使用句子相似度场景的要求

要使用这一场景，用户必须安装或升级 Model Builder 到最新版本 16.14.4。句子相似度场景支持在 CPU 和 GPU 上进行本地训练。对于 GPU，需要一个兼容 CUDA 的 GPU。更多关于配置 GPU 的信息可在 ML.NET GPU 指南中找到。

不再需要安装 GPU 扩展

从 Model Builder 16.14.4 版本开始，不再需要安装 GPU 扩展。在之前的版本中，为了支持 GPU，除了满足硬件要求和安装适当的驱动程序外，还需要安装 Model Builder 的 GPU 扩展。

社区反馈

这一场景的加入受到了社区的积极反馈。例如，一位 Reddit 用户表示，这是他一段时间以来一直在寻找的解决方案，并将在他的项目中应用这一场景。

未来开发计划

Microsoft 还透露了未来几个月在机器学习解决方案方面的开发计划。首先，将扩展深度学习场景，包括对象检测、问题回答和命名实体识别等 API。其次，将更新 ML.NET 中支持的 LightGBM 版本，并通过允许以原生格式加载 LightGBM 模型来改善互操作性。此外，还计划在明年对 AutoML API 进行进一步改进，以简化机器学习工作流程并支持新的场景和定制。

完整变更列表

完整的变更列表可在 Model Builder 的发布说明中找到。