ML.NET Model Builder 中的句子相似性

ML.NET 是一个面向 .NET 开发人员的开源跨平台机器学习框架，可将自定义机器学习模型集成到 .NET 应用中。

新版本的 Model Builder 现已发布！

有什么新变化？

以下是此版本的亮点。您可以在 Model Builder 发行说明中找到所有更改的列表。

要开始使用这些新功能，请安装或升级到最新版本的 Model Builder 16.14.4 或更高版本。

Model Builder中的句子相似性

句子相似性是一项比较两个文本彼此相似程度的任务。

句子相似性的常见用例是信息检索。例如，给出一个搜索查询，返回最相似（相关）的文档。

几个月前，我们发布了句子相似性 API 的预览版，使你能够使用自己的数据训练自定义句子相似性机器学习模型。它通过将 NAS-BERT 的 TorchSharp 实现集成到 ML.NET 中来实现。这与文本分类 API 使用的基于转换器的基础模型相同。使用此模型的预训练版本，句子相似性 API 使用您的数据来微调模型。

今天，我们很高兴地宣布 Model Builder 中的句子相似性方案由 ML.NET 句子相似性 API 提供支持。

借助此新方案，可以使用 Model Builder 中 Microsoft 研究院的最新深度学习技术来训练自定义句子相似性模型。

此方案支持在 CPU 和 GPU 上进行本地训练。对于 GPU，您需要与 CUDA 兼容的 GPU，我们建议至少使用 6 GB 的专用内存。有关设置 GPU 的更多详细信息，请参阅 ML.NET GPU 指南。

获取最新版本的 Model Builder，立即开始训练句子相似性模型。

不再需要 Model Builder GPU 扩展

随着我们继续在 Model Builder 中引入新的深度学习方案，能够在 GPU 上进行训练非常重要。

当我们首次在 Model Builder 中引入 GPU 支持时，除了满足硬件要求和安装相应的驱动程序外，还必须安装 Model Builder GPU 扩展。

我们很高兴地宣布，从 Model Builder 版本 16.14.4 开始，不再需要安装 GPU 扩展。

下一步是什么？

在高级别上，以下项目概述了我们将在未来几个月内重点关注的领域。

深度学习 – 继续扩大深度学习场景覆盖范围。这包括新的方案 API，例如用于对象检测、问答和命名实体识别的文本分类和句子相似性。LightGBM – 升级 ML.NET 中支持的 LightGBM 版本，并通过启用以本机格式加载 LightGBM 模型来提高互操作性。AutoML – 在接下来的一年里，我们计划继续改进 AutoML API，以实现新的场景和自定义，从而简化初学者和体验用户的机器学习工作流程。ML 工具 – 随着 ML.NET 组 API 中提供新的方案和功能，我们计划将它们引入 Model Builder 和 ML.NET CLI，并改善我们工具中的整体用户体验。有关更多详细信息，请参阅 ML.NET 和 Model Builder 路线图。

确认 我们要感谢我们的 Microsoft 研究院和 TorchSharp 合作伙伴帮助我们在 ML.NET 中提供这些新方案和功能。

入门和资源 在 Microsoft Docs 中了解有关 ML.NET、Model Builder 和 ML.NET CLI 的更多信息。

如果遇到任何问题、功能请求或反馈，请在 ML.NET 和 Model Builder 存储库中提交问题。

加入 .NET 开发 Discord 上的 ML.NET 社区 Discord 或#machine-learning 频道。

收听.NET 社区机器学习站立会议。