HyperAIHyperAI

Command Palette

Search for a command to run...

Transformateurs visuels évolutifs avec regroupement hiérarchique

Zizheng Pan Bohan Zhuang† Jing Liu Haoyu He Jianfei Cai

Résumé

Les Transformers visuels (ViT) récemment proposés avec une attention pure ont obtenu des performances prometteuses dans les tâches de reconnaissance d'images, telles que la classification d'images. Cependant, le processus actuel des modèles ViT consiste à maintenir une séquence complète de patches pendant l'inférence, ce qui est redondant et manque de représentation hiérarchique. À cet égard, nous proposons un Transformer visuel hiérarchique (HVT) qui regroupe progressivement les jetons visuels pour réduire la longueur de la séquence et, par conséquent, diminuer le coût computationnel, de manière similaire au sous-échantillonnage des cartes de caractéristiques dans les Réseaux Neuronaux Convolutifs (CNN). Cette approche offre l'avantage majeur de pouvoir augmenter la capacité du modèle en augmentant les dimensions de profondeur/largeur/résolution/taille de patch sans introduire une complexité computationnelle supplémentaire grâce à la réduction de la longueur de la séquence. De plus, nous avons constaté empiriquement que les jetons visuels moyennés contiennent plus d'informations discriminantes que le seul jeton de classe. Pour démontrer l'amélioration de la scalabilité de notre HVT, nous avons mené des expériences approfondies sur la tâche de classification d'images. Avec un nombre comparable d'opérations flottantes (FLOPs), notre HVT surpasse les lignes de base concurrentes sur les jeux de données ImageNet et CIFAR-100. Le code est disponible à l'adresse suivante : https://github.com/MonashAI/HVT


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