PsiAudit: An open-source toolkit for auditing symmetry-organised complexity in equivariant quantum neural networks

Abstract

Parameterised quantum circuits are commonly assessed using measures such as expressibility, gradient behaviour, and entanglement. While useful, these measures do not indicate whether a circuit respects the symmetry it was designed to respect. This is especially important in equivariant quantum machine learning, where symmetry is central to the model. We introduce PsiAudit, an open-source Python toolkit for auditing symmetry-aware quantum neural network ansätze before training. Given an ansatz, a target symmetry, and a state trajectory, PsiAudit reports how the circuit occupies symmetry sectors, maintains coherence between them, fluctuates across the trajectory, and complies with the target symmetry. These outputs are combined into a configurable dashboard-style summary. PsiAudit supports phase, spin, and permutation symmetries, with the permutation audit implemented using Hamming-weight orbits. Tests on five ansatz families, across twenty random seeds and four to eight qubits, show, in the tested setting, that PsiAudit can identify inactive equivariant circuits, recover structure when multiple symmetry sectors are activated, and distinguish ansätze that appear similar under standard diagnostics. The toolkit also includes a unitary-level compliance check, reproducible examples, and a notebook for regenerating the reported results.