Reliable Mislabel Detection for Video Capsule Endoscopy Data

· Source: cs.CV updates on arXiv.org · Field: Health & Wellbeing — Medical Devices & Health Technology, Health & Medical Research · Depth: Expert, extended

Summary

A new framework has been introduced for reliable mislabel detection in medical imaging datasets, specifically validated for Video Capsule Endoscopy (VCE) data. This framework addresses the challenge of noisy annotations in medical datasets, which often require specialized physician expertise and suffer from ambiguous class boundaries. The pipeline was tested on the Kvasir-Capsule and Galar datasets, the two largest publicly available VCE datasets. On Kvasir-Capsule, it successfully detected 456 out of 471 (1% noise) and 2262 out of 2360 (5% noise) injected mislabels. When applied to the Galar dataset, the framework filtered 167,709 samples (4.8%) and corrected 31,650 samples (0.9%). Training a neural network on this cleaned Galar dataset boosted anomaly detection performance, achieving an accuracy of 93.83% and an F1-score of 71.58%, significantly surpassing previous baselines of 37.01% and 54.38%. Clinical validation by three experienced gastroenterologists confirmed 78% of 100 suspected mislabels were indeed incorrect.

Key takeaway

For Machine Learning Engineers developing anomaly detection models for Video Capsule Endoscopy, you should prioritize implementing a robust mislabel detection and cleaning pipeline. This approach significantly improves model accuracy and F1-scores compared to training on uncleaned data. By identifying and correcting noisy labels, you can achieve more reliable diagnostic support and enhance overall classification performance for critical medical applications.

Key insights

Cleaning medical imaging datasets of mislabels significantly improves deep neural network anomaly detection performance.

Principles

Method

The pipeline trains a CNN, then a Gaussian Mixture Model (GMM) to determine noise probability. It corrects top k^c labels with highest noise reduction and filters top k^f labels with highest noise probability.

In practice

Topics

Best for: Computer Vision Engineer, AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.