HyperAIHyperAI

Command Palette

Search for a command to run...

Généralisation de domaine par régularisation par information mutuelle avec des modèles pré-entraînés

Junbum Cha Kyungjae Lee Sungrae Park Sanghyuk Chun

Résumé

La généralisation de domaine (domain generalization, DG) vise à apprendre un modèle généralisé pouvant s'appliquer à un domaine cible inconnu, à partir de sources de données limitées. Les approches antérieures à la DG échouent à extraire des représentations invariantes au domaine uniquement à partir des domaines sources, en raison des écarts de domaine importants entre les domaines d'entraînement et de test. À la place, nous reformulons l'objectif de la DG en utilisant l'information mutuelle avec un modèle oracle, c’est-à-dire un modèle généralisé à tout domaine possible. Nous dérivons une borne inférieure variationnelle tractable en approximant le modèle oracle par un modèle pré-entraîné, ce qui donne lieu à une méthode appelée Mutual Information Regularization with Oracle (MIRO). Nos expériences étendues montrent que MIRO améliore significativement les performances hors distribution. De plus, nos expériences d’échelle révèlent que plus le modèle pré-entraîné est grand, plus l’amélioration des performances apportée par MIRO est importante. Le code source est disponible à l’adresse suivante : https://github.com/kakaobrain/miro.


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