DA-Cramming: Enhancing Cost-Effective Language Model Pretraining with Dependency Agreement Integration

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Kuo, Martin, Zhang, Jianyi, Li, Dongting, Chen, Yiran
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908967001653248
author Kuo, Martin
Zhang, Jianyi
Li, Dongting
Chen, Yiran
author_facet Kuo, Martin
Zhang, Jianyi
Li, Dongting
Chen, Yiran
contents Pretraining language models is still a challenge for many researchers due to its substantial computational costs. As such, there is growing interest in developing more affordable pretraining methods. One notable advancement in this area is the Cramming technique (Geiping and Goldstein, 2022), which enables the pretraining of BERT-style language models using just one GPU in a single day. Building on this innovative approach, we introduce the Dependency Agreement Cramming (DA-Cramming), an efficient framework that integrates information about dependency agreements into the pretraining process. Unlike existing methods that leverage similar semantic information during finetuning, our approach represents a pioneering effort focusing on enhancing the foundational language understanding with semantic information during pretraining. We meticulously design a dual-stage pretraining work flow with four dedicated submodels to capture representative dependency agreements at the chunk level, effectively transforming these agreements into embeddings to benefit the pretraining. Extensive empirical results demonstrate that our method significantly outperforms previous methods across various tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2311_04799
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle DA-Cramming: Enhancing Cost-Effective Language Model Pretraining with Dependency Agreement Integration
Kuo, Martin
Zhang, Jianyi
Li, Dongting
Chen, Yiran
Computation and Language
Artificial Intelligence
Pretraining language models is still a challenge for many researchers due to its substantial computational costs. As such, there is growing interest in developing more affordable pretraining methods. One notable advancement in this area is the Cramming technique (Geiping and Goldstein, 2022), which enables the pretraining of BERT-style language models using just one GPU in a single day. Building on this innovative approach, we introduce the Dependency Agreement Cramming (DA-Cramming), an efficient framework that integrates information about dependency agreements into the pretraining process. Unlike existing methods that leverage similar semantic information during finetuning, our approach represents a pioneering effort focusing on enhancing the foundational language understanding with semantic information during pretraining. We meticulously design a dual-stage pretraining work flow with four dedicated submodels to capture representative dependency agreements at the chunk level, effectively transforming these agreements into embeddings to benefit the pretraining. Extensive empirical results demonstrate that our method significantly outperforms previous methods across various tasks.
title DA-Cramming: Enhancing Cost-Effective Language Model Pretraining with Dependency Agreement Integration
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2311.04799