Layerwise Federated Learning for Heterogeneous Quantum Clients using Quorus
Jason Han, Nicholas S. DiBrita, Daniel Leeds, Jianqiang Li, Jason Ludmir, Tirthak Patel
TL;DR
Quorus tackles the challenge of training quantum models across private clients with heterogeneous circuit depths by introducing layerwise losses and reverse distillation in a federated setting. It offers multiple hardware-aware designs (Layerwise, Ancilla/Blocking, Funnel) to balance shot budget, qubit count, and midcircuit capabilities, while ensuring synchronized objectives among clients. Empirical results show up to a 12.4% improvement in testing accuracy over Q-HeteroFL and higher gradient magnitudes for deeper clients, with validation on IBM QPUs that underscores real-world viability. Collectively, Quorus provides a practical, noise-aware QFL framework that enables scalable quantum machine learning across diverse hardware, supported by open-source implementations. The work advances depth-heterogeneous quantum collaboration, paving the way for privacy-preserving and resource-aware quantum learning on near-term devices.
Abstract
Quantum machine learning (QML) holds the promise to solve classically intractable problems, but, as critical data can be fragmented across private clients, there is a need for distributed QML in a quantum federated learning (QFL) format. However, the quantum computers that different clients have access to can be error-prone and have heterogeneous error properties, requiring them to run circuits of different depths. We propose a novel solution to this QFL problem, Quorus, that utilizes a layerwise loss function for effective training of varying-depth quantum models, which allows clients to choose models for high-fidelity output based on their individual capacity. Quorus also presents various model designs based on client needs that optimize for shot budget, qubit count, midcircuit measurement, and optimization space. Our simulation and real-hardware results show the promise of Quorus: it increases the magnitude of gradients of higher depth clients and improves testing accuracy by 12.4% on average over the state-of-the-art.
