Fine-tuning BERT for Low-Resource Natural Language Understanding via Active Learning

Autor: Grießhaber, Daniel, Maucher, Johannes, Vu, Ngoc Thang
Rok vydání: 2020
Předmět:
Druh dokumentu: Working Paper
Popis: Recently, leveraging pre-trained Transformer based language models in down stream, task specific models has advanced state of the art results in natural language understanding tasks. However, only a little research has explored the suitability of this approach in low resource settings with less than 1,000 training data points. In this work, we explore fine-tuning methods of BERT -- a pre-trained Transformer based language model -- by utilizing pool-based active learning to speed up training while keeping the cost of labeling new data constant. Our experimental results on the GLUE data set show an advantage in model performance by maximizing the approximate knowledge gain of the model when querying from the pool of unlabeled data. Finally, we demonstrate and analyze the benefits of freezing layers of the language model during fine-tuning to reduce the number of trainable parameters, making it more suitable for low-resource settings.
Comment: COLING'2020
Databáze: arXiv