mlsanityData to 7 Oct 2026

Position: Weight Space Should Be a First-Class Generative AI Modality

arXiv 2605.18632 ↗First posted 18 May 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

Neural network checkpoints have quietly become a large-scale data resource: millions of trained weight vectors now exist, each encoding task-, domain-, and architecture-specific knowledge. This position paper argues that model checkpoints should be treated as a first-class data modality, and that generative modeling in weight space should be standardized as a core machine learning primitive. Recent advances demonstrate that neural weights can be synthesized on demand, often matching fine-tuning performance while reducing adaptation cost by orders of magnitude. We contend that these results reflect an underlying structural fact: high-performing models occupy low-dimensional, highly structured regions of weight space shaped by symmetry, flatness, modularity, and shared subspaces. Building on this view, we organize existing methods into a five-stage pipeline, survey applications where the approach is already practical, and clarify current limits: adapter-scale and conditional generation are advancing rapidly, while unrestricted frontier-scale checkpoint synthesis remains open. Our goal is to shift the community's default mindset from optimizing models per task to sampling models from learned weight distributions, accelerating toward an era in which AI systems routinely improve or create other AI systems.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 78 of its 82 references that are in the map. First posted on arXiv 18 May 2026.

Builds on

78 of 82 references.

References · 78

  1. 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
  2. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  3. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  4. 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
  5. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  6. 2016HyperNetworkscited by 1,003
  7. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
  8. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
  9. 2021Carbon Emissions and Large Neural Network Trainingcited by 561
  10. 2019Fantastic Generalization Measures and Where to Find Themcited by 507
  11. 2023DecodingTrust: A Comprehensive Assessment of Trustworthiness in GPT Modelscited by 505
  12. 2017Sharp Minima Can Generalize For Deep Netscited by 462
38 more
  1. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
  2. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  3. 2023Symbolic Discovery of Optimization Algorithmscited by 423
  4. 2019Bayesian Nonparametric Federated Learning of Neural Networkscited by 385
  5. 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
  6. 2024The Platonic Representation Hypothesiscited by 362
  7. 2020Discovering Symbolic Models from Deep Learning with Inductive Biasescited by 335
  8. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
  9. 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 308
  10. 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 289
  11. 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
  12. 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 283
  13. 2014The Loss Surfaces of Multilayer Networkscited by 278
  14. 2020The Lottery Ticket Hypothesis for Pre-trained BERT Networkscited by 275
  15. 2020Understanding the Role of Individual Units in a Deep Neural Networkcited by 254
  16. 2015Net2Net: Accelerating Learning via Knowledge Transfercited by 245
  17. 2021Personalized Federated Learning using Hypernetworkscited by 210
  18. 2018Gradient Descent Happens in a Tiny Subspacecited by 208
  19. 2021Revisiting Model Stitching to Compare Neural Representationscited by 203
  20. 2017SMASH: One-Shot Model Architecture Search through HyperNetworkscited by 200
  21. 2014Qualitatively characterizing neural network optimization problemscited by 186
  22. 2018Gradient descent aligns the layers of deep linear networkscited by 182
  23. 2023HyperDiffusion: Generating Implicit Neural Fields with Weight-Space Diffusioncited by 152
  24. 2019A Function Space View of Bounded Norm Infinite Width ReLU Nets: The Multivariate Casecited by 113
  25. 2020Predicting Neural Network Accuracy from Weightscited by 107
  26. 2020Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Maskscited by 97
  27. 2021Parameter Prediction for Unseen Deep Architecturescited by 89
  28. 2020Relative Flatness and Generalizationcited by 88
  29. 2020Optimizing Mode Connectivity via Neuron Alignmentcited by 85
  30. 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
  31. 2023Learning to Grow Pretrained Models for Efficient Transformer Trainingcited by 81
  32. 2022Deep Model Reassemblycited by 78
  33. 2019Explaining Landscape Connectivity of Low-cost Solutions for Multilayer Netscited by 63
  34. 2022Hyper-Representations as Generative Models: Sampling Unseen Neural Network Weightscited by 63
  35. 2023Graph Metanetworks for Processing Diverse Neural Architecturescited by 55
  36. 2024APOLLO: SGD-like Memory, AdamW-level Performancecited by 50
  37. 2025Text-to-LoRA: Instant Transformer Adaptioncited by 50
  38. 2024Towards Scalable and Versatile Weight Space Learningcited by 40

50 most cited of 78 All 78 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.