Towards Explainability of Classical Neural Network via Quantum Computing
Junyong Lee, Jeihee Cho, Daniel Justice, Shiho Kim · 2024
In recent years, machine learning has achieved remarkable success in tasks such as image classification and solving complex real-world problems. The achievements of deep neural networks (DNNs) have led to the widespread adoption of block-based architectures. However, as the demand for Explainable AI (XAI) grows, there is a need for models that provide greater transparency and interpretability. Our work addresses this need by exploring the potential of quantum-classical hybrid models. By integrating classical neural networks (ClaNNs) with quantum neural networks (QNNs), we aim to develop AI systems that are not only powerful but also interpretable. We propose a framework that combines parameterized quantum circuits (PQCs) with classical layers to create a hybrid block design for QNNs. By adding a linear layer after the PQCs, the flexibility of the block design is increased. This architecture enhances the adaptability of QNN design and facilitates the integration of XAI principles through the comparison of the loss function planes. Our training method, inspired by knowledge distillation, addresses existing challenges in PQC-based QNN designs, providing insights into the optimization landscape. We validate the effectiveness of our hybrid block through experiments on standard datasets, demonstrating its potential to achieve comparable accuracy to traditional DNNs while providing improved interpretability.