IEEE Transactions on Quantum Engineering
Hybrid Quantum Capsule Network: Quantum Transformation Circuit as Vote Transformation in Image Classification
Ridho Nur Rohman Wijaya, Budi Setiyono, Dwi Ratna Sulistyaningrum
Abstract
A capsule network (CapsNet) provides a richer hierarchical representation than conventional neural networks through vector-shaped capsules and intercapsule routing. However, most CapsNet designs still rely on classic matrix-multiplication-based vote transformations, which limit the formation of the prediction vector to linear mappings in Euclidean space. In this article, we propose the hybrid quantum capsule network (HQCapsNet), an architecture that replaces classical vote transformations with a quantum transformation circuit (QTC) based on a parameterized quantum circuit (PQC). We introduce a QTC in two schemes: classwise QTC and fully connected QTC. Furthermore, the routing agreement mechanism is extended via hybrid quantum routing, leveraging various quantum similarity metrics. The evaluation was conducted on several binary image classification datasets, with MNIST, TMNIST, and MNIST-M used as the primary datasets. The experimental analysis included benchmarking of quantum embedding variations, the PQC ansatz, qubit-count analysis, circuit depth, and routing iterations. The experimental results show that the fully connected QTC is the best configuration, achieving consistent accuracy improvements over the classwise QTC, with the most significant improvement from 92.86% to 97.62% on the MNIST-M dataset. Compared to the classical CapsNet, the HQCapsNet also achieves higher performance across all primary datasets, achieving 99.95% accuracy on MNIST and competing with quantum–classical models on comparable split data. These results indicate that integrating quantum computing at the voting transformation step is a promising approach to improving CapsNet’s representation and generalization capabilities.