Preprint
Large Language Models

DeBERTa

June 1, 2020

0

Citations

0

Influential Citations

Venue

2020

Year

Abstract

Enhances BERT and RoBERTa through disentangled attention mechanisms, an enhanced mask decoder, and virtual adversarial training.

Analysis

Why This Paper Matters

DeBERTa addresses key limitations in BERT and RoBERTa by refining the attention mechanism and pre-training objectives. The disentangled attention approach separates content and position information, allowing the model to learn more nuanced relationships between tokens. This is particularly important for tasks requiring precise positional understanding, such as question answering and named entity recognition. Additionally, the enhanced mask decoder improves the model's ability to predict masked tokens by leveraging both content and position cues more effectively. Virtual adversarial training further boosts robustness, making DeBERTa more reliable in noisy or adversarial settings. These innovations collectively push the boundaries of what pre-trained language models can achieve, offering a stronger foundation for downstream NLP applications.

Technical Contributions

  • Disentangled Attention: DeBERTa computes attention scores using separate content-to-content, content-to-position, and position-to-content matrices, enabling the model to capture both semantic and positional dependencies independently.
  • Enhanced Mask Decoder: The decoder uses both the hidden states and the position embeddings of masked tokens to predict the original token, improving the quality of learned representations during pre-training.
  • Virtual Adversarial Training: A regularization technique that adds small perturbations to the input embeddings and encourages the model to produce consistent outputs, enhancing generalization and robustness.

Results

While the abstract does not provide specific numerical results, DeBERTa reportedly outperforms BERT and RoBERTa on standard NLP benchmarks such as GLUE and SQuAD. The improvements are attributed to the combined effect of disentangled attention, the enhanced mask decoder, and virtual adversarial training. Practitioners can expect gains in tasks that require fine-grained understanding of token positions and robust performance under distribution shift.

Significance

DeBERTa's contributions have broader implications for the design of Transformer-based models. The disentangled attention mechanism offers a principled way to handle positional information, which can be adapted to other architectures like vision transformers or multimodal models. The enhanced mask decoder provides a template for improving pre-training objectives beyond simple masked language modeling. Virtual adversarial training is a well-known technique, but its application in the context of large-scale pre-training demonstrates its value in improving model robustness. Overall, DeBERTa represents a meaningful step forward in the evolution of pre-trained language models, influencing both research and practical deployments in NLP.