Classification of quantum correlations via quantum-inspired machine learning

Quantum information theory, and in particular, the theory of quantum state discrimination, has enabled the development of a supervised multi-class classification algorithm. Inspired by the Pretty Good Measurement (PGM), a quantum-inspired classifier named the PGM Classifier has been designed, capable of classifying among multiple classes without resorting to the One versus One or One versus Rest strategy. In the recent past, several approaches of quantum machine learning have been introduced to classify entangled and separable states. In this article, we apply the PGM Classifier to discriminate among factorized states, separable states, and entangled states. We analyze from the simplest case (2 qubits) up to a system of (5 qubits) and offer a comparison between our quantum-inspired classifier and other classical classifiers. Finally, we propose a refined classification of non-locality based on the violation of the Mermin and Svetlichny inequalities.