Quantum Processing Unit (QPU) processing time Prediction with Machine Learning
Lucy Xing, Sanjay Vishwakarma, David Kremer, Francisco Martin-Fernandez, Ismael Faro, Juan Cruz-Benito
TL;DR
The paper addresses predicting QPU processing time for quantum jobs to improve scheduling and resource management in quantum computing systems. It adopts a LightGBM gradient-boosting approach with careful data encoding on a large IBM Quantum dataset, and compares ML predictions to a heuristic baseline. The key contributions include demonstrating that ML predictions are more accurate than the heuristic for both sampler and estimator runtime primitives and providing a practical data-processing and modeling workflow. The findings suggest ML-enabled tooling can enhance quantum workload management and lay groundwork for integrating AI-driven features into quantum computing stacks.
Abstract
This paper explores the application of machine learning (ML) techniques in predicting the QPU processing time of quantum jobs. By leveraging ML algorithms, this study introduces predictive models that are designed to enhance operational efficiency in quantum computing systems. Using a dataset of about 150,000 jobs that follow the IBM Quantum schema, we employ ML methods based on Gradient-Boosting (LightGBM) to predict the QPU processing times, incorporating data preprocessing methods to improve model accuracy. The results demonstrate the effectiveness of ML in forecasting quantum jobs. This improvement can have implications on improving resource management and scheduling within quantum computing frameworks. This research not only highlights the potential of ML in refining quantum job predictions but also sets a foundation for integrating AI-driven tools in advanced quantum computing operations.
