Efficiently recognition of vaginal micro-ecological environment based on Convolutional Neural Network

摘要

Vaginal diseases caused by vaginal micro-ecological abnormalities mainly include Vulvovaginal Candidiasis (VVC), Aerobic Vaginitis (AV), and Bacterial Vaginosis (BV). Severe cases can lead to poor pregnancy outcomes and infertility. AI-based technologies are being deployed with an expectation to relieve doctors of routine, tedious work when implemented correctly in daily microscopy of vaginal micro-ecological abnormalities. In this paper, we built a clinical image dataset of the Gram stain of the vaginal discharge. By comparing the performance of state of art convolutional neural network models, we found the fine-tuning Inception ResNet V2 shows the best classification performance for vaginal diseases. It achieves 96%, 94%, 86% AUC in VVC, AV, BV classification respectively. The result shows that compared with human visual inspection, the method based on deep learning greatly improves the screening sensitivity. Besides, we found that transfer learning can reduce the required manual labeling by roughly 73% (about more than one thousand samples). But for BV, which is difficult to diagnose for both humans and AI. Unlike AV and VVC, it requires more labeled data and is insensitive to the transfer fine-tuning.