Ändra sökning
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf
BRANCH-GAN: IMPROVING TEXT GENERATION WITH (NOT SO) LARGE LANGUAGE MODELS
RISE Research Institutes of Sweden, Säkerhet och transport, Elektrifiering och pålitlighet.ORCID-id: 0000-0003-2811-7481
RISE Research Institutes of Sweden.
RISE Research Institutes of Sweden.
AI Sweden, Sweden.ORCID-id: 0000-0001-5100-0535
Visa övriga samt affilieringar
2024 (Engelska)Ingår i: 12th International Conference on Learning Representations, ICLR 2024, International Conference on Learning Representations, ICLR , 2024Konferensbidrag, Publicerat paper (Refereegranskat)
Abstract [en]

The current advancements in open domain text generation have been spearheaded by Transformer-based large language models. Leveraging efficient parallelization and vast training datasets, these models achieve unparalleled text generation capabilities. Even so, current models are known to suffer from deficiencies such as repetitive texts, looping issues, and lack of robustness. While adversarial training through generative adversarial networks (GAN) is a proposed solution, earlier research in this direction has predominantly focused on older architectures, or narrow tasks. As a result, this approach is not yet compatible with modern language models for open-ended text generation, leading to diminished interest within the broader research community. We propose a computationally efficient GAN approach for sequential data that utilizes the parallelization capabilities of Transformer models. Our method revolves around generating multiple branching sequences from each training sample, while also incorporating the typical next-step prediction loss on the original data. In this way, we achieve a dense reward and loss signal for both the generator and the discriminator, resulting in a stable training dynamic. We apply our training method to pre-trained language models, using data from their original training set but less than 0.01% of the available data. A comprehensive human evaluation shows that our method significantly improves the quality of texts generated by the model while avoiding the previously reported sparsity problems of GAN approaches. Even our smaller models outperform larger original baseline models with more than 16 times the number of parameters. Finally, we corroborate previous claims that perplexity on held-out data is not a sufficient metric for measuring the quality of generated texts.

Ort, förlag, år, upplaga, sidor
International Conference on Learning Representations, ICLR , 2024.
Nyckelord [en]
Computational linguistics; ’current; Computationally efficient; Current modeling; Language model; Modern languages; Parallelizations; Research communities; Sequential data; Text generations; Training dataset; Generative adversarial networks
Nationell ämneskategori
Data- och informationsvetenskap
Identifikatorer
URN: urn:nbn:se:ri:diva-75007Scopus ID: 2-s2.0-85200551619OAI: oai:DiVA.org:ri-75007DiVA, id: diva2:1896475
Konferens
12th International Conference on Learning Representations, ICLR 2024.Vienna, Austria. 7 May 2024 through 11 May 2024
Anmärkning

The research presented in this paper was supported by the Swedish Research Council (grant no. 2022-02909) and by a donation from Meta.

Tillgänglig från: 2024-09-10 Skapad: 2024-09-10 Senast uppdaterad: 2025-09-23Bibliografiskt granskad

Open Access i DiVA

Fulltext saknas i DiVA

Scopus

Person

Carlsson, FredrikSahlgren, MagnusNivre, Joakim

Sök vidare i DiVA

Av författaren/redaktören
Carlsson, FredrikSahlgren, MagnusNivre, Joakim
Av organisationen
Elektrifiering och pålitlighetRISE Research Institutes of SwedenDatavetenskap
Data- och informationsvetenskap

Sök vidare utanför DiVA

GoogleGoogle Scholar

urn-nbn

Altmetricpoäng

urn-nbn
Totalt: 492 träffar
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf