Fine-tuning BERT for Low-Resource Natural Language Understanding via Active Learning (2012.02462v1)

Abstract: Recently, leveraging pre-trained Transformer based LLMs 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 LLM -- 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 LLM during fine-tuning to reduce the number of trainable parameters, making it more suitable for low-resource settings.