devbox
Gausst松鼠会
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

NLP实战：快递单信息抽取-基于ERNIE1.0 预训练模型_ernie 实体识别

作者：Gausst松鼠会 | 2024-03-29 19:30:27

赞

踩

ernie 实体识别

目录

一、介绍

二、代码

2.1前期准备

2.2加载自定义数据集

 2.3数据处理

2.4数据读入

2.5PaddleNLP一键加载预训练模型

2.6设置Fine-Tune优化策略，模型配置

2.7模型训练与评估

​编辑

2.8模型预测

三、总结

原文：

一、介绍

命名实体识别（Named Entity Recognition，简称NER）是自然语言处理（NLP）中的一项任务，目的是识别文本中的命名实体并将其分类为预定义的类别，如人名、地名、组织机构等。命名实体通常指具有独特标识的实体，这些实体在文本中出现时通常会带有特殊的上下文信息。

命名实体识别在很多自然语言处理任务中都是非常重要的预处理步骤，如信息提取、机器翻译、问题回答等。通过对文本中的命名实体进行识别和分类，可以更好地理解文本的含义和结构，为后续的处理任务提供更多的语义信息。

常用的命名实体分类包括人名、地名、组织机构名、日期、时间、货币等，分类的具体数量和类别可以根据不同的应用场景进行调整。

本项目将演示如何使用PaddleNLP语义预训练模型ERNIE完成从快递单中抽取姓名、电话、省、市、区、详细地址等内容，形成结构化信息。辅助物流行业从业者进行有效信息的提取，从而降低客户填单的成本。

2017年之前，对文本处理主要依赖于序列模型Recurrent Neural Network (RNN).

基于BiGRU+CRF的快递单信息抽取项目介绍了如何使用序列模型完成快递单信息抽取任务。

随着深度学习的发展，模型参数数量迅速增加。为了训练这些参数，需要更大的数据集，但对于大部分NLP任务，构建大规模标注数据集很难且成本高。相反，大规模未标注语料库的构建相对容易。因此，我们可以利用这些数据先学习到好的表示，再将其用于其他任务中。最近的研究表明，基于大规模未标注语料库的预训练模型（Pretrained Models, PTM）在NLP任务中表现出色。现在，随着计算能力和训练技巧的提高，PTM也得以不断发展和进步。

本示例展示了以ERNIE(Enhanced Representation through Knowledge Integration)为代表的预训练模型如何Finetune完成序列标注任务。

二、代码

在百度飞桨平台的CodeLab跑就行，类似于Google的colab平台。

2.1前期准备

安装环境

pip install --upgrade paddlenlp

  1. # 下载并解压数据集
  2. from paddle.utils.download import get_path_from_url
  3. URL = "https://paddlenlp.bj.bcebos.com/paddlenlp/datasets/waybill.tar.gz"
  4. get_path_from_url(URL, "./")
  5.  
  6. # 查看预测的数据
  7. !head -n 5 data/test.txt

  1. from functools import partial
  2.  
  3. import paddle
  4. from paddlenlp.datasets import MapDataset
  5. from paddlenlp.data import Stack, Tuple, Pad
  6. from paddlenlp.transformers import ErnieTokenizer, ErnieForTokenClassification
  7. from paddlenlp.metrics import ChunkEvaluator
  8. from utils import convert_example, evaluate, predict, load_dict

2.2加载自定义数据集

推荐使用MapDataset()自定义数据集。

  1. def load_dataset(datafiles):
  2.     def read(data_path):
  3.         with open(data_path, 'r', encoding='utf-8') as fp:
  4.             next(fp)  # Skip header
  5.             for line in fp.readlines():
  6.                 words, labels = line.strip('\n').split('\t')
  7.                 words = words.split('\002')
  8.                 labels = labels.split('\002')
  9.                 yield words, labels
  10.  
  11.     if isinstance(datafiles, str):
  12.         return MapDataset(list(read(datafiles)))
  13.     elif isinstance(datafiles, list) or isinstance(datafiles, tuple):
  14.         return [MapDataset(list(read(datafile))) for datafile in datafiles]
  15.  
  16. # Create dataset, tokenizer and dataloader.
  17. train_ds, dev_ds, test_ds = load_dataset(datafiles=(
  18.         './data/train.txt', './data/dev.txt', './data/test.txt'))
  1. for i in range(5):
  2.     print(train_ds[i])

每条数据包含一句文本和这个文本中每个汉字以及数字对应的label标签。

之后，还需要对输入句子进行数据处理，如切词，映射词表id等。

 2.3数据处理

预训练模型ERNIE对中文数据的处理是以字为单位。PaddleNLP对于各种预训练模型已经内置了相应的tokenizer。指定想要使用的模型名字即可加载对应的tokenizer。

tokenizer作用为将原始输入文本转化成模型model可以接受的输入数据形式。

  1. label_vocab = load_dict('./data/tag.dic')
  2. tokenizer = ErnieTokenizer.from_pretrained('ernie-1.0')
  3.  
  4. trans_func = partial(convert_example, tokenizer=tokenizer, label_vocab=label_vocab)
  5.  
  6. train_ds.map(trans_func)
  7. dev_ds.map(trans_func)
  8. test_ds.map(trans_func)
  9. print (train_ds[0])

2.4数据读入

