Classification of Cells Infected by Malaria with Vision Transformers Models

• Eduardo David Campopiano CPQD http://orcid.org/0009-0005-3613-7257
• Iago Fonseca Marinho Pereira CPQD https://orcid.org/0009-0000-0929-5456
• Thomas William CPQD
• Douglas Henrique Siqueira Abreu PUC-Campinas https://orcid.org/0009-0005-4739-5980
• Guilherme Ribeiro Sales CPQD
• Valentino Corso CPQD
• Cides S. Bezerra CPQD

Abstract

This project develops a system for malaria classification using the Vision Transformer architecture on cellular tissue microscopy images. The implementation aims to assist in medical diagnosis by providing efficient and accurate information, with the potential to contribute to the control of endemic diseases. The system utilizes models from the vit-pytorch library, including the Vision Transformer, Patch Merger, and Vision Transformer for Small Datasets. The Vision Transformer for Small Datasets achieved the highest accuracy of 98.97 for an image size of [150,150].

Keywords: Vision Transformer, Malaria Classification, Cellular Tissue Microscopy, vit-pytorch