AIAny
Icon for item

Circuit Hypernetworks for Quantum-Augmented Diffusion Language Models

Adds token-conditioned quantum residual branches to a frozen masked-diffusion language model: a lightweight hypernetwork emits continuous quantum-circuit coordinates per token, executes a shared sparse IQP-style circuit, and injects classically-expressible expectation readouts back into transformer blocks. Trains only the added branches, scaling to 16–64 qubits with analytic, linear-cost readouts.

Introduction

Why this matters Most work that mixes quantum circuits and LLMs either selects a small fixed circuit or incurs high simulation cost for wider registers. This paper shows a practical compromise: emit continuous, token-specific circuit coordinates from a tiny hypernetwork attached per token, keep the diffusion transformer backbone frozen, and compute circuit expectation values via an exact classical formula whose cost grows only linearly with qubit count. The result is a tractable architectural route to experiment with wider quantum registers inside large language models without relying on classically hard readouts.

Key Findings
  • Architecture: HyperQ adds a token-conditioned quantum residual branch in each transformer block, placed in parallel with the frozen fused QKV projection. A low-rank hypernetwork maps token hidden states to per-token circuit coordinates (single-qubit rotation angles, two-qubit coupling angles, readout axes), which parameterise a shared sparse IQP-style circuit (H — diagonal Rz / Rzz — H) before a readout rotation.
  • Classical efficiency: The required expectation values have an exact closed-form classical expression whose evaluation cost grows linearly with the number of qubits, enabling training of 16, 32 and 64-qubit branches inside a 1.1B-parameter frozen backbone.
  • Empirical improvements: Average downstream benchmark score rises from 47.65 (16 qubits) to 52.80 (32 qubits) and 54.30 (64 qubits). At 64 qubits HyperQ outperforms the frozen backbone (49.59) by 4.71 points and a classical low-rank adapter (50.63) by 3.67 points. HyperQ is fine-tuned on 20,000 prompt–response pairs vs 200,000 for classical baselines.
  • Practical stance: The paper explicitly claims no quantum advantage — readouts remain classically tractable at all tested widths — and demonstrates a bounded paired evaluation using measurements from a 156-qubit superconducting processor while keeping weights fixed.
Who it fits / Tradeoffs

Great fit if you are researching hybrid quantum–classical model architectures or token-level adaptation mechanisms and want a method that: reuses a frozen large backbone; emits continuous, token-specific circuit parameters; and keeps circuit readout classically tractable for wider registers. Look elsewhere if your goal is provable quantum advantage, purely quantum-native models, or production workloads that cannot tolerate the added inference/engineering complexity of per-token emitted circuits and readout projections. The approach prioritises architectural exploration and controlled empirical gains over demonstrating a classically intractable quantum speedup.

Where it sits technically

HyperQ targets model-architecture researchers and ML practitioners exploring quantum-augmented LLMs: it combines ideas from hypernetworks, low-rank adapters, and sparse IQP-style parameterised circuits to deliver token-conditioned quantum branches with analytic readouts. The main methodological novelty is continuous, per-token circuit emission (coordinates in R) rather than discrete circuit selection, plus the demonstration that such emission can be trained at scales up to 64 qubits while keeping classical evaluation feasible.

Information

  • Websitearxiv.org
  • OrganizationsAffiliation: Département d’informatique et de recherche opérationnelle, Université de Montréal, Montréal, QC, Canada, Affiliation: Mila, Quebec AI Institute, Montréal, QC, Canada, Affiliation: McGill University, Montréal, QC, Canada
  • AuthorsXiaoqiang Wang, Mengyang Xiong, Jun Dai, Bang Liu
  • Published date2026/09/21

More Items

Trains a single agent policy with successful trajectories from specialized sibling harnesses when attempts under the base harness all fail. The resulting policy improves under the original harness, with optional task-specific harness routing at inference.

Builds tool-composition environments from real execution traces and generates verifiable long-horizon agent tasks. Results show substantial gains in models’ ability to coordinate tools across environments.

Trains agent models with two supervised fine-tuning stages followed by reinforcement learning, and releases the data-generation, verification, and training pipeline. The recipe skips dedicated mid-training, DPO, and on-policy distillation.