使用paddle.io.DataLoader接口多线程异步加载数据。

  1. ignore_label = -1
  2. batchify_fn = lambda samples, fn=Tuple(
  3.     Pad(axis=0, pad_val=tokenizer.pad_token_id),  # input_ids
  4.     Pad(axis=0, pad_val=tokenizer.pad_token_type_id),  # token_type_ids
  5.     Stack(),  # seq_len
  6.     Pad(axis=0, pad_val=ignore_label)  # labels
  7. ): fn(samples)
  8.  
  9. train_loader = paddle.io.DataLoader(
  10.     dataset=train_ds,
  11.     batch_size=36,
  12.     return_list=True,
  13.     collate_fn=batchify_fn)
  14. dev_loader = paddle.io.DataLoader(
  15.     dataset=dev_ds,
  16.     batch_size=36,
  17.     return_list=True,
  18.     collate_fn=batchify_fn)
  19. test_loader = paddle.io.DataLoader(
  20.     dataset=test_ds,
  21.     batch_size=36,
  22.     return_list=True,
  23.     collate_fn=batchify_fn)

2.5PaddleNLP一键加载预训练模型

快递单信息抽取本质是一个序列标注任务，PaddleNLP对于各种预训练模型已经内置了对于下游任务文本分类Fine-tune网络。以下教程以ERNIE为预训练模型完成序列标注任务。

paddlenlp.transformers.ErnieForTokenClassification()一行代码即可加载预训练模型ERNIE用于序列标注任务的fine-tune网络。其在ERNIE模型后拼接上一个全连接网络进行分类。

paddlenlp.transformers.ErnieForTokenClassification.from_pretrained()方法只需指定想要使用的模型名称和文本分类的类别数即可完成定义模型网络。

  1. # Define the model netword and its loss
  2. model = ErnieForTokenClassification.from_pretrained("ernie-1.0", num_classes=len(label_vocab))

PaddleNLP不仅支持ERNIE预训练模型，还支持BERT、RoBERTa、Electra等预训练模型。 下表汇总了目前PaddleNLP支持的各类预训练模型。您可以使用PaddleNLP提供的模型，完成文本分类、序列标注、问答等任务。同时我们提供了众多预训练模型的参数权重供用户使用，其中包含了二十多种中文语言模型的预训练权重。中文的预训练模型有bert-base-chinese, bert-wwm-chinese, bert-wwm-ext-chinese, ernie-1.0, ernie-tiny, gpt2-base-cn, roberta-wwm-ext, roberta-wwm-ext-large, rbt3, rbtl3, chinese-electra-base, chinese-electra-small, chinese-xlnet-base, chinese-xlnet-mid, chinese-xlnet-large, unified_transformer-12L-cn, unified_transformer-12L-cn-luge等。

更多预训练模型参考：PaddleNLP Transformer API。

更多预训练模型fine-tune下游任务使用方法，请参考：examples。

2.6设置Fine-Tune优化策略，模型配置

适用于ERNIE/BERT这类Transformer模型的迁移优化学习率策略为warmup的动态学习率。

  1. metric = ChunkEvaluator(label_list=label_vocab.keys(), suffix=True)
  2. loss_fn = paddle.nn.loss.CrossEntropyLoss(ignore_index=ignore_label)
  3. optimizer = paddle.optimizer.AdamW(learning_rate=2e-5, parameters=model.parameters())

2.7模型训练与评估

模型训练的过程通常有以下步骤：

  1. 从dataloader中取出一个batch data
  2. 将batch data喂给model，做前向计算
  3. 将前向计算结果传给损失函数，计算loss。将前向计算结果传给评价方法，计算评价指标。
  4. loss反向回传，更新梯度。重复以上步骤。

每训练一个epoch时，程序将会评估一次，评估当前模型训练的效果。

  1. step = 0
  2. for epoch in range(10):
  3.     for idx, (input_ids, token_type_ids, length, labels) in enumerate(train_loader):
  4.         logits = model(input_ids, token_type_ids)
  5.         loss = paddle.mean(loss_fn(logits, labels))
  6.         loss.backward()
  7.         optimizer.step()
  8.         optimizer.clear_grad()
  9.         step += 1
  10.         print("epoch:%d - step:%d - loss: %f" % (epoch, step, loss))
  11.     evaluate(model, metric, dev_loader)
  12.  
  13.     paddle.save(model.state_dict(),
  14.                 './ernie_result/model_%d.pdparams' % step)
  15. # model.save_pretrained('./checkpoint')
  16. # tokenizer.save_pretrained('./checkpoint')

训练十轮，由于没有gpu所以有点慢运行的（6124秒hhh）。

2.8模型预测

训练保存好的模型，即可用于预测。如以下示例代码自定义预测数据，调用predict()函数即可一键预测。

  1. preds = predict(model, test_loader, test_ds, label_vocab)
  2. file_path = "ernie_results.txt"
  3. with open(file_path, "w", encoding="utf8") as fout:
  4.     fout.write("\n".join(preds))
  5. # Print some examples
  6. print(
  7.     "The results have been saved in the file: %s, some examples are shown below: "
  8.     % file_path)
  9. print("\n".join(preds[:10]))

三、总结

ERNIE是百度提出的一种基于持续学习技术的语义理解框架，该框架结合了大规模数据的预训练和多源知识的增强，能够不断吸收来自海量文本数据的语言信息，从而不断提升模型的效果。ERNIE在多个典型的NLP任务上取得了最好的结果，并在多个国际权威评测中获得了多项冠军。ERNIE技术获得了多个荣誉，包括中国人工智能学会优秀科技成果奖和世界人工智能大会最高荣誉SAIL奖。ERNIE在工业界也得到了广泛应用，例如搜索引擎、新闻推荐、广告系统、语音交互和智能客服等领域。全球顶级科技商业杂志《麻省理工科技评论》也报道了ERNIE相关的创新成果，相关技术也被国际顶级学术会议收录。

可以看出ernir1.0在实体抽取项目上准确率高达99%，看得出预训练模型将会是未来的主流。

原文：

飞桨AI Studio - 人工智能学习与实训社区

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/337066
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号