Bacterial Classification Using Deep Structured Convolutional Neural Network for Low Resource Data

Abstract

Bacterial identification is an essential task in medical disciplines and food hygiene. The characteristics of bacteria can be examined under a microscope using culture techniques. However, traditional clinical laboratory culture methods require considerable work, primarily physical and manual effort. An automated process using deep learning technology has been widely used for increasing accuracy and decreasing working costs. In this paper, our research evaluates different types of existing deep CNN models for bacterial contamination classification when low-resource data are used. They are baseline CNN, GCNN, ResNet, and VGGNet. The performance of CNN models was also compared with the traditional machine learning method, including SIFT+SVM. The performance of the DIBaS dataset and our own collected dataset have been evaluated. The results show that VGGNet achieves the highest accuracy. In addition, data augmentation was performed to inflate the dataset. After fitting the model with augmented data, the results show that the accuracy increases significantly. This improvement is consistent in all models and both datasets.

Keywords

• bacterial classification
• deep learning
• convolutional neural network (cnn)
• e-coli