🤖 AI Summary
To address the high computational cost and weak cross-lingual generalization of multilingual models, this paper proposes UniBERT, a lightweight multilingual language model. Methodologically, it innovatively integrates gradient-projection-based adversarial training (FGSM/PGD) with teacher-student knowledge distillation into a unified masked language modeling framework, jointly optimized across Wikipedia corpora in 107 languages to strengthen language-agnostic representation learning. UniBERT employs shared subword tokenization and multilingual tokenization, significantly reducing both pretraining and inference overhead. Empirically, it achieves an average relative performance gain of 7.72% (p = 0.0181) over strong baselines on four cross-lingual tasks—named entity recognition, natural language inference, question answering, and semantic textual similarity—demonstrating the efficacy of synergistically enhancing adversarial robustness and knowledge transfer for multilingual representation learning.
📝 Abstract
This paper presents UniBERT, a compact multilingual language model that leverages an innovative training framework integrating three components: masked language modeling, adversarial training, and knowledge distillation. Pre-trained on a meticulously curated Wikipedia corpus spanning 107 languages, UniBERT is designed to reduce the computational demands of large-scale models while maintaining competitive performance across various natural language processing tasks. Comprehensive evaluations on four tasks -- named entity recognition, natural language inference, question answering, and semantic textual similarity -- demonstrate that our multilingual training strategy enhanced by an adversarial objective significantly improves cross-lingual generalization. Specifically, UniBERT models show an average relative improvement of 7.72% over traditional baselines, which achieved an average relative improvement of only 1.17%, with statistical analysis confirming the significance of these gains (p-value = 0.0181). This work highlights the benefits of combining adversarial training and knowledge distillation to build scalable and robust language models, thereby advancing the field of multilingual and cross-lingual natural language processing.