AI Model Wiki

    Skip to main content
    Back to Glossary

    Neural Architecture Search

    Neural Architecture Search (NAS) is an automated machine-learning technique that algorithmically explores a predefined search space of possible Neural Network designs - including layer types, connectivity patterns, activation functions, and kernel sizes - to discover architectures that optimise a target metric such as validation accuracy, latency, or parameter efficiency. Rather than relying on expert intuition to hand-design every structural choice, NAS treats the architecture itself as a Hyperparameter to be optimised, typically using reinforcement learning, evolutionary algorithms, gradient-based methods (such as DARTS), or Bayesian optimisation.

    Think of it like asking a very patient engineer to try thousands of different blueprint layouts for a building, measuring how each performs under real load conditions, and then handing you the best-performing design - except the engineer works automatically and never sleeps.

    A simplified DARTS-style search updates architecture parameters alongside weight parameters:

    # Conceptual DARTS inner loop
    for step in range(search_steps):
        # Update network weights on training set
        optimizer_weights.zero_grad()
        loss_train = model(x_train, use_arch_params=False)
        loss_train.backward()
        optimizer_weights.step()
    
        # Update architecture parameters on validation set
        optimizer_arch.zero_grad()
        loss_val = model(x_val, use_arch_params=True)
        loss_val.backward()
        optimizer_arch.step()
    

    NAS produced influential architectures such as EfficientNet, NASNet, and MobileNetV3, demonstrating that automated search can outperform manually designed baselines with a fraction of the Inference compute. Related concepts include Gradient Descent (used in differentiable NAS), Overfitting (a risk when the search over-optimises for the proxy task), Knowledge Distillation (often applied after NAS to compress the found architecture), and Quantization (a post-search step to make the discovered model deployment-ready).

    You are most likely to encounter NAS when reading about mobile or edge AI models, hardware-aware model design, or AutoML platforms that promise competitive accuracy within strict latency or memory budgets.

    AI Terminology
    Glossary