CDLM: Cross-Document Language Modeling

Avi Caciularu, Arman Cohan, Iz Beltagy, Matthew E. Peters, Arie Cattan, Ido Dagan

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

40 Scopus citations

Abstract

We introduce a new pretraining approach geared for multi-document language modeling, incorporating two key ideas into the masked language modeling self-supervised objective. First, instead of considering documents in isolation, we pretrain over sets of multiple related documents, encouraging the model to learn cross-document relationships. Second, we improve over recent long-range transformers by introducing dynamic global attention that has access to the entire input to predict masked tokens. We release CDLM (Cross-Document Language Model), a new general language model for multi-document setting that can be easily applied to downstream tasks. Our extensive analysis shows that both ideas are essential for the success of CDLM, and work in synergy to set new state-of-the-art results for several multi-text tasks.

Original languageEnglish
Title of host publicationFindings of the Association for Computational Linguistics, Findings of ACL
Subtitle of host publicationEMNLP 2021
EditorsMarie-Francine Moens, Xuanjing Huang, Lucia Specia, Scott Wen-Tau Yih
PublisherAssociation for Computational Linguistics (ACL)
Pages2648-2662
Number of pages15
ISBN (Electronic)9781955917100
StatePublished - 2021
Event2021 Findings of the Association for Computational Linguistics, Findings of ACL: EMNLP 2021 - Punta Cana, Dominican Republic
Duration: 7 Nov 202111 Nov 2021

Publication series

NameFindings of the Association for Computational Linguistics, Findings of ACL: EMNLP 2021

Conference

Conference2021 Findings of the Association for Computational Linguistics, Findings of ACL: EMNLP 2021
Country/TerritoryDominican Republic
CityPunta Cana
Period7/11/2111/11/21

Bibliographical note

Publisher Copyright:
© 2021 Association for Computational Linguistics.

Funding

The use of cross-document attention has been We thank Doug Downey and Luke Zettlemoyer for recently explored by the Cross-Document Atten-fruitful discussions and helpful feedback, and Yoav tion (CDA) (Zhou et al., 2020). CDA specifi-Goldberg for helping us connect with collaborators cally encodes two documents, using hierarchical on this project. The work described herein was attention networks, with the addition of cross at-supported in part by grants from Intel Labs, the tention between documents, and makes similar-Israel Science Foundation grant 1951/17, the Israeli ity decision between them. Similarly, the recent Ministry of Science and Technology, and the NSF DCS model (Ginzburg et al., 2021) suggested a Grant OIA-2033558. cross-document finetuning scheme for unsuper- vised document-pair matching method (processing only two documents at once). Our CDLM, by contrast, is a general pretrained language model that can be applied to a variety of multi-document downstream tasks, without restrictions on the number of input documents, as long as they fit the input length of the Longformer.

FundersFunder number
Intel Labs
National Science FoundationOIA-2033558
Ministry of science and technology, Israel
makes similar-Israel Science Foundation1951/17

    Fingerprint

    Dive into the research topics of 'CDLM: Cross-Document Language Modeling'. Together they form a unique fingerprint.

    Cite this