arXiv · 2211.00430
VarMAE: Pre-training of Variational Masked Autoencoder for Domain-adaptive Language Understanding
Abstract
Pre-trained language models have achieved promising performance on general benchmarks, but underperform when migrated to a specific domain. Recent works perform pre-training from scratch or continual pre-training on domain corpora. However, in many specific domains, the limited corpus can hardly support obtaining precise representations. To address this issue, we propose a novel Transformer-based language model named VarMAE for domain-adaptive language understanding. Under the masked autoencoding objective, we design a context uncertainty learning module to encode the token's context into a smooth latent distribution. The module can produce diverse and well-formed contextual representations. Experiments on science- and finance-domain NLU tasks demonstrate that VarMAE can be efficiently adapted to new domains with limited resources.
Explore related subjects
Keep this discovery
Dou Hu, Xiaolong Hou, Xiyang Du, Mengyuan Zhou, Lianxin Jiang, Yang Mo, Xiaofeng Shi. 2022-11-01. VarMAE: Pre-training of Variational Masked Autoencoder for Domain-adaptive Language Understanding. https://arxiv.org/abs/2211.00430
Cite the original work for its findings. Save a collection to share your selection of sources.