Generative Modeling of Weights: Generalization or Memorization?
Not on the map. Why?
Abstract (arXiv)
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 four representative, well-known methods on their ability to generate novel model weights, i.e., weights that are different from the checkpoints seen during training. Contrary to claims in prior work, we find that these methods synthesize weights largely by memorization: they produce either replicas, or, at best, simple interpolations of the training checkpoints. Moreover, they 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. Our further analysis suggests that this memorization might result from limited data, overparameterized models, and the underuse of structural priors specific to weight data. These findings highlight the need for more careful design and rigorous evaluation of generative models when applied to new domains. Our code is available at https://github.com/boyazeng/weight_memorization.
Key facts. Cited by 14 papers in the map (data through 14 Oct 2026). Builds on 61 of its 90 references that are in the map. First posted on arXiv 9 Jun 2025.
Builds on
61 of 90 references.
References · 61
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
- 2014Going Deeper with Convolutionscited by 4,241
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
38 more
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,897
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
- 2016Perceptual Losses for Real-Time Style Transfer and Super-Resolutioncited by 2,694
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,269
- 2022Imagen Video: High Definition Video Generation with Diffusion Modelscited by 1,778
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2016HyperNetworkscited by 1,027
- 2025Cosmos World Foundation Model Platform for Physical AIcited by 864
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 464
- 2022Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Modelscited by 424
- 2017The Expressive Power of Neural Networks: A View from the Widthcited by 320
- 2021The Intrinsic Dimension of Images and Its Impact on Learningcited by 315
- 2021Skillful Precipitation Nowcasting using Deep Generative Models of Radarcited by 311
- 2016On the Expressive Power of Deep Neural Networkscited by 306
- 2019Intrinsic dimension of data representations in deep neural networkscited by 282
- 2023Understanding and Mitigating Copying in Diffusion Modelscited by 209
- 2017SMASH: One-Shot Model Architecture Search through HyperNetworkscited by 202
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representationscited by 185
- 2022A Self-Supervised Descriptor for Image Copy Detectioncited by 176
- 2023Score Approximation, Estimation and Distribution Recovery of Diffusion Models on Low-Dimensional Datacited by 174
- 2023Diffusion Models are Minimax Optimal Distribution Estimatorscited by 165
- 2023Holistic Evaluation of Text-To-Image Modelscited by 160
- 2023HyperDiffusion: Generating Implicit Neural Fields with Weight-Space Diffusioncited by 152
- 2018Graph HyperNetworks for Neural Architecture Searchcited by 132
- 2023State of the Art on Diffusion Models for Visual Computingcited by 127
- 2023On Memorization in Diffusion Modelscited by 125
- 2024Measuring Style Similarity in Diffusion Modelscited by 109
- 2023Compositional Abilities Emerge Multiplicatively: Exploring Diffusion Models on a Synthetic Taskcited by 93
- 2021Parameter Prediction for Unseen Deep Architecturescited by 89
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 86
- 2022Hyper-Representations as Generative Models: Sampling Unseen Neural Network Weightscited by 64
- 2024Graph Neural Networks for Learning Equivariant Representations of Neural Networkscited by 59
- 2023Graph Metanetworks for Processing Diverse Neural Architecturescited by 57
- 2020A Non-Parametric Test to Detect Data-Copying in Generative Modelscited by 52
50 most cited of 61 All 61 (JSON)
Built on by
Citing papers per year: 2025: 3; 2026: 11.
14 papers.
Sort:
By area:
- 2026Learning a Generative Meta-Model of LLM Activationscited by 8
- 2026Position: Weight Space Should Be a First-Class Generative AI Modalitycited by 6
- 2026DeepWeightFlow: Re-Basined Flow Matching for Generating Neural Network Weightscited by 5
- 2026You Only Need Minimal RLVR Training: Extrapolating LLMs via Rank-1 Trajectoriescited by 3
- 2026What Linear Probes Miss: Multi-View Probing for Weight-Space Learningcited by 2
- 2026Scaling Linear Mode Connectivity and Merging to Billion Parameter Pretrained Transformerscited by 2
- 2025Flows and Diffusions on the Neural Manifoldcited by 1
- 2025Memorization in 3D Shape Generation: An Empirical Studycited by 1
- 2025Delta Activations: A Representation for Finetuned Large Language Modelscited by 0
- 2026Infinite-Parameter LLMs: Generating and Adapting Weights from Live Datacited by 0
- 2026Permutation-Equivariant Flow Matching for Alignment-Free Neural Weight Generationcited by 0
- 2026On-the-fly Weight Generation: A Hypernetwork Proof of Concept on ARC-1Dcited by 0
- 2026Modeling and Understanding Neural Weights of Humanoid Policies with Diffusion14 Oct 2026
- 2026WIRE: Weight Space Implicit Representation Encoding of Frozen Video LLMs12 Oct 2026
- 2026On-the-fly Weight Generation: A Hypernetwork Proof of Concept on ARC-1D30 Sep 2026
- 2026Permutation-Equivariant Flow Matching for Alignment-Free Neural Weight Generation26 Sep 2026
- 2026Infinite-Parameter LLMs: Generating and Adapting Weights from Live Data16 Sep 2026
- 2026Scaling Linear Mode Connectivity and Merging to Billion Parameter Pretrained Transformers22 Jun 2026
- 2026What Linear Probes Miss: Multi-View Probing for Weight-Space Learning22 May 2026
- 2026You Only Need Minimal RLVR Training: Extrapolating LLMs via Rank-1 Trajectories20 May 2026
- 2026Position: Weight Space Should Be a First-Class Generative AI Modality18 May 2026
- 2026Learning a Generative Meta-Model of LLM Activations6 Feb 2026
- 2026DeepWeightFlow: Re-Basined Flow Matching for Generating Neural Network Weights8 Jan 2026
- 2025Memorization in 3D Shape Generation: An Empirical Study29 Dec 2025
2 more
- 2025Delta Activations: A Representation for Finetuned Large Language Models4 Sep 2025
- 2025Flows and Diffusions on the Neural Manifold14 Jul 2025
- 2026Position: Weight Space Should Be a First-Class Generative AI Modalitycited by 6
- 2026Scaling Linear Mode Connectivity and Merging to Billion Parameter Pretrained Transformerscited by 2
- 2025Flows and Diffusions on the Neural Manifoldcited by 1
- 2025Memorization in 3D Shape Generation: An Empirical Studycited by 1
- 2025Delta Activations: A Representation for Finetuned Large Language Modelscited by 0
- 2026On-the-fly Weight Generation: A Hypernetwork Proof of Concept on ARC-1Dcited by 0
- 2026On-the-fly Weight Generation: A Hypernetwork Proof of Concept on ARC-1D30 Sep 2026
- 2026Scaling Linear Mode Connectivity and Merging to Billion Parameter Pretrained Transformers22 Jun 2026
- 2026Position: Weight Space Should Be a First-Class Generative AI Modality18 May 2026
- 2025Memorization in 3D Shape Generation: An Empirical Study29 Dec 2025
- 2025Delta Activations: A Representation for Finetuned Large Language Models4 Sep 2025
- 2025Flows and Diffusions on the Neural Manifold14 Jul 2025