Source-linked AI summary

Differentiable Quantum Architecture Search

Shi-Xin Zhang, Chang-Yu Hsieh, Shengyu Zhang, Hong Yao

arXiv:2010.08561v2quant-ph

TL;DR

NISQ quantum-circuit design requires adapting architectures to task, noise, and hardware constraints, while existing QAS methods search discrete spaces. DQAS addresses this by jointly optimizing sampled circuit structures and variational parameters in a differentiable probabilistic framework. Across circuit design applications, it discovers task-specific architectures, including noise-resilient circuits and improved QAOA layouts.

  • Problem

    Fixed ansätze and discrete QAS methods limit automated circuit design under NISQ constraints involving noise, depth, connectivity, and problem structure.

    Source: §1 · p. 1
  • Method

    DQAS samples circuit structures from a probabilistic model and jointly optimizes its continuous parameters with circuit variational parameters through gradient-based differentiation.

    Source: §1 · p. 2, §3 · p. 5
  • Results

    DQAS designs circuits across decomposition, error mitigation, and optimization tasks, including a MAX CUT circuit achieving 12 versus 10.39 for P = 1 vanilla QAOA and 11.18 for P = 2 vanilla QAOA.

    Source: §4 · pp. 7, 9
  • Takeaways & Limitations

    DQAS provides a flexible framework for customized quantum-circuit design incorporating native gates, hardware connectivity, and error models.

    Source: §1 · p. 2, §5 · p. 9

    The presented probabilistic model is independent across categories and does not explicitly model correlations between circuit layers; richer models remain future work.

    Source: §3 · p. 6

Abstract

from arXiv · show

Quantum architecture search (QAS) is the process of automating architecture engineering of quantum circuits. It has been desired to construct a powerful and general QAS platform which can significantly accelerate current efforts to identify quantum advantages of error-prone and depth-limited quantum circuits in the NISQ era. Hereby, we propose a general framework of differentiable quantum architecture search (DQAS), which enables automated designs of quantum circuits in an end-to-end differentiable fashion. We present several examples of circuit design problems to demonstrate the power of DQAS. For instance, unitary operations are decomposed into quantum gates, noisy circuits are re-designed to improve accuracy, and circuit layouts for quantum approximation optimization algorithm are automatically discovered and upgraded for combinatorial optimization problems. These results not only manifest the vast potential of DQAS being an essential tool for the NISQ application developments, but also present an interesting research topic from the theoretical perspective as it draws inspirations from the newly emerging interdisciplinary paradigms of differentiable programming, probabilistic programming, and quantum programming.

1. INTRODUCTION

Quantum architecture search automates quantum-circuit design, addressing the limitations of fixed variational ansätze and NISQ hardware. DQAS introduces a general, end-to-end differentiable framework inspired by differentiable neural architecture search.

  • Fixed variational ansätze optimize parameters without adapting circuit structure, despite differences in expressive power and entangling capability.Source: §1 · p. 1
  • NISQ circuit design must account for problem structure, decoherence, limited qubit connectivity, and resource costs such as two-qubit gates and circuit depth.Source: §1 · p. 1
  • QAS denotes automated parameterized-circuit design, while quantum architecture search covers broader quantum-circuit design beyond variational ansätze.Source: §1 · p. 1
  • Existing QAS approaches include greedy, evolutionary or genetic, and reinforcement-learning methods, paralleling common neural architecture search strategies.Source: §1 · p. 1
  • DQAS relaxes quantum architecture choices into a differentiable domain, enabling a general, fully automated, end-to-end circuit-design workflow.Source: §1 · pp. 1–2

2. BACKGROUND AND RELATED WORK

Prior neural and quantum architecture-search methods largely operate over discrete spaces, motivating differentiable search. The QAOA background supplies a structured target for discovering generalized Hamiltonian layouts beyond the vanilla ansatz.

Differentiable Neural Architecture Search

  • NAS seeks task-specific high-performing neural architectures using reinforcement learning, evolutionary algorithms, or differentiable relaxation.Source: §2 · p. 2
  • DARTS relaxes discrete architecture components into continuous variables optimized by differentiation and gradient descent, reducing training time and hardware requirements.Source: §2 · p. 2

Related works on QAS

  • Earlier QAS studies mainly used evolutionary, greedy, or reinforcement-learning methods for QAOA and VQE-related circuit-structure tasks.Source: §2 · pp. 2–3
  • These QAS methods search discrete quantum architectures, motivating DQAS as an end-to-end differentiable alternative.Source: §2 · p. 3

