HyperAIHyperAI

Command Palette

Search for a command to run...

vor 4 Monaten

Lernen von Pseudolabels für semi- und schwach überwachte semantische Segmentierung

{Shiguang Shan Meina Kan Jie Zhang Yude Wang}

Abstract

In diesem Paper befassen wir uns mit der semi- und schwach überwachten semantischen Segmentierung (SWSSS), bei der viele Bildklassifizierungslabels auf Bildebene und nur wenige Pixelannotierungen zur Verfügung stehen. Wir sind der Überzeugung, dass der entscheidende Aspekt zur Lösung des SWSSS-Problems die Erzeugung hochwertiger Pseudolabels ist, und gehen diesem Problem aus zwei Perspektiven entgegen. Erstens führen wir eine klassenbewusste Kreuzentropie-(CCE-)Verlustfunktion für das Netzwerktraining ein. Im Vergleich zur herkömmlichen Kreuzentropie verlangt die CCE-Funktion, dass das Modell lediglich zwischen gleichzeitig auftretenden Klassen unterscheiden muss, wodurch das Lernziel für die Generierung von Pseudolabels vereinfacht wird. Zweitens schlagen wir eine progressive Kreuztrainings-(PCT-)Methode vor, die einen dynamischen Bewertungsmechanismus nutzt, um zwischen zwei Netzwerken eine Kreuzüberwachung aufzubauen, wobei schrittweise hochwertige Vorhersagen als zusätzliche Überwachung für das Training eingesetzt werden. Unser Ansatz verbessert die Qualität der generierten Pseudolabels erheblich, insbesondere in Szenarien mit extrem begrenzten Annotationen. Umfangreiche Experimente zeigen, dass unsere Methode die derzeitigen State-of-the-Art-Verfahren deutlich übertrifft.

Benchmarks

BenchmarkMethodikMetriken
semi-supervised-semantic-segmentation-on-15PCT (DeepLab v3+ with ResNet-50 pretrained on ImageNet-1K)
Validation mIoU: 77.26%
semi-supervised-semantic-segmentation-on-21PCT (DeepLab v3+ with ResNet-50 pretrained on ImageNet-1K)
Validation mIoU: 71.35
semi-supervised-semantic-segmentation-on-4PCT (DeepLab v3+ with ResNet-50 pretrained on ImageNet-1K)
Validation mIoU: 75.52%
semi-supervised-semantic-segmentation-on-9PCT (DeepLab v3+ with ResNet-50 pretrained on ImageNet-1K)
Validation mIoU: 76.47

KI mit KI entwickeln

Von der Idee bis zum Start — beschleunigen Sie Ihre KI-Entwicklung mit kostenlosem KI-Co-Coding, sofort einsatzbereiter Umgebung und den besten GPU-Preisen.

KI-Co-Coding
Sofort einsatzbereit GPUs
Beste Preise
Jetzt starten

Hyper Newsletters

Abonnieren Sie unsere neuesten Updates
Wir werden die neuesten Updates der Woche in Ihren Posteingang liefern um neun Uhr jeden Montagmorgen
Unterstützt von MailChimp