From architecture search to architecture generation
Neural Architecture Search (NAS) treats network design as an optimization problem: try candidates, train them, keep the best, and start over for the next task. Neural Architecture Generation (NAG) instead trains a generative model that has learned what good architectures look like, and samples new ones on demand. The sections below explain the difference, with small experiments that run in your browser. Open any section to read it.
Why automate architecture design?
Structure matters, and it is expensive to get right
Many of deep learning's landmark results came from a new architecture rather than more data or a new optimizer.
Deep convolutional networks won ImageNet (Krizhevsky et al., 2017), residual connections made very deep networks trainable (He et al., 2016), and attention replaced recurrence in sequence models (Vaswani et al., 2017). Each of these was designed by hand, by choosing layer types, their order, how they connect, and many hyperparameters.
The design space these choices define is enormous and highly structured, and exploring it by trial and error is slow (Bergstra et al., 2013). As models grew, manual design became a bottleneck for scale, reproducibility and reuse across tasks. The natural next step was to let an algorithm do the exploring.
That algorithm has taken two broad forms. The first, search, looks for the single best architecture for one task. The second, generation, learns a model of good architectures that can be reused.
Search: the NAS view
Architecture design as optimization
Neural Architecture Search frames design as a discrete optimization problem over a space of candidates that a human defined in advance (Elsken et al., 2019).
Three ingredients
A search space: which architectures are allowed. To keep it manageable, many methods search only for a small repeated motif, a cell, and stack copies of it into a full network (Zoph et al., 2018).
A performance estimate: how good a candidate is, usually its validation accuracy after training, or a cheaper proxy for it.
Written out, NAS solves a bi-level problem. The outer level looks for the architecture \(A\) in the search space \(\mathcal{S}\) with the highest reward \(R\). The inner level trains the weights \(w\) of each candidate:
Every evaluation of \(R\) hides a full training run, which is why early NAS cost thousands of GPU-days (Real et al., 2017). Weight sharing (Pham et al., 2018), low-fidelity training (Li et al., 2018) and learning-curve extrapolation (Domhan et al., 2015) made each evaluation cheaper, but evaluation still plays the same role: an external score that decides which candidates survive.
Try it: the toy search space used on this page
Every demo on this page uses the same small cell, in the style of NAS-Bench-201 (Dong & Yang, 2020). It has four nodes, and each of its six edges carries one of five operations: none, skip connection, 1×1 convolution, 3×3 convolution or average pooling. That makes 56 = 15,625 cells, few enough for the browser to score every one of them. That is how the demos can show exact answers.
The cell, left to right from input to output. Each edge is labeled with its operation. A dashed line means "none", and a cell with no complete path from input to output is degenerate.
Accuracy of all 15,625 cells on the selected task. The marker shows the cell on the left.
The accuracies are a hand-designed synthetic function, not NAS-Bench-201 data. It rewards convolutions, depth with a parallel skip connection, and extra capacity on harder tasks, and it adds a small deterministic "noise". This page uses it to illustrate ideas, not to rank methods.
The limits of search
As the field matured, the costs of this framing became clearer (White et al., 2023):
The search space decides what can be found. A space designed by humans biases discovery toward variations of known designs. Inside well-designed spaces, random search is a surprisingly strong baseline (Li & Talwalkar, 2020)(Yu et al., 2019), a sign that much of the result comes from the space rather than the strategy.
Every new task starts from scratch. A search produces one architecture for one dataset and one budget. The knowledge gathered along the way is thrown away.
Evaluation stays outside. The score filters candidates but never becomes part of a reusable model of what makes architectures good.
These limits have led some authors to ask whether NAS is full automation or optimization inside human-imposed boundaries (Baymurzina et al., 2022).
From search to generation
Learning a distribution instead of finding a point
Neural Architecture Generation treats architectures as samples from a learned probability distribution \(p_\theta(A)\), and trains a generative model to put most of its probability on good designs.
Instead of an \(\arg\max\), the target becomes a distribution that favors high reward. A convenient choice weighs every architecture exponentially by its reward (Soin et al., 2025):
\[ p^*(A) \propto \exp\big(\beta\, R(A)\big). \]
The temperature parameter \(\beta \ge 0\) controls how picky the distribution is. With \(\beta = 0\) every architecture is equally likely. As \(\beta \to \infty\), all the mass collapses onto the single best architecture, and sampling from \(p^*\) gives back exactly the NAS answer \(A^*\). In this sense, NAS is the zero-temperature limit of NAG.
A generator is then trained to approximate this target, \(\min_\theta D\big(p_\theta \,\|\, p^*\big)\) for some divergence \(D\). Once trained, it can be conditioned on a description of the dataset \(D\) or on constraints \(C\) such as a latency budget, and draw architectures from \(p_\theta(A \mid D, C)\) (Lee et al., 2021)(Lomurno et al., 2024).
Try it: slide the temperature from "anything goes" to "only the best"
This demo computes the exact \(p^*(A)\) over all 15,625 cells as you move the slider.
Probability mass of \(p^*\) in each accuracy bin (0.5-point bins). The dashed marker is the best cell, \(A^*\).
Summary of \(p^*\) at this temperature
Expected accuracy
Probability of drawing the best cell
Effective number of architectures (exp of the entropy)
What to look for
At β = 0 the histogram simply shows how accuracies are spread over the space: most cells are mediocre.
As β grows, mass moves to the right and the effective number of architectures falls from thousands to a handful. It is still a distribution, though, so several different good cells keep a share.
At the far right the distribution is essentially a point mass on \(A^*\). That is NAS.
Keeping a distribution rather than a point is not only more general. It makes diversity, uncertainty and trade-offs visible and controllable, and it is what allows the learned model to be reused, as the next section shows.
Amortization: pay once, sample many times
NAS loop versus a trained generator
A search pays its full cost on every new task. A generator pays a large cost once, during training, and afterwards produces architectures for new tasks by inference alone.
This is amortized inference. The expensive exploration of architecture–performance pairs is done once, across many training tasks, and stored in the weights of a shared model \(p_\theta(A \mid D)\). Given the descriptor of a new dataset, the model proposes architectures in a forward pass (Lee et al., 2021)(Hemmi et al., 2024). The same idea underlies "train once, specialize many times" approaches to deployment (Cai et al., 2020).
Try it: a new task, searched versus generated
The toy tasks are described by two numbers, difficulty and input resolution, and the best architecture changes with them. When this section opens, a small conditional generator is trained on 16 training tasks, using the evaluations of one 300-step evolutionary search per task, weighted as in \(p^* \propto \exp(\beta R)\). Then pick any new task and compare.
The generator trains when this section opens.
NAS: best accuracy found so far
NAG: best of 10 samples
Best cell (known here by enumeration)
NAS: 300 evaluations per task
NAG: 4,800 to train once, then 10 per task
Best cell found by NAS.Best of the generator's samples.
What to look for
After 300 evaluations, the search usually finds the very best cell. The generator's samples come close, typically within a point of accuracy, without any search. Only its 10 samples are evaluated, to pick the best of them.
The cost chart is the honest part. Training the generator took 4,800 evaluations, so it only pays off after about 17 new tasks. Amortization helps when design problems repeat, not when there is only one.
Try tasks far from the training grid (difficulty or resolution near 0 or 1). A generator can only interpolate what its training tasks covered.
Three pillars of a generator
Representation, mechanism, guidance
A NAG method can be read along three interdependent axes: how architectures are represented, which mechanism produces them, and what guides the generator toward good ones.
Representation
Architectures are graphs, usually directed acyclic graphs of operations. A generator needs a space in which to model them. The options include:
Discrete encodings, such as adjacency matrices with operation labels. They are exact, but hard to interpolate.
Continuous latent embeddings learned by graph autoencoders (Zhang et al., 2019)(Li et al., 2020)(Xiang, 2023). They are smooth and support gradients, but decoding a latent point may give an invalid or duplicate graph.
Learned discrete tokens, such as vector-quantized codes that a language model can generate as a sequence (Poddenige et al., 2025).
Probabilistic structure: distributions over the components themselves, such as the probability of each operation on each edge (Muravev et al., 2021). The generator in the amortization demo is of this kind.
Unlike a search space, which only lists what is allowed, a representation space models how architectures relate to each other. Whether such representations help even without supervision is a question of its own (Yan et al., 2020).
Mechanism
Almost every family of generative models has been applied to architectures. The four mini-demos below illustrate the main ideas on the toy space. Each is an illustration of the principle, not a reimplementation of the cited methods.
Latent optimization: encode, climb, decode
Methods such as NAO (Luo et al., 2019) embed architectures in a continuous space, train a predictor of accuracy on it, climb the predictor's gradient, and decode the result. Here, the 2-D map's horizontal axis was fitted to predict accuracy and its vertical axis to predict size, from 120 evaluated cells. Each pixel is colored by the true accuracy of the cell it decodes to.
Click to choose a starting point. Horizontal axis: predicted accuracy. Vertical axis: size (larger cells toward the top). Light = low true accuracy, dark = high. Gray pixels decode to a degenerate cell. The ring marks the true best cell.
What to look for: many pixels decode to the same cell, and whole regions decode to degenerate cells. That is the mismatch between a continuous space and discrete graphs. The climb also stops where the smooth predictor peaks, which is not where the true best cell is.
Diffusion: denoise a random cell into a good one
Diffusion models learn to reverse a gradual noising process (Ho et al., 2020). Applied to architecture graphs, a predictor can steer the denoising toward high accuracy (An et al., 2024). Here, the training data is an archive of 300 good cells, noise replaces an edge's operation at random, and a linear predictor fitted on the archive provides the guidance.
Press "Denoise one" to watch a cell form over 12 steps.
Gray bars: accuracy of the 300 archive cells. After "Generate 300", orange bars: the generated cells. The markers show the archive's best cell and the true best cell.
What to look for: the exact denoiser is the best possible fit to its training data, yet it can only reproduce archive cells (novelty 0) (Carlini et al., 2023). The per-edge denoiser recombines edges into new cells, and with guidance it can reach cells better than anything in the archive. Stronger guidance raises accuracy and lowers uniqueness.
GFlowNets: sample in proportion to reward
A Generative Flow Network builds an object step by step and is trained so that complete objects are sampled with probability proportional to their reward (Bengio et al., 2026), which has been applied to architecture generation (Soin et al., 2025). Here, the same policy builds a cell one edge at a time, trained with two objectives: reinforcement learning, which maximizes expected reward, and a GFlowNet objective (trajectory balance) with reward \(R = \exp(\beta\,\text{accuracy})\).
Target: proportional to reward
GFlowNet
Reinforcement learning
Press "Train both". Training runs 200,000 construction episodes per policy.
After training
GFlowNet
Reinforcement learning
Expected accuracy
–
–
Effective number of architectures
–
–
Distinct top-1% cells in 1,000 samples
–
–
What to look for: reinforcement learning converges on a single cell, a point estimate, just like search. The GFlowNet keeps probability on many good cells, roughly in proportion to their reward, so its samples stay diverse.
Budget-conditioned generation: one model, every budget
Deployment adds constraints such as size, latency or energy. Instead of one search per budget, Pareto-aware generators learn \(p_\theta(A \mid \lambda)\) for a whole range of budgets \(\lambda\) (Guo et al., 2022)(Lomurno et al., 2024). Here, a generator was trained on ten budgets. Move the slider to ask for any budget, including ones it never saw.
Every valid cell (gray), the exact Pareto front (line), the budget (dashed) and 50 samples from the generator (blue if within budget, orange if over).
What to look for: the samples follow the budget as it moves, with no retraining and no search. Not every sample fits, because this small generator picks each edge independently, but a cell's size is free to compute, so over-budget samples are simply discarded before any training.
What pulls a generator toward good architectures? The studies fall into two groups:
Predictor-guided: a surrogate model estimates accuracy and provides gradients or scores that steer generation, as in the latent and diffusion demos above. DiffusionNAG is one example (An et al., 2024).
Predictor-free: guidance comes from conditioning built into training, structural priors, or true evaluations turned into rewards, as in the GFlowNet demo. Examples include classifier-free guidance in graph diffusion (Asthana et al., 2024) and GFlowNet-based generation (Soin et al., 2025).
Either way, the defining move from NAS to NAG is that the performance signal is internalized into the generator's parameters instead of acting as an external filter after the fact.
Some systems proposed and critiqued by large language models (Rahman & Chakraborty, 2024)(Yang et al., 2025) escape fixed search spaces, but still rely on iterative evaluate-and-refine loops, so they behave more like search than like a learned generative model.
Evaluating a generator
Judging a distribution, not a single winner
NAS results are usually summarized by the best architecture found and its cost. A generator produces a distribution, and a single number cannot describe a distribution.
Metrics borrowed from generative modeling
Validity, uniqueness, novelty. Are samples well-formed graphs? Are they different from each other? Are they different from the training data (Zhang et al., 2019)(Poddenige et al., 2025)? The diffusion demo reports all three, and shows how a model can have perfect validity and zero novelty.
Divergences. The forward KL penalizes a generator that misses good regions. The reverse KL penalizes one that puts mass on bad regions. They are complementary, and ranking models with only one metric can mislead (Theis et al., 2016)(Betzalel et al., 2022).
Reward alignment and query efficiency. How well sampling frequencies track true performance (Soin et al., 2025), and how many expensive evaluations training needed (Lukasik et al., 2022).
The benchmark paradox
Tabular benchmarks such as NAS-Bench-101 (Ying et al., 2019) and NAS-Bench-201 (Dong & Yang, 2020) store the trained accuracy of every architecture in a fixed space, so any method can be scored by lookup. That gives reproducibility and fair comparison, but only if the generator is confined to that fixed space, which removes the thing NAG is supposed to be good at: generating beyond it.
The alternatives have their own problems. Training every generated model makes results depend on training recipes (Yang et al., 2020). Scoring with a surrogate predictor imports the predictor's biases, and a predictor trained on familiar designs tends to undervalue novel ones, the same flaw found in feature-based metrics for images (Stein et al., 2023).
Open problems
Where the field is heading
NAG is young, and several challenges still limit its reliability and adoption.
Validity by construction
Continuous latent spaces and stochastic denoising can produce graphs that are not valid networks, so samples often need repair or rejection. Building constraints such as acyclicity and connectivity into the generation process itself is an open direction. Discrete learned tokens are one early attempt (Poddenige et al., 2025).
Beyond closed search spaces
Most generators still operate over the fixed operation sets of benchmark spaces, so gains may reflect better interpolation within a table rather than new design principles. Evaluation protocols built for generative models, measuring diversity, transfer and controllability, are still missing.
Trustworthy guidance
Architecture–performance landscapes are large and rugged, and a single surrogate predictor rarely generalizes across them, so generators can end up optimizing the predictor's errors. Progressive guidance from ensembles of weaker predictors is one response (Zhang & Zhuang, 2025).
Any budget, instantly
Ideally, one generator would map any hardware budget to a near-optimal architecture and share structure across budgets, as in the budget demo, instead of treating each budget as a separate problem (Guo et al., 2022)(Lomurno et al., 2024).
Zero-shot architectural priors
Most generators need thousands of evaluated architectures before they are useful. Priors extracted from existing benchmarks (Xie et al., 2025) and training-light evolutionary dynamics (Zhou et al., 2025) reduce this cold start. The long-term goal is a foundation distribution over architectures that works across tasks.
Explaining generated designs
Generated architectures rarely come with a reason. Agent-style critique loops that justify each design choice (Yang et al., 2025) hint at generators whose decisions can be inspected, which matters for trust and for learning something from what they produce.
About the study
The paper behind this page
This page is based on From Search to Synthesis: A Systematic Mapping Study of Neural Architecture Generation, by Thiago C. S. Barbieri, Edesio Alcobaça and André C. P. L. F. de Carvalho (Institute of Mathematical and Computer Sciences, University of São Paulo). The manuscript is currently under review, and a link will be added here once it is published.
The page draws on the study's framing and bibliography. The demos and all their numbers were made for this page.
References
Every work cited on this page
Agiollo, A., Omicini, A. (2022). GNN2GNN: Graph neural networks to generate neural networks. Proceedings of the Thirty-Eighth Conference on Uncertainty in Artificial Intelligence 180, 32–42. linkCited in: Three pillars of a generator
Bengio, Y., Lahlou, S., Deleu, T., Hu, E. J., Tiwari, M., Bengio, E. (2026). GFlowNet Foundations. arXiv preprint. linkCited in: Three pillars of a generator
Bergstra, J., Yamins, D., Cox, D. (2013). Making a Science of Model Search: Hyperparameter Optimization in Hundreds of Dimensions for Vision Architectures. Proceedings of the 30th International Conference on Machine Learning 28, 115–123. linkCited in: Why automate architecture design?
Betzalel, E., Penso, C., Navon, A., Fetaya, E. (2022). A Study on the Evaluation of Generative Models. arXiv preprint. linkCited in: Evaluating a generator
Cai, H., Gan, C., Wang, T., Zhang, Z., Han, S. (2020). Once-for-All: Train One Network and Specialize it for Efficient Deployment. arXiv preprint. linkCited in: Amortization: pay once, sample many times
Carlini, N., Hayes, J., Nasr, M., Jagielski, M., Sehwag, V., Tramèr, F., Balle, B., Ippolito, D., Wallace, E. (2023). Extracting Training Data from Diffusion Models. arXiv preprint. linkCited in: Three pillars of a generator
Domhan, T., Springenberg, J. T., Hutter, F. (2015). Speeding up automatic hyperparameter optimization of deep neural networks by extrapolation of learning curves. Proceedings of the 24th International Conference on Artificial Intelligence , 3460–3468. linkCited in: Search: the NAS view
Elsken, T., Metzen, J. H., Hutter, F. (2019). Neural architecture search: a survey. J. Mach. Learn. Res. 20(1), 1997–2017.
Cited in: Search: the NAS view
Guo, Y., Chen, Y., Zheng, Y., Chen, Q., Zhao, P., Chen, J., Huang, J., Tan, M. (2022). Pareto-aware Neural Architecture Generation for Diverse Computational Budgets. arXiv preprint. linkCited in: Three pillars of a generator, Open problems
He, K., Zhang, X., Ren, S., Sun, J. (2016). Deep Residual Learning for Image Recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) , 770–778. doi:10.1109/CVPR.2016.90Cited in: Why automate architecture design?
Hemmi, K., Tanigaki, Y., Onishi, M. (2024). NAVIGATOR-D3: Neural Architecture Search Using Variational Graph Auto-encoder Toward Optimal Architecture Design for Diverse Datasets. Artificial Neural Networks and Machine Learning – ICANN 2024 , 292–307.
Cited in: Amortization: pay once, sample many times
Ho, J., Jain, A., Abbeel, P. (2020). Denoising Diffusion Probabilistic Models. arXiv preprint. linkCited in: Three pillars of a generator
Li, J., Liu, Y., Liu, J., Wang, W. (2020). Neural Architecture Optimization with Graph VAE. arXiv preprint. linkCited in: Three pillars of a generator
Li, L., Jamieson, K., DeSalvo, G., Rostamizadeh, A., Talwalkar, A. (2018). Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimization. arXiv preprint. linkCited in: Search: the NAS view
Li, L., Talwalkar, A. (2020). Random Search and Reproducibility for Neural Architecture Search. Proceedings of The 35th Uncertainty in Artificial Intelligence Conference 115, 367–377. linkCited in: Search: the NAS view
Lukasik, J., Jung, S., Keuper, M. (2022). Learning Where to Look – Generative NAS is Surprisingly Efficient. Computer Vision – ECCV 2022 , 257–273.
Cited in: Evaluating a generator
Naeem, M. F., Oh, S. J., Uh, Y., Choi, Y., Yoo, J. (2020). Reliable fidelity and diversity metrics for generative models. Proceedings of the 37th International Conference on Machine Learning (ICML).
Cited in: Evaluating a generator
Pham, H., Guan, M., Zoph, B., Le, Q., Dean, J. (2018). Efficient Neural Architecture Search via Parameters Sharing. Proceedings of the 35th International Conference on Machine Learning 80, 4095–4104. linkCited in: Search: the NAS view
Rahman, M. H., Chakraborty, P. (2024). LeMo-NADe: Multi-Parameter Neural Architecture Discovery with LLMs. arXiv preprint. linkCited in: Three pillars of a generator
Real, E., Moore, S., Selle, A., Saxena, S., Suematsu, Y. L., Tan, J., Le, Q., Kurakin, A. (2017). Large-Scale Evolution of Image Classifiers. arXiv preprint. linkCited in: Search: the NAS view
Real, E., Aggarwal, A., Huang, Y., Le, Q. V. (2019). Regularized evolution for image classifier architecture search. Proceedings of the AAAI Conference on Artificial Intelligence. doi:10.1609/aaai.v33i01.33014780Cited in: Search: the NAS view
Rezaei, S. S. C., Han, F. X., Niu, D., Salameh, M., Mills, K., Lian, S., Lu, W., Jui, S. (2021). Generative Adversarial Neural Architecture Search. arXiv preprint. doi:10.48550/arXiv.2105.09356Cited in: Three pillars of a generator
Sajjadi, M. S. M., Bachem, O., Lucic, M., Bousquet, O., Gelly, S. (2018). Assessing Generative Models via Precision and Recall. arXiv preprint. linkCited in: Evaluating a generator
Stein, G., Cresswell, J. C., Hosseinzadeh, R., Sui, Y., Ross, B. L., Villecroze, V., Liu, Z., Caterini, A. L., Taylor, J. E. T., Loaiza-Ganem, G. (2023). Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion models. arXiv preprint. linkCited in: Evaluating a generator
Theis, L., van den Oord, A., Bethge, M. (2016). A note on the evaluation of generative models. arXiv preprint. linkCited in: Evaluating a generator
Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, Ł., Polosukhin, I. (2017). Attention is all you need. Proceedings of the 31st International Conference on Neural Information Processing Systems , 6000–6010.
Cited in: Why automate architecture design?
White, C., Neiswanger, W., Savani, Y. (2020). BANANAS: Bayesian Optimization with Neural Architectures for Neural Architecture Search. arXiv preprint. linkCited in: Search: the NAS view
White, C., Safari, M., Sukthanker, R., Ru, B., Elsken, T., Zela, A., Dey, D., Hutter, F. (2023). Neural Architecture Search: Insights from 1000 Papers. arXiv preprint. linkCited in: Search: the NAS view
Xiang, L. (2023). Efficient Automated Neural Architecture Design. PhD thesis, University of Warwick. linkCited in: Three pillars of a generator
Xie, J., Ji, H., Sun, Y. (2025). Prior Knowledge Guided Neural Architecture Generation. Proceedings of the 42nd International Conference on Machine Learning 267, 68671–68686. linkCited in: Open problems
Yan, S., Zheng, Y., Ao, W., Zeng, X., Zhang, M. (2020). Does Unsupervised Architecture Representation Learning Help Neural Architecture Search?. arXiv preprint. linkCited in: Three pillars of a generator
Yang, A., Esperança, P. M., Carlucci, F. M. (2020). NAS evaluation is frustratingly hard. arXiv preprint. linkCited in: Evaluating a generator
Yang, Z., Zeng, W., Jin, S., Qian, C., Luo, P., Liu, W. (2025). NADER: Neural Architecture Design via Multi-Agent Collaboration. 2025 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) , 4452–4461. doi:10.1109/CVPR52734.2025.00420Cited in: Three pillars of a generator, Open problems
Ying, C., Klein, A., Real, E., Christiansen, E., Murphy, K., Hutter, F. (2019). NAS-Bench-101: Towards Reproducible Neural Architecture Search. arXiv preprint. linkCited in: Evaluating a generator
Yu, K., Sciuto, C., Jaggi, M., Musat, C., Salzmann, M. (2019). Evaluating the Search Phase of Neural Architecture Search. arXiv preprint. linkCited in: Search: the NAS view
Zhang, M., Jiang, S., Cui, Z., Garnett, R., Chen, Y. (2019). D-VAE: a variational autoencoder for directed acyclic graphs. Proceedings of the 33rd International Conference on Neural Information Processing Systems , 1588–1600.
Cited in: Three pillars of a generator, Evaluating a generator
Zoph, B., Le, Q. V. (2017). Neural Architecture Search with Reinforcement Learning. arXiv preprint. linkCited in: Search: the NAS view
Zoph, B., Vasudevan, V., Shlens, J., Le, Q. V. (2018). Learning Transferable Architectures for Scalable Image Recognition. arXiv preprint. linkCited in: Search: the NAS view