XLNet:通过排列语言建模突破BERT的预训练局限
长期信息 · 《XLNet: Generalized Autoregressive Pretraining for Language Understanding》(2019)
在自然语言处理领域,无监督预训练是提升下游任务性能的关键。长期以来,自回归(AR)语言模型和基于去噪自动编码器的模型(如BERT)是两种最成功的方法。然而,它们各自存在明显的局限性。传统的AR模型只能单向编码上下文,难以捕捉深度双向信息;而BERT虽然能利用双向上下文,但其依赖掩码符号导致预训练与微调阶段的数据分布不一致(即“预训练-微调差异”),且假设被预测的词之间相互独立,忽略了自然语言中普遍存在的长距离依赖。
针对这些痛点,卡内基梅隆大学与谷歌AI团队提出了XLNet。这是一种广义自回归预训练方法,旨在结合AR模型和BERT的优势并规避其缺陷。XLNet的核心创新在于“排列语言建模”。不同于传统AR模型固定的前向或后向分解顺序,XLNet最大化所有可能排列顺序下的期望对数似然。这意味着对于序列中的每个位置,其上下文可以包含左右两侧的词。从统计期望上看,模型学会了利用来自所有位置的信息,从而在不破坏自回归性质的前提下实现了双向上下文的捕捉。
此外,由于XLNet保持了自回归形式,它不依赖于输入数据的损坏(如BERT的[MASK]操作),因此消除了预训练与微调之间的差异。同时,自回归目标允许使用乘积法则来分解联合概率,自然地建模了词与词之间的依赖关系,克服了BERT独立性假设的简化问题。为了进一步提升性能,XLNet还集成了Transformer-XL的段递归机制和相对位置编码,使其在处理长文本序列时表现更佳。实验表明,在可比设置下,XLNet在问答、自然语言推理、情感分析和文档排序等20项任务上均优于BERT,往往具有显著优势。
建议阅读《XLNet: Generalized Autoregressive Pretraining for Language Understanding》(2019年)。这篇论文深入剖析了BERT和传统自回归模型的优缺点,并展示了如何通过排列语言建模这一巧妙设计,在保持自回归特性的同时实现双向上下文学习。对于希望理解大规模语言模型预训练机制演进、特别是如何平衡双向性与自回归优势的读者,这是一篇必读的经典技术文献。
信息源:《XLNet: Generalized Autoregressive Pretraining for Language Understanding》(2019)