Improving Multi-Class Classification for Endomicroscopic Images by Semi-Supervised Learning

Abstract

Optical Endomicroscopy (OE) is a newly-emerged biomedical imaging modality that can help physicians make real-time clinical decisions about patients' grade of dysplasia. However, the performance of applying medical imaging classification for computer-aided diagnosis is primarily limited by the lack of labeled images. To improve the classification performance, we propose a semi-supervised learning algorithm that can incorporate large sets of unlabeled images. Our real-world endo-microscopic imaging datasets consist of 425 labeled images and 2,826 unlabeled ones. With semi-supervised learning algorithms, we improved multi-class classification performance over supervised learning algorithms by around 10% in all evaluation metrics, namely precision, recall, F1 score and Cohen-Kappa statistics.