Capacity Matters: a Proof-of-Concept for Transformer Memorization on Real-World Data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Changalidis, Anton, Härmä, Aki
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913898953703424
author Changalidis, Anton
Härmä, Aki
author_facet Changalidis, Anton
Härmä, Aki
contents This paper studies how the model architecture and data configurations influence the empirical memorization capacity of generative transformers. The models are trained using synthetic text datasets derived from the Systematized Nomenclature of Medicine (SNOMED) knowledge graph: triplets, representing static connections, and sequences, simulating complex relation patterns. The results show that embedding size is the primary determinant of learning speed and capacity, while additional layers provide limited benefits and may hinder performance on simpler datasets. Activation functions play a crucial role, and Softmax demonstrates greater stability and capacity. Furthermore, increasing the complexity of the data set seems to improve the final memorization. These insights improve our understanding of transformer memory mechanisms and provide a framework for optimizing model design with structured real-world data.
format Preprint
id arxiv_https___arxiv_org_abs_2506_14704
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Capacity Matters: a Proof-of-Concept for Transformer Memorization on Real-World Data
Changalidis, Anton
Härmä, Aki
Computation and Language
This paper studies how the model architecture and data configurations influence the empirical memorization capacity of generative transformers. The models are trained using synthetic text datasets derived from the Systematized Nomenclature of Medicine (SNOMED) knowledge graph: triplets, representing static connections, and sequences, simulating complex relation patterns. The results show that embedding size is the primary determinant of learning speed and capacity, while additional layers provide limited benefits and may hinder performance on simpler datasets. Activation functions play a crucial role, and Softmax demonstrates greater stability and capacity. Furthermore, increasing the complexity of the data set seems to improve the final memorization. These insights improve our understanding of transformer memory mechanisms and provide a framework for optimizing model design with structured real-world data.
title Capacity Matters: a Proof-of-Concept for Transformer Memorization on Real-World Data
topic Computation and Language
url https://arxiv.org/abs/2506.14704