HyperAIHyperAI

Command Palette

Search for a command to run...

MTI-Net : Réseaux d'interaction de tâches multi-échelle pour l'apprentissage multi-tâches

Simon Vandenhende Stamatios Georgoulis Luc Van Gool

Résumé

Dans cet article, nous mettons en avant l'importance de prendre en compte les interactions entre tâches à plusieurs échelles lors de la distillation d'informations de tâches dans un cadre d'apprentissage multi-tâches. Contrairement à l'idée reçue, nous démontrons que des tâches présentant une forte affinité à une échelle donnée ne sont pas nécessairement conservées à d'autres échelles, et inversement. Nous proposons une nouvelle architecture, nommée MTI-Net, qui s'appuie sur cette observation en trois points. Premièrement, elle modélise explicitement les interactions entre tâches à chaque échelle grâce à une unité de distillation multi-échelle et multi-modale. Deuxièmement, elle propage les informations de tâches distillées des échelles inférieures vers les échelles supérieures via un module de propagation de caractéristiques. Troisièmement, elle agrège les caractéristiques de tâches affinées issues de toutes les échelles à l'aide d'une unité d'agrégation de caractéristiques afin de produire les prédictions finales par tâche. Des expérimentations étendues sur deux jeux de données de marquage dense multi-tâches montrent que, contrairement aux approches antérieures, notre modèle multi-tâches exploite pleinement le potentiel de l'apprentissage multi-tâches : une empreinte mémoire réduite, un nombre moindre d'opérations calculatoires, et une meilleure performance par rapport à l'apprentissage mono-tâche. Le code est mis à disposition publiquement : https://github.com/SimonVandenhende/Multi-Task-Learning-PyTorch.


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
MTI-Net : Réseaux d'interaction de tâches multi-échelle pour l'apprentissage multi-tâches | Articles | HyperAI