Logo: to the web site of Uppsala University

uu.sePublications from Uppsala University
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Memory-Augmenting Decoder-Only Language Models through Encoders: (Student Abstract)
Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology. Uppsala University, Disciplinary Domain of Medicine and Pharmacy, Faculty of Medicine, Department of Women's and Children's Health.
Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Division Vi3. Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Computerized Image Analysis and Human-Computer Interaction.ORCID iD: 0000-0002-3309-3552
2024 (English)In: THIRTY-EIGTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 38 NO 21 / [ed] Wooldridge, M Dy, J Natarajan, S, Assoc Advancement Artificial Intelligence , 2024, Vol. 38, p. 23494-23496, article id 21Conference paper, Published paper (Refereed)
Abstract [en]

The Transformer architecture has seen a lot of attention in recent years also thanks to its ability to scale well and allow massive parallelism during training. This has made possible the development of Language Models (LMs) of increasing size and the discovery of latent abilities that completely outclass traditional methods e.g. rule-based systems. However, they also introduced new issues, like their inability to retain the history of previous interactions due to their stateless nature or the difficulty in controlling their generation. Different attempts have been made to address these issues, e.g. a 'brute force' approach to solving the memory issue is to include the full conversation history in the context window, a solution that is limited by the quadratic scalability of Transformers. In this work, we explore computationally practical solutions to the memory problem. We propose to augment the decoder-only architecture of (most) Large LMs with a (relatively small) memory encoder. Its output is prepended to the decoder's input in a similar fashion to recent works in Adapters and the original Transformer architecture. Initial experiments show promising results, however future work is needed to compare with State-of-the-Art methods.

Place, publisher, year, edition, pages
Assoc Advancement Artificial Intelligence , 2024. Vol. 38, p. 23494-23496, article id 21
Series
AAAI Conference on Artificial Intelligence, ISSN 2159-5399, E-ISSN 2374-3468
National Category
Computer Sciences Software Engineering
Identifiers
URN: urn:nbn:se:uu:diva-537573ISI: 001239989100140OAI: oai:DiVA.org:uu-537573DiVA, id: diva2:1895322
Conference
38th AAAI Conference on Artificial Intelligence (AAAI) / 36th Conference on Innovative Applications of Artificial Intelligence / 14th Symposium on Educational Advances in Artificial Intelligence, FEB 20-27, 2024, Vancouver, CANADA
Funder
Swedish Research Council, 2022-06725Available from: 2024-09-05 Created: 2024-09-05 Last updated: 2024-09-05Bibliographically approved

Open Access in DiVA

No full text in DiVA

Authority records

Galatolo, AlessioWinkle, Katie

Search in DiVA

By author/editor
Galatolo, AlessioWinkle, Katie
By organisation
Department of Information TechnologyDepartment of Women's and Children's HealthDivision Vi3Computerized Image Analysis and Human-Computer Interaction
Computer SciencesSoftware Engineering

Search outside of DiVA

GoogleGoogle Scholar

urn-nbn

Altmetric score

urn-nbn
Total: 121 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf