paper-with-me

홈 › Papers

Generative Modeling of Weights: Generalization or Memorization?

2025-06-09 · Boya Zeng, Yida Yin, Zhiqiu Xu, Zhuang Liu

Generative models, with their success in image and video generation, have recently been explored for synthesizing effective neural network weights. These approaches take trained neural network checkpoints as training data, and aim to generate high-performing neural network weights during inference. In this work, we examine four representative methods on their ability to generate novel model weights, i.e., weights that are different from the checkpoints seen during training. Surprisingly, we find that these methods synthesize weights largely by memorization: they produce either replicas, or at best simple interpolations, of the training checkpoints. Current methods fail to outperform simple baselines, such as adding noise to the weights or taking a simple weight ensemble, in obtaining different and simultaneously high-performing models. We further show that this memorization cannot be effectively mitigated by modifying modeling factors commonly associated with memorization in image diffusion models, or applying data augmentations. Our findings provide a realistic assessment of what types of data current generative models can model, and highlight the need for more careful evaluation of generative models in new domains. Our code is available at https://github.com/boyazeng/weight_memorization.

📄 PDF Abstract BibTeX arXiv:2506.07998

Code (1)

boyazeng/weight_memorization 공식 구현 pytorch

Tasks

MemorizationVideo Generation

Methods 이 논문이 사용한 방법론

Diffusion Diffusion models generate samples by gradually removing noise from a signal, and their training objective can be expressed as a reweighted variational lower-bound…

Similar Papers 제목 키워드 기반

Generalization of Diffusion Models Arises with a Balanced Representation Space

2025-12-24 · Zekai Zhang, Xiao Li, Xiang Li, Lianghe Shi 외 arxiv

Diffusion models excel at generating high-quality, diverse samples, yet they risk memorizing training data when overfit to the training objective. We analyze the distinctions between memorization and generalization in di…

Representation Learning

On the Memorization of Consistency Distillation for Diffusion Models

2026-04-26 · Bingqing Jiang, Difan Zou arxiv

Diffusion models are central to modern generative modeling, and understanding how they balance memorization and generalization is critical for reliable deployment. Recent work has shown that memorization in diffusion mod…

On Training Sample Memorization: Lessons from Benchmarking Generative Modeling with a Large-scale Competition

2021-06-06 · Ching-Yuan Bai, Hsuan-Tien Lin, Colin Raffel, Wendy Chih-wen Kan

Many recent developments on generative models for natural images have relied on heuristically-motivated metrics that can be easily gamed by memorizing a small sample from the true distribution or training a model directl…

BenchmarkingMemorization

Generalization-Memorization Machines

2022-07-08 · Zhen Wang, Yuan-Hai Shao

Classifying the training data correctly without over-fitting is one of the goals in machine learning. In this paper, we propose a generalization-memorization mechanism, including a generalization-memorization decision an…

Memorization

Memorization in 3D Shape Generation: An Empirical Study

2025-12-29 · Shu Pu, Boya Zeng, Kaichen Zhou, Mengyu Wang 외 arxiv

Generative models are increasingly used in 3D vision to synthesize novel shapes, yet it remains unclear whether their generation relies on memorizing training shapes. Understanding their memorization could help prevent t…