A Trainable-Embedding Quantum Physics-Informed Framework for Multi-Species Reaction-Diffusion Systems
Ban Q. Tran, Nahid Binandeh Dehaghani, A. Pedro Aguiar, Rafal Wisniewski, Susan Mengel
TL;DR
This work introduces extended Trainable-Embedding Quantum Physics-Informed Neural Networks (x-TE-QPINNs) to solve multi-species reaction–diffusion PDEs by comparing classical embedding (FNN) and fully quantum embedding (QNN) approaches within a unified physics-informed training framework. The method integrates a trainable embedding network, a hardware-efficient variational quantum circuit, and species-specific readouts, all enforced by a PDE-residual loss, boundary conditions, and initial conditions. Key contributions include a controlled comparison of embedding strategies, a differentiable training pipeline combining parameter-shift quantum gradients with automatic differentiation, and empirical results on 1D and 2D RD systems showing quantum embeddings can match or surpass classical embeddings in certain regimes while offering insights into resource requirements. This work provides architectural guidance for resource-efficient quantum PDE solvers and highlights the trade-offs between embedding design, gradient structure, and quantum hardware constraints in near-term quantum computing contexts.
Abstract
Physics-informed neural networks (PINNs) and hybrid quantum-classical extensions provide a promising framework for solving partial differential equations (PDEs) by embedding physical laws directly into the learning process. In this work, we study embedding strategies for trainable embedding quantum physics-informed neural networks (TE-QPINNs) in the context of nonlinear reaction-diffusion (RD) systems. We introduce an extended TE-QPINN (x-TE-QPINN) architecture that supports both classical and fully quantum embeddings, enabling a controlled comparison between feedforward neural network-based feature maps and parameterized quantum circuit embeddings. The first architecture is the classical embedding feed-forward neural network-based TE-QPINN (FNN-TE-QPINN), while the latter variant is a purely quantum one, referred to as quantum embedding neural network-based TE-QPINN (QNN-TE-QPINN). The proposed framework employs hardware-efficient variational quantum circuits and species-specific readout operators to approximate coupled multi-field dynamics while enforcing governing equations, boundary conditions, and initial conditions through a physics-informed loss function. By isolating the embedding mechanism while keeping the variational ansatz, loss formulation, and optimization procedure fixed, we analyze the impact of embedding design on gradient structure, parameter scaling, and quantum resource requirements. Numerical experiments on one- and two-dimensional RD equations demonstrate that quantum embeddings can replace classical embeddings without degradation in solution accuracy and, in certain regimes, exhibit improved optimization behavior compared to classical PINNs and hybrid quantum models with fixed embeddings. These results provide architectural insight into hybrid quantum PDE solvers and inform the design of resource-efficient quantum physics-informed learning methods.
