Deep Learning-based Classification of Viruses Using Transmission Electron Microscopy Images

Mohd Mohsin Ali, Rakesh Chandra Joshi, Malay Kishore Dutta, Radim Burget, Anzhelika Mezina

Abstract: Humans have a strong urge to categorize natural organisms, and the categorization of viruses becomes more challenging. Viruses are not visible with the naked eyes, and their automatic classification based on images obtained with Transmission Electron Microscopy (TEM) can help a lot in the medical field. Their classification is more challenging due to their complicated intracellular structures and lighting conditions to capture the TEM images. The proposed architecture has been developed for the classification of the 14 different types of viruses. The dataset has been split into the training set, validation set and test set. The proposed model obtained better experimental results with 96.90% classification accuracy on the validation set and 96.10 <sup xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">%</sup> on the test set of unseen images. The performance of the proposed model has been compared with state-of-the-art pre-trained deep-learning models such that XceptionNet, MobileNet and DenseNet201. The model is accurate and computationally less complex, which supports faster processing suitable for microscopic cell image analysis for different medical applications.