Out of distribution detection in deep learning-based scattering pattern classification

Abstract

Intrinsic biophysical cell properties hold an enormous potential for cell class and state classification in microfluidics, allowing to avoid the need of cost intensive fluorescence labelling. Several methods can accomplish cell identification, while convolutional neural networks show an outstanding performance compared to other state-of-the-art classification methods, regarding accuracy and speed. In fact, neural networks show high performance for known image class prediction but struggles when unknown (out of distribution) image classes need to be identified. In such a scenario no prior knowledge of the unknown cell class can be used for the model training, which inevitably results in image misclassification. In fact, to distinguish unknown cell classes, a neural network must first construct an in-distribution of known images to afterwards detect out of distribution as unknowns, which is also called open-set classification assumption. Ones, a new cell class is identified, the neural network can be retrained with the obtained knowledge to dynamically update its cell class database. This process can be simply repeated for each new detected cell class. We applied this open-set idea to scattering pattern snapshots of different classes of living cells obtained in microfluidics. Our outcome shows a proof-of-concept for openset based convolutional neural network for cell image classification, which can be applied to a wide range of single cell classification approaches to reduce uncertainty in machine learning based technologies.