English
Related papers

Related papers: Can Neural Network Memorization Be Localized?

200 papers

Memorisation is a natural part of learning from real-world data: neural models pick up on atypical input-output combinations and store those training examples in their parameter space. That this happens is well-known, but how and where are…

Computation and Language · Computer Science 2024-08-12 Verna Dankers , Ivan Titov

The neural network memorization problem is to study the expressive power of neural networks to interpolate a finite dataset. Although memorization is widely believed to have a close relationship with the strong generalizability of deep…

Machine Learning · Computer Science 2024-11-04 Lijia Yu , Xiao-Shan Gao , Lijun Zhang , Yibo Miao

Over-parameterized deep neural networks are able to achieve excellent training accuracy while maintaining a small generalization error. It has also been found that they are able to fit arbitrary labels, and this behaviour is referred to as…

Machine Learning · Computer Science 2021-12-17 Futong Liu , Tao Lin , Martin Jaggi

Can we localize the weights and mechanisms used by a language model to memorize and recite entire paragraphs of its training data? In this paper, we show that while memorization is spread across multiple layers and model components,…

Computation and Language · Computer Science 2024-04-01 Niklas Stoehr , Mitchell Gordon , Chiyuan Zhang , Owen Lewis

In practice, deep neural networks are often able to easily interpolate their training data. To understand this phenomenon, many works have aimed to quantify the memorization capacity of a neural network architecture: the largest number of…

Machine Learning · Statistics 2024-12-09 Sjoerd Dirksen , Patrick Finke , Martin Genzel

Generative models have recently been explored for synthesizing neural network weights. These approaches take neural network checkpoints as training data and aim to generate high-performing weights during inference. In this work, we examine…

Machine Learning · Computer Science 2025-10-06 Boya Zeng , Yida Yin , Zhiqiu Xu , Zhuang Liu

Deep Neural Networks (DNNs) are increasingly being used in software engineering and code intelligence tasks. These are powerful tools that are capable of learning highly generalizable patterns from large datasets through millions of…

Machine Learning · Computer Science 2022-09-15 Md Rafiqul Islam Rabin , Aftab Hussain , Mohammad Amin Alipour , Vincent J. Hellendoorn

Overparameterized deep networks that generalize well have been key to the dramatic success of deep learning in recent years. The reasons for their remarkable ability to generalize are not well understood yet. When class labels in the…

Machine Learning · Computer Science 2026-02-03 Simran Ketha , Venkatakrishnan Ramaswamy

Understanding how large neural networks avoid memorizing training data is key to explaining their high generalization performance. To examine the structure of when and where memorization occurs in a deep network, we use a recently developed…

Machine Learning · Computer Science 2021-06-01 Cory Stephenson , Suchismita Padhy , Abhinav Ganesh , Yue Hui , Hanlin Tang , SueYeon Chung

Neural networks often learn simple explanations that fit the majority of the data while memorizing exceptions that deviate from these explanations.This behavior leads to poor generalization when the learned explanations rely on spurious…

Machine Learning · Computer Science 2024-12-11 Reza Bayat , Mohammad Pezeshki , Elvis Dohmatob , David Lopez-Paz , Pascal Vincent

We examine the role of memorization in deep learning, drawing connections to capacity, generalization, and adversarial robustness. While deep networks are capable of memorizing noise data, our results suggest that they tend to prioritize…

Despite the empirical advances of deep learning across a variety of learning tasks, our theoretical understanding of its success is still very restricted. One of the key challenges is the overparametrized nature of modern models, enabling…

Machine Learning · Computer Science 2023-02-24 Sotiris Anagnostidis , Gregor Bachmann , Lorenzo Noci , Thomas Hofmann

When do diffusion models reproduce their training data, and when are they able to generate samples beyond it? A practically relevant theoretical understanding of this interplay between memorization and generalization may significantly…

Machine Learning · Computer Science 2025-08-26 Sam Buchanan , Druv Pai , Yi Ma , Valentin De Bortoli

Deep learning models frequently exploit spurious features in training data to achieve low training error, often resulting in poor generalization when faced with shifted testing distributions. To address this issue, various methods from…

Machine Learning · Computer Science 2025-02-11 Geraldin Nanfack , Eugene Belilovsky

Models for learning probability distributions such as generative models and density estimators behave quite differently from models for learning functions. One example is found in the memorization phenomenon, namely the ultimate convergence…

Machine Learning · Statistics 2021-03-03 Hongkang Yang , Weinan E

We investigate how Large Language Models (LLMs) distinguish between memorization and generalization at the neuron level. Through carefully designed tasks, we identify distinct neuron subsets responsible for each behavior. Experiments on…

Computation and Language · Computer Science 2025-07-10 Ko-Wei Huang , Yi-Fu Fu , Ching-Yu Tsai , Yu-Chieh Tu , Tzu-Ling Cheng , Cheng-Yu Lin , Yi-Ting Yang , Heng-Yi Liu , Keng-Te Liao , Da-Cheng Juan , Shou-De Lin

Deep learning algorithms are well-known to have a propensity for fitting the training data very well and often fit even outliers and mislabeled data points. Such fitting requires memorization of training data labels, a phenomenon that has…

Machine Learning · Computer Science 2020-08-11 Vitaly Feldman , Chiyuan Zhang

We study the interplay between memorization and generalization of overparameterized networks in the extreme case of a single training example and an identity-mapping task. We examine fully-connected and convolutional networks (FCN and CNN),…

Machine Learning · Statistics 2020-01-10 Chiyuan Zhang , Samy Bengio , Moritz Hardt , Michael C. Mozer , Yoram Singer

It has been observed \citep{zhang2016understanding} that deep neural networks can memorize: they achieve 100\% accuracy on training data. Recent theoretical results explained such behavior in highly overparametrized regimes, where the…

Machine Learning · Computer Science 2019-09-27 Rong Ge , Runzhe Wang , Haoyu Zhao

Neural networks are known to exploit spurious artifacts (or shortcuts) that co-occur with a target label, exhibiting heuristic memorization. On the other hand, networks have been shown to memorize training examples, resulting in…

Machine Learning · Computer Science 2024-02-05 Rachit Bansal , Danish Pruthi , Yonatan Belinkov
‹ Prev 1 2 3 10 Next ›