Semi-supervised learning from coarse histopathology labels

Abstract

ABSTRACT Ultrasound imaging is commonly used to guide sampling the prostate tissue in transrectal biopsies, followed by detection of cancer through histopathological analysis and coarse labelling of sampled tissue. Ideally, the procedure should be improved by developing machine learning solutions that can identify the presence of cancer in ultrasound images to guide the biopsy procedure. Training a fully supervised learning model using coarse histopathology labels suffers from weakly annotated data which introduce label noise for each image pixel. To address this challenge, we propose a semi-supervised framework for learning with noisy labels. We leverage a two-component mixture model to cluster the training data into clean and noisy label samples based on their loss values. Then, during the semi-supervised training phase, we utilise the well-known MixMatch algorithm which incorporates consistency regularisation, entropy minimisation, and the Mixup regularisation as well as the cross-entropy loss function for noisy and clean sets, respectively. We evaluate the proposed framework with prostate ultrasound data obtained from 71 subjects, while sampling 264 biopsy cores. We achieve balanced accuracy, sensitivity, and specificity of 78.6%, 80.0%, and 77.1%, respectively. In a detailed comparison study, we demonstrate that our proposed framework outperforms the fully supervised method with state-of-the-art robust loss functions.