Quantum entanglement is one of the essential resources involved in quantum information processing tasks. However, its detection for usage remains a challenge. The Bell-type inequality for relative entropy of coherence serves as an entanglement witness for pure entangled states. However, it does not perform reliably for mixed entangled states. This paper constructs a classifier by employing the relationship between coherence and entanglement for supervised machine learning methods. This method encodes multiple Bell-type inequalities for the relative entropy of coherence into an artificial neural network to detect the entangled and separable states in a quantum dataset.

Quantum entanglement is a unique phenomenon of quantum mechanics, which has no classical counterpart and gives quantum systems their advantage in computing, communication, sensing, and metrology. In quantum sensing and metrology, utilizing an entangled probe state enhances the achievable precision more than its classical counterpart. Noise in the probe state preparation step can cause the system to output unentangled states, which might not be resourceful. Hence, an effective method for the detection and classification of tripartite entanglement is required at that step. However, current mathematical methods cannot robustly classify multiclass entanglement in tripartite quantum systems, especially in the case of mixed states. In this paper, we explore the utility of artificial neural networks for classifying the entanglement of tripartite quantum states into fully separable, biseparable, and fully entangled states. We employed Bell’s inequality for the dataset of tripartite quantum states and train the deep neural network for multiclass classification. This entanglement classification method is computationally efficient due to using a small number of measurements. At the same time, it also maintains generalization by covering a large Hilbert space of tripartite quantum states.