微调Mistral 7B以实现命名实体识别 (NER)

2024年04月23日 由 alex 发表 42 0

在自然语言处理(NLP)领域,命名实体识别(NER)被认为是一项关键任务,应用范围广泛,包括信息提取和问题解答系统。


Mistral 7B 是由 Mistral AI 开发的一种创新型开源大型语言模型,它的推出为 NLP 领域带来了重大变革。本文旨在深入探讨 Mistral 7B 在处理 NER 任务方面的能力,重点介绍如何对这一先进模型进行微调,以实现卓越的实体识别性能。


命名实体识别简介

命名实体识别(NER)是 NLP 的一项基本任务,主要用于识别文本中的命名实体并对其进行分类,例如人名、地点、组织、日期等。命名实体的精确识别对于信息检索、情感分析和知识图谱构建等众多下游应用至关重要。


传统的命名实体识别系统依赖于人工规则创建和特征工程技术,这些技术往往缺乏跨领域的可扩展性和鲁棒性。


然而,Mistral 7B 等深度学习和预训练语言模型的出现彻底改变了这一领域。这些模型提供了一种利用大量文本数据的数据驱动方法,从而提高了 NER 任务的性能和可扩展性。


锐意进取: Mistral 7B 与其他开源 LLM 的比较

Mistral 7B 作为领先的竞争者脱颖而出,提供了性能、效率和可访问性的独特融合。与同类产品相比,Mistral 7B 融合了创新的架构元素,如分组查询关注和滑动窗口关注,从而提高了推理速度和序列处理能力。


这些进步不仅使 Mistral 7B 与众不同,还凸显了它在各种自然语言处理(NLP)任务中的卓越适应性和弹性。此外,Mistral 7B 的 Apache 2.0 许可和社区驱动的开发方法营造了协作和创新的氛围,巩固了其在寻求最先进语言理解模型的研究人员和从业人员中的首选地位。


揭幕 Mistral 7B:NLP 领域的游戏规则改变者

Mistral 7B 是语言模型领域的创新力量,它拥有令人印象深刻的 70 亿个参数,经过精心设计,可以应对错综复杂的自然语言。


Mistral 7B 采用了分组查询关注(GQA)和滑动窗口关注(SWA)等最先进的功能,表现出无与伦比的性能和效率,成为包括命名实体识别在内的 NLP 任务的有力选择。


GQA 的加入使 Mistral 7B 能够加快推理速度,实现对文本数据的实时处理,这对于在动态环境中运行的 NER 任务来说是必不可少的能力。此外,SWA 使 Mistral 7B 能够有效地处理不同长度的序列,确保在各种文本输入中都能保持稳定的性能。


针对 NER 微调 Mistral 7B:释放其全部潜能

尽管 Mistral 7B 在预训练状态下表现出了令人印象深刻的能力,但通过微调过程,它的潜力得以充分发挥。在微调过程中,会对模型参数进行调整,以适应特定任务和数据集的复杂性。针对命名实体识别(NER)任务对 Mistral 7B 进行微调,需要在有注释的数据上进行额外的训练,使模型能够掌握特定领域的模式,提高识别实体的准确性。


安装微调 Mistral 所需的 Python 软件包和库


1


2


使用 UBIAI 工具进行数据标注和结构设计

在对 Mistral 7B 等大型语言模型进行微调时,数据的质量和相关性是对模型的性能和有效性产生重大影响的关键因素。利用高质量、有良好注释的数据可确保模型从准确且有代表性的示例中学习,从而提高其泛化和生成可靠输出的能力。


本教程的数据来源于 UBIAI,该平台以其强大的注释工具和全面的 NLP 功能而著称。利用 UBIAI 的注释功能,可以确保用于微调的数据经过精心标注,并针对手头的特定任务进行定制,从而最大限度地提高训练过程的有效性。


3


4


该代码从包含发票信息的文件中读取 JSON 数据。它将数据组织成结构化格式,确保没有重复注释。处理后的结果将被存储起来,用于进一步分析和微调 Mistral 的命名实体识别 (NER) 任务。


5


6


7


不同标签: {'人'、'产品'、'设施'、'金钱'、'事件'、'日期'、'组织'、'地点'}。


数据集由包含名称、文本和注释的文档组成。这些注释包括个人、组织、地点、设施、事件等各种实体,为命名实体识别(NER)任务提供了结构化数据。


8


9


现在,该数据集增加了一个全面的提示,划定了实体提取任务的说明。每条指令都为提取各种标签(包括日期、个人、组织、地点、设施、事件、金钱和产品)中的实体提供了精确的指导。提示强调了回答的准确性和相关性的重要性,提供了确保提取信息的一致性和全面性的指导原则。


这种结构化的提示有助于简化实体识别任务的数据处理,与 NER(命名实体识别)流程的目标完全一致。


10


11


这一过程包括将数据子集转换为字符串表示法,将其附加到列表中,并保存到 CSV 文件中。随后,CSV 文件被读入 Pandas DataFrame,并转换为 Hugging Face Datasets对象,以便进一步处理。


12


13


使用 BitsAndBytesConfig 配置加载 Mistral 7B 基本模型,可实现量化,从而提高内存使用效率和推理速度。此外,还加载了与 Mistral 7B 相关的标记符,确保为文本处理任务提供适当的填充和序列末端标记符设置。


14


15


本节代码使用 Weights & Biases (W&B) 平台对 Mistral 7B 的微调过程进行监控。通过使用所提供的 API 密钥与 W&B 进行身份验证,训练进度和性能指标将受到持续的实时监控。wandb.init() "函数会在指定项目("微调 Mistral 7B UBIAI")中启动新的运行,并将作业类型指定为 "训练"。启用匿名访问可确保训练数据的安全,同时促进协作监测和分析。


16


初始化训练过程的超参数,定义输出目录、历时次数、批量大小、学习率和记录频率等设置。此外,它还会配置 SFTTrainer,指定模型、数据集、标记化器和其他相关参数,这些参数对微调 Mistral 7B 以完成给定任务至关重要。


17


18


19


试试我们的模型 使用用户提示测试 Mistral 7B

所提供的函数流将用户提示作为输入,并使用 Mistral 7B 模型根据该提示生成文本。它利用模型的 generate 方法,以用户提示为上下文生成文本。生成的文本将作为输出返回。


20


21


结果


22


结论

为命名实体识别微调 Mistral 7B 是实体识别任务的变革性方法,可提供更高的性能、适应性和效率。随着企业不断利用 NER 进行信息提取、知识发现等工作,Mistral 7B 所释放的功能为 NLP 的突破性进展铺平了道路。通过利用 Mistral 7B 的强大功能并针对 NER 进行微调,从业人员可以在理解和处理文本数据方面开启新的可能性,从而推动不同领域的创新。

文章来源:https://medium.com/ubiai-nlp/fine-tuning-mistral-7b-for-named-entity-recognition-ner-bbb96af918d3
欢迎关注ATYUN官方公众号
商务合作及内容投稿请联系邮箱:bd@atyun.com
评论 登录
热门职位
Maluuba
20000~40000/月
Cisco
25000~30000/月 深圳市
PilotAILabs
30000~60000/年 深圳市
写评论取消
回复取消