HyperAIHyperAI

Command Palette

Search for a command to run...

Un cadre d’apprentissage semi-supervisé simple pour la détection d’objets

Kihyuk Sohn Zizhao Zhang Chun-Liang Li Han Zhang Chen-Yu Lee Tomas Pfister

Résumé

L’apprentissage semi-supervisé (SSL) présente un potentiel important pour améliorer les performances prédictives des modèles d’apprentissage automatique en exploitant des données non étiquetées. Bien qu’il ait connu des progrès remarquables récemment, la portée des démonstrations dans le domaine du SSL s’est principalement limitée aux tâches de classification d’images. Dans cet article, nous proposons STAC, un cadre SSL simple mais efficace pour la détection d’objets visuels, accompagné d’une stratégie d’augmentation de données. STAC utilise des pseudo-étiquettes à haute confiance pour des objets localisés provenant d’images non étiquetées, et met à jour le modèle en imposant une consistance à travers des augmentations fortes. Nous introduisons des protocoles expérimentaux pour évaluer les performances de la détection d’objets semi-supervisée sur MS-COCO, et démontrons l’efficacité de STAC tant sur MS-COCO que sur VOC07. Sur VOC07, STAC améliore le score AP0.5^{0.5}0.5 de 76,30 à 79,08 ; sur MS-COCO, STAC atteint une efficacité deux fois supérieure en matière de données en obtenant un mAP de 24,38 avec seulement 5 % de données étiquetées, contre 23,86 % pour la méthode supervisée basée sur 10 % de données étiquetées. Le code est disponible à l’adresse suivante : https://github.com/google-research/ssl_detection/.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp
Un cadre d’apprentissage semi-supervisé simple pour la détection d’objets | Articles | HyperAI