Brief review on QAOA

  • QAOA alternates phase or cost evolution with mixer evolution, using variational parameters to address combinatorial optimization problems.Source: §2 · p. 3
  • Generalized QAOA separates preparation, mixer, phase/problem, and cost Hamiltonians, creating layouts that DQAS can automatically search beyond vanilla QAOA.Source: §2 · p. 3
  • QAOA uses relatively few variational parameters, potentially easing training compared with same-depth hardware-efficient VQE or quantum neural-network ansätze.Source: §2 · p. 3

3. METHODS

DQAS jointly searches circuit structure and variational parameters by sampling discrete architectures from a differentiable probabilistic model. It supports task-specific objectives, broad circuit-design settings, and scalable parameter sharing.

Overview

  • DQAS treats circuit design as bi-optimization: it selects unitaries for a circuit layout while identifying suitable variational parameters.Source: §3 · p. 4

Circuit encoding and operation pool

  • The framework represents circuits as sequences of unitary placeholders filled from an operation pool, including one-, two-, or higher-level parameterized blocks.Source: §3 · p. 4
  • Each placeholder can reuse an operation-pool unitary, allowing DQAS to assemble circuits with replacement.Source: §3 · p. 4

Objectives

  • Objectives cover hybrid quantum-classical tasks, supervised learning, state preparation, and circuit compilation, including expectation-based losses and reference-circuit fidelity.Source: §3 · pp. 4–5

Sampling the structures

  • DQAS samples discrete structures k from a probabilistic model parameterized by continuous α, then optimizes α and circuit parameters θ through automatic differentiation.Source: §3 · p. 5

Filling the circuit parameters

  • Parameter sharing stores operation parameters in a shared pool, while Monte Carlo architecture sampling reaches an exponential architecture space with moderate resources.Source: §3 · pp. 5–6

Probabilistic models

  • The default independent-category probabilistic model omits explicit inter-layer correlations; richer energy-based or autoregressive models are left for future work.Source: §3 · p. 6

4. APPLICATIONS

DQAS is applied to several NISQ circuit-design tasks, including noise-resilient QFT circuits and QAOA-like MAXCUT ansatz search. It discovers correlated gate insertions, circuit layouts, and reduced ansatz structures automatically.

Applications

  • DQAS is presented as a versatile tool for quantum circuit design, with examples spanning unitary decomposition, quantum error mitigation, and hybrid quantum-classical algorithms.The implementations use quantum simulation backends based on Cirq/TensorFlow Quantum or TensorNetwork/TensorCircuit.Source: §4 · p. 6

Quantum error mitigation

  • DQAS inserts single-qubit gates into idle slots to maximize noisy-circuit output fidelity under a specified error model.For QFT-3, six placeholders are searched; the model assumes 2% bit-flip errors between gates and about 20% during long idle periods.Source: §4 · p. 7
  • With an expanded gate pool, DQAS places correlated T gates in separated QFT-3 gaps, improving fidelity from 0.33 for the basic circuit to 0.6 for the QEM circuit.The figure reports these fidelities relative to the ideal circuit on noisy hardware.Source: §4 · p. 7, Figure 2 · p. 7
  • For QFT-4, DQAS searches 12 gaps and finds noise-mitigation architectures with long-range correlations and multiple inserted gates collectively forming identity operations.The reported example includes three inserted gates on qubit 0 and correlated insertions on qubit 2.Source: §4 · p. 7, Figure 3 · p. 8

QAOA ansatz searching

  • On an Erdős–Rényi MAXCUT instance, DQAS reaches the exact cut value 12 with a reduced ansatz and nearly 100% one-shot success probability, versus 47% for P = 2 vanilla QAOA.The reduced ansatz uses about the same quantum resources as P = 1 QAOA while outperforming the vanilla P = 2 circuit.Source: §4 · p. 9

5. DISCUSSIONS

DQAS is presented as a versatile framework for automated, hardware-aware quantum circuit design across several NISQ applications. The work highlights reduced ansatz search, including shallower QAOA circuits that outperform conventional layouts, while noting related QAS efforts.

  • DQAS can be tailored to specific quantum hardware through customized noise models and native gate sets.Source: §5 · p. 9
  • The demonstrated applications include state preparation, unitary decomposition, noiseless and noisy hybrid quantum-classical circuit design, and reduced ansatz design.Source: §5 · p. 9
  • DQAS reformulates quantum-circuit and hybrid quantum-classical algorithm design as an automated differentiable architecture-search problem.Source: §5 · p. 9
  • A related QAS paper independently used random-search and evolutionary methods, and the two works together support QAS for quantum-circuit optimization.Source: §5 · p. 9

A. Glossary for ingredients of DQAS

DQAS uses a probabilistic super-network representation and Monte Carlo score-function gradients to search quantum circuits with implementable sampled unitary circuits. This probabilistic representation is more than an optional memory-saving device because direct super-network implementation is costly on NISQ hardware.

