掩码语言模型(Masked Language Model)mlm

本文深入解析Transformer架构及其在自然语言处理中的应用,包括Mask机制、Seq2Seq模型以及BERT预训练模型的工作原理。通过源码分析,理解如何利用Transformer提升语言理解能力。

https://www.cnblogs.com/anai/p/11645953.html
bert 论文

在这里插入图片描述

从语言模型到Seq2Seq:Transformer如戏,全靠Mask
https://zhuanlan.zhihu.com/p/69106080

深度学习 — > NLP — >Improving Language Understanding by Generative Pre-Training

https://zhuanlan.zhihu.com/p/44121378
https://zhuanlan.zhihu.com/p/32544778
https://blog.youkuaiyun.com/qq_33876194/article/details/98943383

https://zhuanlan.zhihu.com/p/93061413

[# Transformer 源码中 Mask 机制的实现
GPT解读(论文 + TensorFlow实现)

BERT源码分析(PART III)
Bert系列(三)——源码解读之Pre-trainhttps://www.jianshu.com/p/22e462f01d8c

https://www.jianshu.com/p/ff43575ab2b0

在这里插入图片描述

BERT模型中引入了掩码语言模型Masked Language ModelMLM)任务和下一句预测(Next Sentence Prediction,NSP)任务,这两项任务相辅相成,使模型在多个层面获得系统性的语言知识,为后续的下游任务迁移学习奠定了坚实的基础[^1]。 #### 掩码语言模型Masked Language Model)任务 - **核心思想**:随机掩盖输入中的部分词汇,然后让模型预测这些被掩盖词汇的原始内容。这种方式打破了传统语言模型只能单向建模的限制,使模型能够深度掌握词级语义关系。 - **作用**:通过该任务,模型可以学习到词汇之间的上下文依赖关系,在词汇、句法层面获得丰富的语言知识,为处理各种涉及语义理解的下游任务提供基础。例如在文本分类、命名实体识别等任务中,模型对词汇语义的准确理解至关重要。 以下是一个简单的示例,展示如何使用Hugging Face的`transformers`库进行MLM任务的预测: ```python from transformers import BertTokenizer, BertForMaskedLM import torch # 加载预训练的BERT模型和分词器 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased') model = BertForMaskedLM.from_pretrained('bert-base-uncased') # 输入包含[MASK]的句子 text = "The [MASK] is a large mammal." inputs = tokenizer(text, return_tensors='pt') # 进行预测 with torch.no_grad(): outputs = model(**inputs) # 获取预测结果 logits = outputs.logits mask_token_index = (inputs.input_ids == tokenizer.mask_token_id)[0].nonzero(as_tuple=True)[0] predicted_token_id = logits[0, mask_token_index].argmax(axis=-1) predicted_word = tokenizer.decode(predicted_token_id) print(f"预测的词是: {predicted_word}") ``` #### 下一句预测(Next Sentence Prediction)任务 - **核心思想**:要求模型判断两个句子是否在原文中相邻。具体做法是,在预训练时,一半的情况下选择真正相邻的两个句子作为输入,另一半情况选择随机的两个句子。 - **作用**:促使模型在句子层面学习到更丰富的篇章语义,让模型能够理解句子之间的逻辑关系和连贯性,使模型在处理涉及篇章结构理解的下游任务,如问答系统、文本生成等时表现更好。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值