Increasing The Performance of Cognitively Inspired Data-Efficient Language Models via Implicit Structure Building (2310.20589v1)

Abstract: In this paper, we describe our submission to the BabyLM Challenge 2023 shared task on data-efficient LLM (LM) pretraining (Warstadt et al., 2023). We train transformer-based masked LLMs that incorporate unsupervised predictions about hierarchical sentence structure into the model architecture. Concretely, we use the Structformer architecture (Shen et al., 2021) and variants thereof. StructFormer models have been shown to perform well on unsupervised syntactic induction based on limited pretraining data, and to yield performance improvements over a vanilla transformer architecture (Shen et al., 2021). Evaluation of our models on 39 tasks provided by the BabyLM challenge shows promising improvements of models that integrate a hierarchical bias into the architecture at some particular tasks, even though they fail to consistently outperform the RoBERTa baseline model provided by the shared task organizers on all tasks.