C. Derivation on gradients of probabilistic model parameters

The gradient derivation defines the objective’s forward-pass evaluation and applies a score-function approach to obtain gradients for probabilistic model parameters. The derivation concludes with the gradient formula used in DQAS.

D. General hyperparameters for DQAS training.

DQAS training depends on choices for optimization, batching, encoding, probabilistic modeling, regularization, and parameter reuse. The authors identify these ingredients as adjustable and leave extensive investigation of their effects to future work, without guaranteeing optimal hyperparameters.

E. Training techniques implemented in DQAS

DQAS training can incorporate techniques adapted from differentiable neural architecture search to improve robustness, reduce redundancy, control resources, and explore complex landscapes.

F. DQAS application in state preparation and circuit compiling

DQAS designs circuits for state preparation and unitary compilation by optimizing task-specific objectives over candidate quantum structures. Examples include GHZ-state preparation and a depth-5 Bell-state circuit.

State preparation

  • DQAS searches for GHZ-state preparation circuits from |0^n⟩ while progressively reducing depth until the objective can no longer be achieved.The operation pool includes parameterized Ry gates and CNOT gates subject to available connectivity.Source: F. DQAS application in state… · p. 17
  • The GHZ objective uses final-state distance, and DQAS tunes Ry(θ) to approximate a Hadamard when Hadamard is absent from the operation pool.Source: F. DQAS application in state… · p. 17

Unitary decomposition

Circuit compiling

G. Hyperparameter settings and training ingredients in experiments

The experiments vary operation pools, objectives, noise, batch sizes, and search strategies across state preparation, Bell circuits, QEM, and QAOA applications.

State preparation

  • The GHZ3 experiment uses Ry and nearest-neighbor CNOT operations, batch size 128, zero circuit-parameter initialization, and Adam with learning rate 0.15 for α.Source: G. Hyperparameter settings and… · p. 18

Bell circuit

Quantum error mitigation

  • QEM operation pools contain discrete gates including X, Y, Z, S, T, and identity, with the identity representing an idle qubit state.The QFT-3 setup uses batch size 256.Source: G. Hyperparameter settings and… · p. 18
  • QEM maximizes fidelity between noisy and ideal outputs, evaluating each circuit on a sampled input because fidelity variation across inputs was numerically small.Source: G. Hyperparameter settings and… · p. 18
  • QFT-4 search uses p = 12 slots, but its large search space can trap optimization locally; restricting single-qubit gates can reduce multistart requirements.Source: G. Hyperparameter settings and… · p. 18

QAOA

H. Further results on QEM of QFT circuit

DQAS-designed QEM circuits improve fidelity over bare and human-designed QEM circuits for QFT-3 and QFT-4 under the described bit-flip-noise setting.

Comparison

QFT-3

QFT-4

I. Further results on QAOA ansatz searching

Further results show DQAS can recover standard QAOA layouts, discover competitive alternatives, and customize circuits for individual MAXCUT instances. The study also examines hardware-aware objectives, search strategies, transfer to larger systems, and limitations from rugged landscapes and objective choice.

Hardware-aware search

  • DQAS incorporates hardware constraints by translating circuit layers into native gates and adding resource penalties to the search objective.For the stated gate costs, an xx-layer on a degree-3 regular graph has weight 27/2, with weights adjustable as hyperparameters.Source: §I · p. 20

Search-space design

  • Block encoding combines primitive operations into multi-layer blocks, helping DQAS discover the standard QAOA layout for P = 3.Examples include zz-rx-block and yy-rz-block constructions while retaining the Hadamard layer in the operation pool.Source: §I · pp. 20–21

Alternative layouts

  • For an n = 8 degree-3 regular-graph ensemble, DQAS found an H, yy, rx, zz, rx architecture with MAX CUT expectation around 8.8, similar to same-depth QAOA.The result demonstrates that DQAS can identify competitive layouts beyond the vanilla QAOA structure.Source: §I · p. 21

Ensemble learning

  • DQAS rediscovered the QAOA layout for Erdős–Rényi graph ensembles with n = 10, p = 0.3 and n = 8, p = 0.4.The study notes that Erdős–Rényi ensembles usually have greater variance in exact MAX CUT values.Source: §I · p. 21

Instance learning

  • For individual MAXCUT instances, DQAS produced a yy, zz, yy circuit with expected energy −8.0 versus −7.75 for same-depth vanilla QAOA.Customized architectures may outperform generic QAOA when circuit depth is restricted, although similar-performing candidates can create local-minimum traps.Source: §I · p. 22
Loading 2010.08561v2…