HyperAIHyperAI

Command Palette

Search for a command to run...

Détection d'objets semi-supervisée avec apprentissage contrastif par objet et incertitude de régression

Honggyu Choi Zhixiang Chen Xuepeng Shi Tae-Kyun Kim

Résumé

La détection d'objets semi-supervisée (SSOD) vise à améliorer les performances de détection en exploitant des données non étiquetées supplémentaires. Le cadre enseignant-étudiant s'est révélé prometteur pour la SSOD, dans lequel un réseau enseignant génère des pseudo-étiquettes pour les données non étiquetées afin d’assister l’entraînement d’un réseau étudiant. Étant donné que ces pseudo-étiquettes sont bruitées, leur filtrage s’avère crucial pour exploiter pleinement le potentiel de ce cadre. Contrairement aux méthodes existantes suboptimales, nous proposons une méthode de filtrage des pseudo-étiquettes en deux étapes, appliquée séparément aux têtes de classification et de régression dans un cadre enseignant-étudiant. Pour la tête de classification, nous introduisons OCL (Object-wise Contrastive Learning), une méthode de régularisation de l’apprentissage des représentations d’objets qui utilise les données non étiquetées pour améliorer le filtrage des pseudo-étiquettes en renforçant la discriminabilité des scores de classification. Cette approche a pour objectif de regrouper les objets appartenant à la même classe tout en éloignant ceux appartenant à des classes différentes. Pour la tête de régression, nous proposons par ailleurs RUPL (Regression-Uncertainty-guided Pseudo-Labeling), une méthode qui apprend l’incertitude aleatoire de la localisation des objets afin de guider le filtrage des étiquettes. En filtrant conjointement les pseudo-étiquettes des têtes de classification et de régression, le réseau étudiant reçoit une guidance plus précise du réseau enseignant pour la tâche de détection d’objets. Les résultats expérimentaux sur les jeux de données Pascal VOC et MS-COCO démontrent l’efficacité de notre méthode, qui atteint des performances compétitives par rapport aux approches existantes.


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