基于近邻样本联合学习模型的疟疾识别算法

Neighbor sample joint learning for malaria parasite recognition

疟疾早期诊断可以有效防止疾病暴发.深度学习在细胞形态和组织图像检测等任务中具有出色能力.已有许多基于深度学习的疟疾研究,但它们主要用于环状体和红细胞二分类.本文首次研究疟疾多阶段识别,并提出近邻样本联合学习(neighbor sample joint learning,NSJL)模型.NSJL包括卷积神经网络(convolutional neural network,CNN)特征学习、领域相关性挖掘和图特征嵌入.它提取CNN特征,并将其与K近邻(K-nearset neighbor,K-NN)建立的邻域图传入图卷积网络(graph convolutional network,GCN).为评估NSJL,将其与先进方法比较,结果表明NSJL模型可达92.50%准确率,92.84%精确度,92.50%召回率和92.52%F1分数,至少高于其他方法7%的准确率表明其优秀疟疾识别能力.

Malaria is a serious and fatal infectious disease and early detection of the patient's infection severity can effectively curb the outbreak of this infectious disease.Deep learning has been verified to have excellent capability in image classification and disease diagnosis in many challenging tasks,such as cell detection and histological image classification.There exist many deep learning studies on malaria parasite recognition with successful applications,but they mainly focus on binary classification of single ring stage and red blood cells.The most important disadvantages of them are ignoring other stages of malaria parasites,including Trophozoite,Gametocytes and Schizont.In this paper,we are the first to study the multi-stage malaria parasite recognition problem,and propose a novel Neighbor Sample Joint Learning(NSJL)for this challenging task.Specifically,NSJL consists of CNN(Convolutional Neural Network)feature learning,neighbour correlation mining and graph representation modules.The method firstly extracts CNN representations from each parasite image and then establishes the neighbour correlations among CNN features by K-Nearest Neighbour(K-NN)graph building algorithms,with operating Graph Convolutional Network(GCN)on CNN features and their correlations.To evaluate the performance of our NSJL model,we compare it with several advanced existing methods,and our model can achieve a high accuracy of 92.50%,precision of 92.84%,recall of 92.50%and F1-score of 92.52%.The comparison with existing advanced methods verifies the NSJL model has excellent capability for recognizing multi-stage malaria parasite,which is higher than the compared methods by at least 7%in accuracy.