HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Inconsistency Masks: Removing the Uncertainty from Input-Pseudo-Label Pairs

Michael R. H. Vorndran Bernhard F. Roeck

Inconsistency Masks: Removing the Uncertainty from Input-Pseudo-Label Pairs

Abstract

Efficiently generating sufficient labeled data remains a major bottleneck in deep learning, particularly for image segmentation tasks where labeling requires significant time and effort. This study tackles this issue in a resource-constrained environment, devoid of extensive datasets or pre-existing models. We introduce Inconsistency Masks (IM), a novel approach that filters uncertainty in image-pseudo-label pairs to substantially enhance segmentation quality, surpassing traditional semi-supervised learning techniques. Employing IM, we achieve strong segmentation results with as little as 10% labeled data, across four diverse datasets and it further benefits from integration with other techniques, indicating broad applicability. Notably on the ISIC 2018 dataset, three of our hybrid approaches even outperform models trained on the fully labeled dataset. We also present a detailed comparative analysis of prevalent semi-supervised learning strategies, all under uniform starting conditions, to underline our approach's effectiveness and robustness. The full code is available at: https://github.com/MichaelVorndran/InconsistencyMasks

Code Repositories

michaelvorndran/inconsistencymasks
Official
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
lesion-segmentation-on-isic-2018AIM++ (256x256, 1.5m parameters, 10% labeled data, no pretraining)
mean Dice: 0.85
semi-supervised-medical-image-segmentation-on-6AIM++ (256x256, 1.5m parameters, 10% labeled data, no pretraining)
Dice Score: 0.85
semi-supervised-semantic-segmentation-on-43IM++ (416x208, 2.7m parameters, no pretraining)
Mean IoU (class): 0.428
semi-supervised-semantic-segmentation-on-suimAIM+ (256x256, 2.7m parameters, 10% labeled data, no pretraining)
Mean IoU (class): 0.482

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp