TADPOLE: Task ADapted Pre-Training via AnOmaLy DEtection

Vivek Madan, Ashish Khetan, Zohar Karnin


Abstract
The paradigm of pre-training followed by finetuning has become a standard procedure for NLP tasks, with a known problem of domain shift between the pre-training and downstream corpus. Previous works have tried to mitigate this problem with additional pre-training, either on the downstream corpus itself when it is large enough, or on a manually curated unlabeled corpus of a similar domain. In this paper, we address the problem for the case when the downstream corpus is too small for additional pre-training. We propose TADPOLE, a task adapted pre-training framework based on data selection techniques adapted from Domain Adaptation. We formulate the data selection as an anomaly detection problem that unlike existing methods works well when the downstream corpus is limited in size. It results in a scalable and efficient unsupervised technique that eliminates the need for any manual data curation. We evaluate our framework on eight tasks across four different domains: Biomedical, Computer Science, News, and Movie reviews, and compare its performance against competitive baseline techniques from the area of Domain Adaptation. Our framework outperforms all the baseline methods. On small datasets with less than 5K training examples, we get a gain of 1.82% in performance with additional pre-training for only 5% steps compared to the originally pre-trained models. It also compliments some of the other techniques such as data augmentation known for boosting performance when downstream corpus is small; highest performance is achieved when data augmentation is combined with task adapted pre-training.
Anthology ID:
2021.emnlp-main.463
Volume:
Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing
Month:
November
Year:
2021
Address:
Online and Punta Cana, Dominican Republic
Editors:
Marie-Francine Moens, Xuanjing Huang, Lucia Specia, Scott Wen-tau Yih
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
5732–5746
Language:
URL:
https://aclanthology.org/2021.emnlp-main.463
DOI:
10.18653/v1/2021.emnlp-main.463
Bibkey:
Cite (ACL):
Vivek Madan, Ashish Khetan, and Zohar Karnin. 2021. TADPOLE: Task ADapted Pre-Training via AnOmaLy DEtection. In Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing, pages 5732–5746, Online and Punta Cana, Dominican Republic. Association for Computational Linguistics.
Cite (Informal):
TADPOLE: Task ADapted Pre-Training via AnOmaLy DEtection (Madan et al., EMNLP 2021)
Copy Citation:
PDF:
https://aclanthology.org/2021.emnlp-main.463.pdf
Video:
 https://aclanthology.org/2021.emnlp-main.463.mp4
Data
IMDb Movie ReviewsSSTSciERC