quant-phJul 13, 2026

Fixed-Protocol Amortized MPS Tomography with Conformalized Predictive Uncertainty

Authors: Jian XuDelu ZengJohn PaisleyQibin Zhao

Organizations: 1RIKEN iTHEMS · 2RIKEN AIP · 3South China University of Technology · 4Columbia University

Abstract

Quantum state tomography is sample-starved, and the states one prepares live on a narrow, learnable manifold. A k=0k{=}0 prior-only control shows that on concentrated families a prior estimate is already near-optimal, so ``high fidelity at few measurements'' can be family memorization rather than tomography; genuine measurement-efficiency needs a model that conditions on the measurements and demonstrably uses them. On a shared matrix-product-state (MPS) core parameterization we study two routes. ApproachA learns a generative prior over MPS cores with measurement-guided posterior inference (gold-standard-validated, but whose few-measurement accuracy the control shows is largely the prior). ApproachB, our main proposal, is a \emph{fixed-protocol amortized} MPS estimator trained once with a gauge-invariant fidelity loss; we deliberately do not rest it on a permutation-invariant set encoder (a plain MLP matches it). The decisive lever is the measurement design: motivated by the fact that local reduced density matrices determine a χχ-MPS, conditioning on an \emph{informative local} Pauli set rather than random strings turns a modest, memorization-prone estimator into a high-fidelity one ( ⁣0.95\approx\!0.95, up to +0.59+0.59 over prior-only, decisively passing a shuffled-measurement control). A dropout ensemble, conformally recalibrated, gives  ⁣90%\approx\!90\%-coverage intervals -- including for observables never measured, where a shot-based interval does not exist. Quality holds as the system grows (fidelity 0.900.90 at n=10n{=}10, gain \emph{growing} in nn; 0.880.88 at bond dimension χ=4χ{=}4), the parameterization is polynomial (native contraction to 2020 qubits), and we close the loop on IBM hardware (55 states at 0.970.97 from hardware-measured Paulis).

Explore similar work

CardsList