Abstract: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 parameters. At the same time, their large capacity can render them prone to memorizing data points. Recent work suggests that the memorization risk manifests especially strongly when the training dataset is noisy, involving many ambiguous or questionable samples, and memorization is the only recourse. The goal of this paper is to evaluate and compare the extent of memorization and generalization in neural code intelligence models. It aims to provide insights on how memorization may impact the learning behavior of neural models in code intelligence systems. To observe the extent of memorization in models, we add random noise to the original training dataset and use various metrics to quantify the impact of noise on various aspects of training and testing. We evaluate several state-of-the-art neural code intelligence models and benchmarks based on Java, Python, and Ruby codebases. Our results highlight important risks: millions of trainable parameters allow the neural networks to memorize anything, including noisy data, and provide a false sense of generalization. We observed all models manifest some forms of memorization. This can be potentially troublesome in most code intelligence tasks where they rely on rather noise-prone and repetitive data sources, such as code from GitHub. To the best of our knowledge, we provide the first study to quantify memorization effects in the domain of software engineering and code intelligence systems. This work raises awareness and provides new insights into important issues of training neural models in code intelligence systems that are usually overlooked by software engineering researchers.

Leveraging Unlabeled Data to Track Memorization

Learn to Forget: Memorization Elimination for Neural Networks.

Detecting Memorization in Large Language Models

Memory Aware Synapses: Learning what (not) to forget

Uncovering Latent Memories: Assessing Data Leakage and Memorization Patterns in Frontier AI Models

Searching to Exploit Memorization Effect in Deep Learning with Noisy Labels.

Mitigating Memorization In Language Models

Memorization and Generalization in Neural Code Intelligence Models

Measuring Catastrophic Forgetting in Neural Networks

Measuring Forgetting of Memorized Training Examples

Memorization in deep learning: A survey

Early-Learning Regularization Prevents Memorization of Noisy Labels

Confidence Adaptive Regularization for Deep Learning with Noisy Labels

Measuring memorization through probabilistic discoverable extraction

Exploring Memorization in Adversarial Training

MemHunter: Automated and Verifiable Memorization Detection at Dataset-scale in LLMs

Embedding Space Selection for Detecting Memorization and Fingerprinting in Generative Models

MILD: Modeling the Instance Learning Dynamics for Learning with Noisy Labels

Scalability of memorization-based machine unlearning

Overcoming Catastrophic Forgetting with Unlabeled Data in the Wild

Multi-Objective Interpolation Training for Robustness to Label Noise