Kaiwu-PyTorch-Plugin: Bridging Deep Learning and Photonic Quantum Computing for Energy-Based Models and Active Sample Selection
Organizations: Beijing QBoson Quantum Technology Co., Ltd. · Chinese Academy of Sciences
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.
Figures & tables
| Immune | HLCA | Lung | ||||
|---|---|---|---|---|---|---|
| Model | Bio. | Batch | Bio. | Batch | Bio. | Batch |
| VAE Kingma and Welling (2013) | 0.55 | 0.36 | 0.54 | 0.34 | 0.56 | 0.39 |
| scVI Ergen et al. (2025) | 0.65 | 0.59 | 0.72 | 0.44 | 0.52 | 0.47 |
| LDVAE Ergen et al. (2025) | 0.55 | 0.58 | 0.65 | 0.43 | 0.53 | 0.47 |
| scPoli De Donno et al. (2023) | 0.65 | 0.59 | 0.71 | 0.43 | 0.55 | 0.47 |
| QBM-VAE | 0.69 | 0.61 | 0.76 | 0.48 | 0.67 | 0.49 |
| Model | Ours | SEDD | MLDM |
|---|---|---|---|
| PPL | 22.54 | 24.56 | 23.83 |
Hybrid Quantum-Classical ML
Hybrid quantum-classical machine learning combines quantum circuits or annealers with classical models for learning, prediction, generation, and optimization.
See all 179 papers