🤖 AI Summary
This work addresses the low sampling efficiency and data redundancy inherent in classical energy-based models by integrating a coherent Ising machine into the PyTorch ecosystem for the first time, enabling seamless fusion of photonic quantum computing with deep learning frameworks. Leveraging quantum-accelerated Boltzmann sampling and active sample selection, the authors develop hybrid quantum-classical architectures such as QBM-VAE and Q-Diffusion. Evaluated on single-cell and OpenWebText datasets, the proposed approach achieves state-of-the-art performance, demonstrating the efficacy of quantum-enhanced energy-based models in efficient training and active learning scenarios.
📝 Abstract
This paper introduces the Kaiwu-PyTorch-Plugin (KPP) to bridge Deep Learning and Photonic Quantum Computing across multiple dimensions. KPP integrates the Coherent Ising Machine into the PyTorch ecosystem, addressing classical inefficiencies in Energy-Based Models. The framework facilitates quantum integration in three key aspects: accelerating Boltzmann sampling, optimizing training data via Active Sampling, and constructing hybrid architectures like QBM-VAE and Q-Diffusion. Empirical results on single-cell and OpenWebText datasets demonstrate KPPs ability to achieve SOTA performance, validating a comprehensive quantum-classical paradigm.