Optimized deep CNN with rotation-driven features for malaria parasite detection
摘要
Manual microscopic analysis of blood smears for malaria detection is labor-intensive, time-consuming, and prone to human error. To address these limitations, this study proposes a self-supervised learning approach using RotNet, a rotation-based model, to differentiate between healthy and malaria-infected cells. RotNet learns image representations by predicting the rotation angle of input images, eliminating the need for manually labeled data. This method uses synthetic rotation labels to train the model, significantly reducing dependency on human annotations. The core contribution of this work lies in using embeddings and weights from a rotation-based pretext task on unlabeled images to improve classification accuracy in a downstream convolutional neural network (CNN) task. Key highlights include: (1) an efficient and accurate deep learning solution for malaria detection, (2) a novel rotation-based self-supervised strategy, and (3) a practical approach to tackle the scarcity of labeled medical data. The proposed method was tested on the National Institutes of Health (NIH) malaria dataset, achieving a remarkable AUC of 99.8%, surpassing previously reported results. Moreover, similar accuracy was attained using only 10% of the labeled data compared to fully supervised training, highlighting the effectiveness of self-supervised pretraining. Overall, RotNet demonstrates high potential as a robust, scalable, and efficient tool for automated malaria diagnosis.