Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Structural Self-Supervised Objectives for Transformers

About

This thesis focuses on improving the pre-training of natural language models using unsupervised raw data to make them more efficient and aligned with downstream applications. In the first part, we introduce three alternative pre-training objectives to BERT's Masked Language Modeling (MLM), namely Random Token Substitution (RTS), Cluster-based Random Token Substitution (C-RTS), and Swapped Language Modeling (SLM). These objectives involve token swapping instead of masking, with RTS and C-RTS aiming to predict token originality and SLM predicting the original token values. Results show that RTS and C-RTS require less pre-training time while maintaining performance comparable to MLM. Surprisingly, SLM outperforms MLM on certain tasks despite using the same computational budget. In the second part, we proposes self-supervised pre-training tasks that align structurally with downstream applications, reducing the need for labeled data. We use large corpora like Wikipedia and CC-News to train models to recognize if text spans originate from the same paragraph or document in several ways. By doing continuous pre-training, starting from existing models like RoBERTa, ELECTRA, DeBERTa, BART, and T5, we demonstrate significant performance improvements in tasks like Fact Verification, Answer Sentence Selection, and Summarization. These improvements are especially pronounced when limited annotation data is available. The proposed objectives also achieve state-of-the-art results on various benchmark datasets, including FEVER (dev set), ASNQ, WikiQA, and TREC-QA, as well as enhancing the quality of summaries. Importantly, these techniques can be easily integrated with other methods without altering the internal structure of Transformer models, making them versatile for various NLP applications.

Luca Di Liello• 2023

Related benchmarks

TaskDatasetResultRank
Natural Language UnderstandingGLUE (test)
SST-2 Accuracy92.3
416
SummarizationXSum (test)
ROUGE-219.3
231
Answer SelectionWikiQA (test)
MAP0.899
149
Natural Language UnderstandingGLUE (test dev)--
81
Answer Sentence SelectionTREC-QA (test)
MAP90.5
63
Fact VerificationFEVER (dev)
Label Accuracy82.1
57
Answer Sentence SelectionASNQ (test)
P@171.3
45
Answer Sentence SelectionWikiQA
P@183.5
36
Fact VerificationFEVER (test)
LA Score75.36
32
Answer Sentence SelectionTREC-QA
P@192.6
24
Showing 10 of 12 rows

Other info

Code

Follow for update