HyperAIHyperAI

Command Palette

Search for a command to run...

Transformateur sans attention masquée pour la segmentation d'instances 3D

Xin Lai Yuhui Yuan Ruihang Chu Yukang Chen Han Hu Jiaya Jia

Résumé

Récemment, les méthodes basées sur les transformateurs ont dominé la segmentation d’instances 3D, où l’attention sur les masques est couramment utilisée. Plus précisément, les requêtes d’objets sont guidées par les masques d’instances initiaux lors de la première attention croisée, puis se réfinent itérativement selon un mécanisme similaire. Toutefois, nous observons que le pipeline d’attention sur les masques conduit généralement à une convergence lente en raison de masques d’instances initiaux à faible rappel. Par conséquent, nous abandonnons l’architecture d’attention sur les masques au profit d’une tâche auxiliaire de régression de centres. Grâce à cette régression de centres, nous surmontons efficacement le problème du faible rappel et réalisons l’attention croisée en imposant un prior positionnel. Pour atteindre cet objectif, nous proposons une série de conceptions sensibles à la position. Premièrement, nous apprenons une distribution spatiale des positions 3D comme requêtes initiales de position. Ces requêtes couvrent densesment l’espace 3D, permettant ainsi de capturer facilement les objets dans une scène avec un haut rappel. En outre, nous introduisons un encodage de position relative pour l’attention croisée et une réduction itérative pour améliorer la précision des requêtes de position. Les expérimentations montrent que notre approche converge 4 fois plus rapidement que les méthodes existantes, établit un nouveau record sur le benchmark de segmentation d’instances 3D ScanNetv2, et démontre également des performances supérieures sur diverses bases de données. Le code et les modèles sont disponibles à l’adresse suivante : https://github.com/dvlab-research/Mask-Attention-Free-Transformer.


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
Transformateur sans attention masquée pour la segmentation d'instances 3D | Articles | HyperAI