Exponential Quantum Communication Advantage in Distributed Inference and Learning

Training and inference with large machine learning models that far exceed the memory capacity of individual devices necessitates the design of distributed architectures, forcing one to contend with communication constraints. We present a framework for distributed computation over a quantum network in which data is encoded into specialized quantum states. We prove that for models within this framework, inference and training using gradient descent can be performed with exponentially less communication compared to their classical analogs, and with relatively modest overhead relative to standard gradient-based methods. We show that certain graph neural networks are particularly amenable to implementation within this framework, and moreover present empirical evidence that they perform well on standard benchmarks. To our knowledge, this is the first example of exponential quantum advantage for a generic class of machine learning problems that hold regardless of the data encoding cost. Moreover, we show that models in this class can encode highly nonlinear features of their inputs, and their expressivity increases exponentially with model depth. We also delineate the space of models for which exponential communication advantages hold by showing that they cannot hold for linear classification. Our results can be combined with natural privacy advantages in the communicated quantum states that limit the amount of information that can be extracted from them about the data and model parameters. Taken as a whole, these findings form a promising foundation for distributed machine learning over quantum networks.

Paper

Similar papers

Peer review

Reviewer K89H6/10 · confidence 4/52024-07-11

Summary

This paper studies distributed learning and inference based on graph neural networks but with quantum communications between the distributed agents. The authors show that quantum networks reduce the communication complexity inference and gradient computation of distributed models. This reduction is shown to be exponential in communication complexity. The main idea of the paper is to have the data encoded into qubits using amplitude encoding. With that one gets an exponential reduction in space and hence communication complexity. As for the gradient estimation, the problem is shown to be converted to Aaronson's Shadow tomography.

Strengths

The paper proposes an exponential advantage in communication complexity for the task of distributed inference and gradient estimation. This is a nice set of advantages that can be employed in practical problems. Also the lower bounds are nice contributions.

Weaknesses

The result of the paper is not surprising given the known results in quantum communication complexity. Moreover, the results seem to be directly derived from known papers on Shadow tomography and the Threshold search.

Questions

Can you elaborate on the run-time of the proposed method?

Rating

6

Confidence

4

Soundness

4

Presentation

4

Contribution

2

Limitations

Nothing significant.

Reviewer mkaP7/10 · confidence 4/52024-07-15

Summary

The paper investigates the exponential quantum advantage in communication complexity over the tasks of training or inference in multiple types of compositional distributed quantum circuit model. The authors also study the quantum communication advantage in a specific class of shallow graph neural networks, and then they conduct experiments showing that it can achieve performance comparable to certain existing classical models and benchmarks while maintaining the exponential communication advantage. At the end, the authors explore the expressivity of the compositional model.

Strengths

I think this paper is one of the good examples of applying results and tools from quantum complexity theory to problems in distributed computing.

Weaknesses

(1) Lemma 6 addresses the quantum communication complexity of the quadratic graph network inference problem, for which I think a lower bound would make more sense. However, (I am not sure if I miss anything or not), instead of giving a lower bound, Lemma 6 provides an upper bound, which I find confusing. (2) The paper assumes that the input is amplitude encoded. It is well-known that preparing such quantum states is generally not efficient. This is an important factor to consider regarding the feasibility of the proposed systems, and thus, it is my main concern with the paper. However, I don't see much formal treatment of this issue in the paper.

Questions

My questions are those 2 points in Weaknesses.

Rating

7

Confidence

4

Soundness

3

Presentation

3

Contribution

3

Limitations

Yes, the authos give a fair discussion on possible limitations.

Reviewer W5VQ6/10 · confidence 2/52024-07-23

Summary

This paper studies the quantum advantage on communication of distributed learning. It places a common quantum neural network model in the (two-party) communication scenario, where the data x is assigned to Alice, and the parameterized unitary operators of each layer are alternately given to Alice and Bob in order. The paper shows exponential quantum advantage on communication for problems of estimating the loss function (some function on data after the action of unitary operators) and estimating the gradients of the loss function with respect to each parameter of the unitary operators. The quantum upper bound for estimating the loss function is by encoding data x (size-N vector) into O(log N)-qubit state. The quantum upper bound for estimating the gradients is by a reduction to a shadow tomography problem. The classical lower bound is by a reduction from a problem proposed by Raz, also a round-based lower bound by a reduction from pointer-chasing. For comparison, the paper shows that no exponential quantum advantage on communication of linear classification, where Alice and Bob are given data x and y as vectors to determine the sign of inner product of x and y. The classical upper bound is by applying Johnson-Lindenstrauss and the quantum lower bound is by a reduction from gap-Hamming. The paper also shows exponential quantum advantage in communcication for quadratic graph network inference, and provides experimental results for the above scenarios.

Strengths

- A new research direction on quantum advantage of communication complexity for learning problems, providing a series of results with solid theoretical proofs and experimental data to support them.

Weaknesses

- For technical part, the upper bounds are straightforward, and the lower bound reductions look standard. - These learning tasks (estimating the loss function in quantum neural networks, linear classification, and quadratic graph network inference) seem to have weak correlations, making the paper appear somewhat like a mere compilation of results.

Questions

- Minor comment: the upper bound in Lemma 1 should be O(L log N) instead of O(log N)?

Rating

6

Confidence

2

Soundness

3

Presentation

3

Contribution

2

Limitations

- The claim of QUANTUM advantage on QUANTUM neural networks seems a bit like cheating, although the model and problem are described using classical information. - The two-party version of quantum neural networks seems artificial to me. The data and parameters of each single unitary operator might be stored distributively between parties, just like in problem 7 of linear classification, where data x and y are assigned to Alice and Bob.

Reviewer W5VQ2024-08-09

Thank you for the response. Indeed the classical neural networks can be expressed as quantum circuits, but I am not sure whether converting classical neural networks into quantum circuits introduces much additional communication in your distributed setting.

Authorsrebuttal2024-08-09

In our setting, each party constructs a set of unitaries locally, requiring no communication. Note that each unitary can be arbitrarily complex, since this is done locally. Each unitary can also be a nonlinear function of classical data or model parameters. The total communication complexity of inference will then scale linearly with the number of times the quantum state encoding the features has to be transferred back and forth between the players when evaluating the resulting circuit. A possible concern here is that it might be hard to construct expressive models in this way, or models that approximate the action of realistic classical neural networks. We show in section 4 however that for certain classes of graph GNNs, models of this form require only a single round of communication to run inference and are expressive enough to perform well on standard benchmarks.

Reviewer mkaP2024-08-14

Thanks for the response. As for comments on W2, do you mean in general performing amplitude encoding of input data takes time linear in N?

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC