掩码语言模型

技术 EN:Masked Language Model 别名:MLM 发布:2026-08-23 浏览:0
摘要
掩码语言模型(MLM)是 BERT 使用的预训练目标:随机遮盖输入中约 15% 的词元,让模型从双向上下文中预测被遮词元,从而学到深度上下文表征。

定义

掩码语言模型(Masked Language Model, MLM)是双向预训练语言模型的训练目标:随机遮盖部分输入词元,要求模型依据被遮词元左右两侧的上下文预测原词,与自回归的「从左到右预测」形成对比。 齐e慧享

原理与结构

BERT 将约 15% 的 WordPiece 词元替换为 [MASK](其中 80% 换掩码符、10% 换随机词、10% 保持原词),再用交叉熵在遮词位置计算损失。由于可见双向上下文,模型学到的是「深度双向表征」,适合理解类任务。 齐e慧享

关键事实

  • 提出:BERT(Devlin et al., 2018, arXiv:1810.04805)。
  • 预训练-微调不一致:推理时并不存在 [MASK],需要缓解策略。
  • 变体:RoBERTa 去掉 NSP 并加大数据/训练时长,效果更好。

典型应用

文本分类、命名实体识别、关系抽取、问答与句子相似度等自然语言理解任务。 齐e慧享

局限与争议

无法直接做自回归生成;[MASK] 造成的分布偏移;双向建模在生成任务上不如 decoder-only 架构。 齐e慧享

参考资料

发现内容有误或想补充资料? 贡献新词条 词条编辑指南 我要投稿(进入审核)