HyperAIHyperAI

Command Palette

Search for a command to run...

Amélioration du réglage des prompts visuels pour les transformateurs vision auto-entraînés

Seungryong Yoo Eunji Kim Dahuin Jung Jungbeom Lee Sungroh Yoon

Résumé

Visual Prompt Tuning (VPT) est une méthode efficace d’adaptation des Vision Transformers (ViTs) préentraînés aux tâches spécifiques. Elle exploite des tokens apprenables supplémentaires, appelés « prompts », qui orientent les ViTs préentraînés, maintenant leurs paramètres gelés. Bien que le VPT ait démontré son efficacité avec les ViTs supervisés, il se comporte souvent de manière inférieure lorsqu’il est appliqué à des ViTs auto-supervisés. À partir d’observations empiriques, nous constatons que l’efficacité du VPT dépend largement des blocs ViT avec lesquels les tokens de prompt interagissent. Plus précisément, le VPT améliore les performances sur les tâches de classification d’images pour MAE et MoCo v3 lorsque les tokens de prompt sont insérés dans des blocs ultérieurs plutôt que dans le premier bloc. Ces observations suggèrent l’existence d’un emplacement optimal pour l’insertion des tokens de prompt au sein des ViTs auto-supervisés. Malheureusement, identifier ces blocs optimaux pour chaque ViT auto-supervisé dans des scénarios futurs divers est un processus coûteux. Pour atténuer ce problème, nous proposons une méthode simple mais efficace qui apprend une porte (gate) pour chaque bloc ViT afin d’ajuster son intervention sur les tokens de prompt. Grâce à cette approche, les tokens de prompt sont influencés sélectivement par les blocs nécessitant une orientation pour l’adaptation à la tâche. Notre méthode surpasser les variantes du VPT sur les tâches de classification d’images FGVC et VTAB, ainsi que sur la segmentation sémantique ADE20K. Le code est disponible à l’adresse suivante : https://github.com/ryongithub/GatedPromptTuning.


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