Les chercheurs de Vector s'attaquent aux défis concrets de l'IA lors de l'ICML 2025

10 juillet 2025

Rechercher

Des chercheurs de pointe de Vector présentent leurs travaux les plus novateurs lors de la Conférence internationale sur l'apprentissage automatique (ICML) de cette année, qui se tiendra du 13 au 19 juillet 2025 à Vancouver, au Canada, et en ligne. Grâce à plusieurs articles acceptés, les chercheurs de Vector s'attaquent à certains des défis les plus urgents en intelligence artificielle : rendre les systèmes d'IA plus sûrs et plus fiables, et développer de nouveaux outils pour les soins de santé et la surveillance environnementale.

Vous trouverez ci-dessous 45 articles et présentations par affiche acceptés, présentés par les membres du corps professoral de Vector, les professeurs affiliés à Vector et les chercheurs postdoctoraux émérites de Vector. Les articles marqués d'un astérisque (*) sont des articles phares.

Extraction adaptative d'informations latentes à l'aide du langage naturel

Jimmy Wang, Tom Zollo, Richard Zemel (membre du corps professoral de Vector), Hongseok Namkoong

Abstrait

Recueillir des informations pour réduire l'incertitude sur une entité latente est une compétence essentielle dans de nombreux domaines d'application, comme l'évaluation des acquis individuels des élèves, le diagnostic de maladies sous-jacentes ou la compréhension des préférences des utilisateurs. Bien que le langage naturel soit un outil puissant à cette fin, les grands modèles de langage (GML) et les algorithmes d'ajustement existants ne disposent pas de mécanismes permettant de collecter stratégiquement des informations afin d'affiner leur compréhension de l'entité latente. Nous proposons un cadre d'élicitation adaptatif qui réduit activement l'incertitude sur l'entité latente en simulant des réponses contrefactuelles. La modélisation probabiliste d'une entité latente abstraite étant complexe, nous validons et affinons les méthodes de quantification de l'incertitude basées sur les GML en utilisant la perplexité sur des observations futures masquées produites par l'entité latente. Notre cadre permet le développement de stratégies sophistiquées de collecte d'informations, et nous démontrons sa polyvalence par des expériences de sondage d'opinion dynamique et d'évaluation adaptative des élèves.

Résumé

Nous proposons un cadre d'utilisation des modèles linéaires d'apprentissage (MLA) pour poser des questions pertinentes sur des variables et des entités non observables directement. Parmi les applications potentiellement importantes, mentionnons la construction d'un questionnaire de diagnostic dynamique optimisant les renseignements recueillis sur la santé d'un patient ou la génération d'un ensemble personnalisé de questions d'évaluation permettant de mieux cerner les besoins d'apprentissage d'un étudiant.

AutoÉlicit : Utilisation de grands modèles linguistiques pour l’obtention d’informations a priori d’experts en modélisation prédictive

Alexander Capstick, Rahul G. Krishnan (membre du corps professoral de Vector), Payam Barnaghi

Abstrait

Les grands modèles linguistiques (LLM) acquièrent une vaste gamme d'informations dans divers domaines. Cependant, leur complexité de calcul, leur coût et leur manque de transparence limitent souvent leur application directe aux tâches prédictives où la confidentialité et l'interprétabilité sont primordiales. Dans des domaines tels que la santé, la biologie et les finances, les modèles linéaires spécialisés et interprétables conservent une valeur considérable. Dans ces domaines, les données étiquetées peuvent être rares ou coûteuses à obtenir. Les distributions a priori bien spécifiées sur les paramètres du modèle peuvent réduire la complexité d'échantillonnage de l'apprentissage par inférence bayésienne ; cependant, l'obtention de ces distributions a priori auprès d'experts peut s'avérer fastidieuse. Nous introduisons donc AutoElicit pour extraire des connaissances des LLM et construire des distributions a priori pour les modèles prédictifs. Nous montrons que ces distributions a priori sont informatives et peuvent être affinées à l'aide du langage naturel. Nous menons une étude comparative approfondie entre AutoElicit et l'apprentissage en contexte et montrons comment sélectionner le modèle le plus approprié parmi les deux méthodes. Nous constatons qu'AutoElicit produit des distributions a priori qui permettent de réduire considérablement l'erreur par rapport à des distributions a priori non informatives, en utilisant moins d'étiquettes, et surpasse systématiquement l'apprentissage en contexte. Nous démontrons qu'AutoElicit permet d'économiser plus de 6 mois d'efforts d'étiquetage lors de la construction d'un nouveau modèle prédictif des infections urinaires à partir d'enregistrements de capteurs chez des personnes atteintes de démence.

Résumé

Dans ce travail, nous proposons AutoElicit, une méthode utilisant des modèles linguistiques pour faciliter les tâches de modélisation prédictive, notamment dans le domaine de la santé. Plus précisément, nous présentons une méthode permettant d'utiliser des modèles de langage pour extraire les distributions a priori d'experts pour les modèles prédictifs linéaires et montrons comment ces experts peuvent contribuer à ce processus. Nous comparons ensuite les prédictions a posteriori obtenues avec celles issues de l'apprentissage en contexte, où les modèles de langage font directement les prédictions. À partir des données de notre étude sur la démence, nous démontrons qu'AutoElicit permet de gagner plus de six mois de travail d'étiquetage lors de la construction d'un nouveau modèle prédictif des infections urinaires à partir des enregistrements de capteurs des participants.

L'effet papillon : les trajectoires d'entraînement des réseaux neuronaux sont très sensibles aux conditions initiales

Gül Sena Altıntaş, Devin Kwok, Colin Raffel (membre du corps professoral de Vector), David Rolnick

Abstrait

L'entraînement des réseaux de neurones est intrinsèquement sensible à l'initialisation et au hasard induit par la descente de gradient stochastique. Cependant, l'impact de ces effets sur les réseaux obtenus, que ce soit au niveau des poids des modèles ou des fonctions sous-jacentes apprises, reste incertain. Dans ce travail, nous montrons que, durant une phase initiale « chaotique », une perturbation extrêmement faible suffit à faire diverger des trajectoires d'entraînement par ailleurs identiques – un effet qui s'atténue rapidement au cours de l'entraînement. Nous quantifions cette divergence par (i) la distance L² entre les paramètres, (ii) la similarité des vecteurs de paramètres mesurée par alignement de permutations et, surtout, (iii) la barrière de perte lors de l'interpolation entre les réseaux. Nous montrons ainsi comment des perturbations affectant différents hyperparamètres ou réglages fins orientent les trajectoires d'entraînement vers des minima de perte distincts. Nos résultats éclairent la stabilité de l'entraînement des réseaux de neurones et ont des implications pratiques pour les techniques de réglage fin et de fusion de modèles.

Résumé

En raison du bruit, deux réseaux de neurones entraînés à partir d'un même point de départ aléatoire peuvent apprendre plusieurs solutions différentes à un même problème, tandis que les réseaux préentraînés tendent à apprendre la même solution. Nous ignorons cependant quand et comment les réseaux passent de l'apprentissage de solutions différentes à l'apprentissage d'une solution unique. Pour répondre à cette question, nous entraînons des copies jumelles de réseaux de neurones de manière identique, en introduisant une infime modification (perturbation) dans l'une des copies pendant l'entraînement. Nous constatons que pour les réseaux initialisés aléatoirement, même une modification minime (bien inférieure aux effets aléatoires typiques) entraîne l'apprentissage de solutions différentes, tandis que les réseaux pré-entraînés n'apprennent des solutions différentes que lorsque des modifications beaucoup plus importantes que les effets aléatoires sont appliquées. Nos résultats sont importants car nous devons souvent réentraîner et combiner les connaissances de plusieurs grands réseaux (comme les grands modèles de langage). Certaines méthodes étant plus performantes avec des solutions similaires qu'avec des solutions différentes, nous pouvons adapter nos méthodes de réentraînement ou de combinaison de modèles afin de cibler au mieux chaque cas.

Apprentissage de modèles calibrés et sensibles aux valeurs avec des modèles d'environnement probabilistes

Claas Voelcker, Anastasiia Pedan, Arash Ahmadian, Romina Abachi, Igor Gilitschenski (membre affilié de la faculté Vector), Amir-Massoud Farahmand (membre affilié de la faculté Vector)

Abstrait

L'idée d'un apprentissage de modèles tenant compte de la valeur, selon laquelle les modèles doivent produire des estimations de valeur précises, a pris de l'importance dans l'apprentissage par renforcement basé sur des modèles. La perte MuZero, qui pénalise la prédiction de la fonction de valeur d'un modèle par rapport à la fonction de valeur réelle, a été utilisée dans plusieurs travaux empiriques importants. Cependant, l'étude théorique de ses forces et faiblesses demeure limitée. Dans cet article, nous analysons la famille des pertes d'apprentissage de modèles tenant compte de la valeur, dont la perte populaire MuZero. Nous montrons que ces pertes, telles qu'elles sont généralement utilisées, sont des pertes de substitution non calibrées, ce qui signifie qu'elles ne retrouvent pas toujours le modèle et la fonction de valeur corrects. Forts de cette observation, nous proposons des corrections pour résoudre ce problème. De plus, nous étudions l'interaction entre l'étalonnage de la perte, les architectures de modèles latents et les pertes auxiliaires couramment utilisées lors de l'entraînement d'agents de type MuZero. Nous montrons que si les modèles déterministes peuvent suffire à prédire des valeurs précises, l'apprentissage de modèles stochastiques calibrés demeure avantageux.

Résumé

Cet article analyse l'apprentissage de modèles tenant compte de la valeur (VAML), notamment la fonction de perte MuZero, dans le cadre de l'apprentissage par renforcement basé sur un modèle. Les fonctions de perte VAML permettent d'entraîner un modèle à prédire avec précision la valeur d'une action dans chaque état, plutôt que de l'entraîner à prédire les états eux-mêmes avec précision.

Apprentissage par renforcement distributionnel catégoriel avec divergence de Kullback-Leibler : convergence et asymptotique

Tyler Kastner, Mark Rowland, Yunhao Tang, Murat Erdogdu (membre du corps professoral de Vector), Amir-Massoud Farahmand (membre associé du corps professoral de Vector)

Abstrait

Nous étudions le problème de l'apprentissage par renforcement distributionnel à l'aide de paramétrisations catégorielles et d'une perte de divergence de Kullback-Leibler (KL). Les travaux antérieurs analysant l'apprentissage par renforcement distributionnel catégoriel ont utilisé une perte basée sur la distance de Cramér, simplifiant l'analyse mais créant un écart entre la théorie et la pratique. Nous présentons une version préconditionnée de l'algorithme et démontrons sa convergence garantie. Nous dérivons ensuite la variance asymptotique des estimations catégorielles sous différents régimes de taux d'apprentissage, la comparons à celle de l'apprentissage par renforcement classique et analysons l'influence de l'approximation de fonctions linéaires sur ces mises à jour. Finalement, nous validons empiriquement nos résultats et menons une étude comparative des avantages relatifs des pertes KL, ce qui nous permet d'en tirer des enseignements pratiques.

Résumé

Une approche courante de l'apprentissage par renforcement profond consiste à utiliser des pertes de classification pour saisir l'éventail des résultats futurs possibles. Les travaux théoriques antérieurs étudiant cet algorithme modifient la perte utilisée afin de simplifier l'analyse, ce qui crée un décalage entre la théorie et la pratique. Dans ce travail, nous étudions directement ces algorithmes d'apprentissage avec la perte de classification couramment utilisée en pratique : la divergence de Kullback-Leibler (KL). Nous démontrons qu'avec quelques modifications de la dynamique (l'utilisation d'une matrice de préconditionnement), les mises à jour convergent de manière prouvée. Nous étudions également l'efficacité de ces méthodes par rapport à l'apprentissage par renforcement standard et nous prouvons des résultats sur la variance exacte de ces algorithmes à l'approche de la convergence. Tout au long de notre analyse, nous obtenons plusieurs enseignements précieux pour quiconque utilise ces méthodes en pratique, notamment comment modifier le taux d'apprentissage en fonction du nombre d'atomes (un hyperparamètre distinct) et comment le nombre et la position de ces atomes influencent l'erreur commise.

Réseaux neuronaux de graphiques de déplacement

Wei Zhuo, Han Yu, Guang Tan, Xiaoxiao Li (membre du corps professoral de Vector)

Abstrait

Les réseaux de neurones graphiques (GNN) ont démontré une efficacité remarquable dans l'apprentissage à partir de données structurées en graphes. Cependant, leur application aux graphiques orientés (digraphes) présente des défis uniques, principalement dus à l'asymétrie inhérente aux relations entre les nœuds. Les GNN traditionnels excellent dans la capture des relations unidirectionnelles, mais peinent à encoder les dépendances mutuelles entre les chemins, comme les plus courts chemins asymétriques typiques des digraphes. Face à cette lacune, nous introduisons les réseaux de neurones graphiques à temps de trajet (CGNN), une approche qui intègre de manière transparente le temps de trajet de chaque nœud dans le système de transmission de messages. Le principe fondamental des CGNN repose sur une méthode efficace de calcul du temps de trajet, utilisant un laplacien de digraphe nouvellement formulé. Ce temps de trajet est ensuite intégré au processus d'agrégation de voisinage, les contributions des voisins étant pondérées en fonction de leur temps de trajet respectif vers le nœud central de chaque couche. Cela permet aux CGNN de capturer directement les relations asymétriques mutuelles dans les digraphes. De nombreuses expériences confirment les performances supérieures des CGNN. Le code source de CGNN est disponible anonymement ici.

Résumé

De nombreux réseaux de neurones graphiques (GNN) traitent les graphes orientés (digraphes) comme des ensembles d'arêtes à sens unique, ne parvenant ainsi pas à capturer la connectivité asymétrique aller-retour qui détermine l'intensité des interactions entre deux nœuds. Cette limitation est évidente sur les médias sociaux, où un fan peut contacter instantanément une célébrité, mais où les interactions en retour sont rares. Nous introduisons les réseaux de neurones graphiques à trajets (CGNN) pour modéliser explicitement cette asymétrie. CGNN exploite un laplacien de digraphe novateur (DiLap) associé à un recâblage de graphes léger et basé sur les caractéristiques. Cela garantit la parcimonie et l'irréductibilité du graphe, facilitant le calcul efficace des temps de trajet déterministes, définis comme le nombre moyen d'étapes pour un parcours aléatoire d'un nœud à un autre et retour. Ces temps de déplacement servent de pondération pour les messages des voisins, permettant aux nœuds mutuellement accessibles d'exercer une plus grande influence lors de l'agrégation. Le temps de trajet domicile-travail capture naturellement des interactions mutuelles réalistes, comme la dynamique abonnés-célébrités sur les médias sociaux ou le trafic Web bidirectionnel ; par conséquent, CGNN fournit un cadre plus précis, interprétable et largement applicable pour l'apprentissage à partir de réseaux orientés.

Un cadre global pour analyser la convergence d'Adam : combler le fossé avec la SGD

Ruinan Jin, Xiao Li, Yaoliang Yu (membre du corps professoral de Vector), Baoxiang Wang

Abstrait

L'estimation adaptative des moments (Adam) est un algorithme d'optimisation fondamental en apprentissage profond, largement reconnu pour sa flexibilité face aux taux d'apprentissage adaptatifs et son efficacité dans le traitement de données à grande échelle. Cependant, malgré son succès pratique, la compréhension théorique de la convergence d'Adam a été limitée par des hypothèses strictes, telles que la quasi-certitude de la borne inférieure du gradient stochastique ou la borne supérieure du gradient uniforme, plus restrictives que celles généralement requises pour l'analyse de la descente de gradient stochastique (SGD). Dans cet article, nous présentons un cadre novateur et complet pour analyser les propriétés de convergence d'Adam. Ce cadre offre une approche polyvalente pour établir la convergence d'Adam. Plus précisément, nous démontrons qu'Adam atteint la convergence asymptotique (au sens de la dernière itération) au sens quasi-sûr et au sens L₁ sous des hypothèses assouplies, généralement utilisées pour la SGD, à savoir la L-régularité et l'inégalité ABC. Par ailleurs, sous les mêmes hypothèses, nous montrons qu'Adam atteint des bornes de complexité d'échantillonnage non asymptotiques semblables à celles de la SGD.

Résumé

Adam est l'une des méthodes d'optimisation les plus populaires pour l'entraînement des modèles d'apprentissage profond. Son efficacité pratique repose sur sa capacité à ajuster automatiquement sa vitesse d'apprentissage. Cependant, jusqu'à présent, comprendre précisément son fonctionnement nécessitait des hypothèses mathématiques très fortes, souvent irréalistes. Dans cet article, nous présentons un nouveau cadre théorique démontrant qu'Adam peut réussir dans des conditions beaucoup plus souples et réalistes, semblables à celles requises pour analyser l'algorithme plus fondamental SGD (descente de gradient stochastique). Nos résultats montrent qu'Adam est non seulement performant en pratique, mais qu'il bénéficie également de solides garanties théoriques, contribuant ainsi à combler le fossé entre son succès empirique et sa compréhension formelle. Ces travaux pourraient aussi faciliter l'analyse d'autres méthodes d'optimisation similaires.

Gardien confidentiel : Interdiction cryptographique de l’abus de l’abstention de modèles

Stephan Rabanser, Ali Shahin Shamsabadi, Olive Franzese, Xiao Wang, Adrian Weller, Nicolas Papernot (membre du corps professoral de Vector)

Abstrait

Les prédictions prudentes – où un modèle d'apprentissage machine s'abstient en cas d'incertitude – sont cruciales pour limiter les erreurs préjudiciables dans les applications critiques pour la sécurité. Dans ce travail, nous identifions une nouvelle menace : une institution malhonnête peut exploiter ces mécanismes pour discriminer ou refuser injustement des services sous couvert d'incertitude. Nous démontrons la réalité de cette menace en introduisant une attaque induisant l'incertitude, appelée Mirage, qui réduit délibérément la confiance dans des régions d'entrée ciblées, désavantageant ainsi subrepticement certains individus. Entre-temps, Mirage conserve des performances prédictives élevées sur l'ensemble des points de données. Pour contrer cette menace, nous proposons Confidential Guardian, un cadre qui analyse les mesures d'étalonnage sur un ensemble de données de référence afin de détecter toute réduction artificielle de la confiance. De plus, il utilise des preuves à divulgation nulle de connaissance d'inférence vérifiée pour s'assurer que les scores de confiance rapportés proviennent bien du modèle déployé. Cela empêche le fournisseur de fabriquer arbitrairement des valeurs de confiance pour le modèle tout en protégeant les renseignements confidentiels de ce dernier. Nos résultats confirment que Confidential Guardian empêche efficacement le détournement de prédictions prudentes, en fournissant des garanties vérifiables que l'abstention reflète une véritable incertitude du modèle plutôt qu'une intention malveillante.

Résumé

Lorsque les systèmes d'intelligence artificielle (IA) sont incertains, ils choisissent souvent de s'abstenir de toute prédiction. Ce comportement prudent aide à éviter des erreurs préjudiciables dans des domaines critiques tels que la médecine, les finances ou la justice pénale. Mais que se passerait-il si ce mécanisme même, conçu pour garantir la sécurité, était détourné à des fins malveillantes ? Nos travaux révèlent une possibilité inquiétante : une organisation pourrait délibérément faire apparaître son système d'IA comme incertain pour certaines personnes, non pas en raison d'une réelle difficulté de la tâche, mais pour leur refuser discrètement des services tels que des prêts ou des prestations sociales. Nous appelons cette stratégie trompeuse « Mirage », une attaque qui réduit la confiance de l'IA dans des cas spécifiques tout en conservant de bonnes performances globales. Il devient donc difficile pour les observateurs externes de déceler toute activité suspecte. Pour prévenir de tels abus, nous présentons Confidential Guardian, un nouveau système permettant à des auditeurs indépendants de vérifier si le comportement prudent d'une IA est authentique ou artificiel. Ce système analyse le comportement de l'IA sur des cas de test fiables et le vérifie à l'aide d'une technique garantissant son intégrité, sans pour autant dévoiler le fonctionnement interne du modèle. Nos conclusions mettent en lumière un danger caché dans les systèmes d'IA actuels et offrent une voie vers plus de transparence et d'équité, garantissant que la prudence soit utilisée pour la sécurité et non à des fins discriminatoires.

Modèles de mouvement direct pour l'évaluation des vidéos générées

Kelsey Allen (membre du corps professoral de Vector), Carl Doersch, Guangyao Zhou, Mohammed Suhail, Danny Driess, Ignacio Rocco, Yulia Rubanova, Thomas Kipf, Mehdi SM Sajjadi, Kevin Murphy, Joao Carreira, Sjoerd van Steenkiste

Abstrait

Une limite actuelle des modèles de génération vidéo est qu'ils produisent des images d'apparence plausible, mais un mouvement peu réaliste – un problème mal pris en compte par FVD et d'autres méthodes courantes d'évaluation des vidéos générées. Nous allons plus loin que FVD en développant une métrique qui mesure mieux la plausibilité des interactions entre objets et le mouvement. Notre approche novatrice repose sur l'auto-encodage de trajectoires de points et fournit des caractéristiques de mouvement permettant de comparer les distributions de vidéos (d'une vidéo générée et d'une vidéo de référence, ou jusqu'à deux ensembles de données), ainsi que les erreurs de reconstruction pour évaluer le mouvement de vidéos individuelles. Nous montrons que l'utilisation de trajectoires de points, au lieu de la reconstruction de pixels ou de la reconnaissance d'actions, aboutit à une métrique nettement plus sensible aux distorsions temporelles dans les données synthétiques. Cette mesure permet de prédire avec une meilleure précision que de nombreuses alternatives les évaluations humaines de la cohérence temporelle et du réalisme des vidéos générées par des modèles open source.

Résumé

Les modèles d'intelligence artificielle actuels qui créent des vidéos produisent souvent des images individuelles d'apparence crédible, mais les mouvements des objets dans ces vidéos manquent de réalisme. Les méthodes existantes de contrôle de la qualité vidéo ont du mal à détecter ces mouvements imparfaits et nécessitent généralement l'accès à un ensemble de vidéos plutôt que de pouvoir s'appliquer à une seule. Nous avons créé une nouvelle méthode de mesure de la qualité vidéo qui se concentre spécifiquement sur la fluidité des mouvements et des interactions des objets. Notre méthode fonctionne en suivant des points sur les objets tout au long de la vidéo et en utilisant ces informations pour comprendre le mouvement. Cela nous permet d'évaluer le réalisme des mouvements, même pour des vidéos individuelles. Nous avons constaté que notre nouvelle approche, basée sur le suivi de ces points, est bien plus performante que les autres méthodes pour détecter les mouvements étranges ou artificiels dans les vidéos générées par ordinateur. Elle permet aussi de mieux reproduire ce que l'œil humain considère comme réaliste et cohérent dans les vidéos produites par l'IA. De plus, notre méthode permet de localiser précisément les passages d'une vidéo où le mouvement semble incorrect, ce qui facilite la compréhension des erreurs commises.

Prédiction conditionnelle disparate dans les classificateurs multiclasses

Sivan Sabato (membre du corps professoral de Vector), Eran Treister, Elad Yom-Tov

Abstrait

Nous proposons des méthodes d'audit d'équité des classificateurs multiclasses sous l'hypothèse d'égalité des chances, en estimant l'écart à cette égalité lorsque le classificateur n'est pas parfaitement équitable. Nous généralisons aux classificateurs multiclasses la mesure de prédiction conditionnelle disparate (DCP), initialement proposée par Sabato et Yom-Tov (2020) pour les classificateurs binaires. La DCP est définie comme la fraction de la population pour laquelle le classificateur prédit des probabilités de prédiction conditionnelle différentes de la prédiction de référence commune la plus proche. Nous fournissons de nouvelles méthodes d'optimisation locale pour estimer la DCP multiclasse dans deux cas de figure : l'un où les matrices de confusion conditionnelles de chaque sous-population protégée sont connues, et l'autre où elles ne peuvent être estimées, par exemple en raison de l'inaccessibilité du classificateur ou de l'absence de données individuelles de bonne qualité. Ces méthodes permettent de détecter les classificateurs susceptibles de traiter une fraction significative de la population de manière inéquitable. Des expériences démontrent la précision des méthodes. Le code des expériences est fourni en annexe.

Résumé

De nombreux systèmes d'apprentissage machine prennent des décisions qui ont un impact sur la vie des individus, comme l'octroi de prêts ou la recommandation de traitements médicaux. Lorsque l'accès au système sous-jacent est difficile, il devient plus complexe de vérifier son équité de traitement pour tous les groupes de personnes. De plus, les contrôles d'équité existants ne tiennent souvent pas compte des cas où ces systèmes gèrent plus de deux résultats possibles. Nous proposons de nouvelles méthodes pour vérifier l'équité de ces systèmes décisionnels multiclasses. Nous nous appuyons sur une mesure d'équité appelée Prédiction Conditionnelle Disparate (PCD), qui évalue le nombre de personnes recevant des prédictions s'écartant d'une situation de référence équitable. Nous étendons cette mesure aux systèmes prenant en charge plus de deux résultats et proposons deux méthodes d'estimation de la PCD : l'une pour les cas où nous disposons de données détaillées sur le comportement du système pour différents groupes, et l'autre pour les cas où nous n'avons pas accès au système ou à des données individuelles de haute qualité. Ces outils facilitent la détection des systèmes décisionnels susceptibles de traiter une part importante de la population de manière inéquitable et aident ainsi les organisations et les organismes de réglementation à identifier et à corriger les biais, pour des résultats plus équitables pour tous.

Désapprentissage rapide et précis des données de réglage fin pour les LLM

Andrei Muresanu, Anvith Thudi, Michael Zhang, Nicolas Papernot (membre du corps professoral de Vector)

Abstrait

Les modèles d'apprentissage machine modernes sont coûteux à entraîner, et la difficulté de supprimer a posteriori des données d'entraînement spécifiques suscite une préoccupation croissante. Réaliser un désapprentissage exact dans les pipelines d'apprentissage profond – c'est-à-dire produire des modèles comme si certaines données n'avaient jamais été utilisées pour l'entraînement – ​​reste un problème ouvert. Dans cet article, nous réexaminons le désapprentissage exact en apprentissage profond et montrons que, pour les grands modèles de langage (LLM), il est possible de désapprendre efficacement et exactement les « données de réglage fin » (les données utilisées pour adapter un modèle préentraîné). Ce résultat est basé sur deux observations. Premièrement, on peut utiliser l'apprentissage en contexte pour adapter le LLM à l'ensemble de données de réglage fin, au lieu d'algorithmes basés sur la descente de gradient stochastique (SGD). Deuxièmement, nous montrons qu'un apprentissage en contexte précis peut être réalisé avec l'algorithme k-moyennes quantifié, ce qui permet des opérations de désapprentissage en temps quasi constant. Notre évaluation empirique montre que cette méthode de désapprentissage offre des performances similaires aux alternatives de réglage fin, tout en réduisant considérablement les coûts de désapprentissage. Notre étude souligne également la nécessité de nouvelles mesures du coût du désapprentissage lors de l'adaptation de l'algorithme d'apprentissage afin d'accélérer les opérations de désapprentissage.

Résumé

Après le déploiement d'un modèle, il peut s'avérer nécessaire de « désapprendre » une partie des données d'entraînement initiales. Le désapprentissage précis de ces données est généralement coûteux en apprentissage profond, et nous avons démontré dans cet article qu'il peut être efficace lors de l'adaptation d'un modèle linéaire préentraîné (LLM) à une tâche. Ce résultat découle de l'observation qu'un algorithme d'apprentissage parfois efficace ajoute des exemples d'entraînement en amont de l'instruction fournie au LLM. Nous avons étudié différentes méthodes pour désapprendre cette sélection d'exemples et constaté qu'il est possible de le faire à un coût indépendant de la taille du modèle et de l'ensemble de données. Nous avons également observé que toutes les tentatives précédentes d'accélération du désapprentissage augmentaient le coût d'inférence et avons proposé de nouvelles mesures pour quantifier ce compromis.

FedOne : Apprentissage fédéré efficace en termes de requêtes pour l’apprentissage discret par invites en boîte noire

Ganyu Wang, Jinjie Fang, Maxwell (Juncheng) Yin, Bin Gu, Xi Chen, Boyu Wang (affilié à la faculté Vector), Yi Chang, Charles X. Ling

Abstrait

L'apprentissage discret par invites en boîte noire (BDPL) est une méthode d'optimisation des invites qui optimise les invites discrètes sans accéder aux paramètres ni aux gradients du modèle, rendant ainsi possible l'optimisation des invites sur un modèle de langage étendu (LLM) basé sur le cloud. L'adaptation de l'apprentissage fédéré (FL) au BDPL pourrait améliorer davantage les performances d'optimisation des invites en exploitant des données provenant de sources diverses. Cependant, toutes les recherches antérieures sur l'optimisation fédérée des invites en boîte noire ont négligé le coût important des requêtes associé au service LLM basé sur le cloud. Pour combler cette lacune, nous avons mené une analyse théorique de l'efficacité des requêtes dans le contexte de l'optimisation fédérée des invites en boîte noire. Nos résultats ont révélé que la dégradation de FedAvg pour n'activer qu'un seul client par itération, une stratégie que nous avons appelée FedOne, permettait une efficacité optimale des requêtes dans l'apprentissage fédéré des invites en boîte noire. Forts de cette observation, nous avons proposé le cadre FedOne, une méthode d'apprentissage discret par invites en boîte noire fédérée conçue pour maximiser l'efficacité des requêtes lors des interactions avec des LLM basés sur le cloud. Nous avons mené des expériences numériques sur différents aspects de notre cadre, démontrant une amélioration significative de l'efficacité des requêtes, ce qui concorde avec nos résultats théoriques.

Résumé

Les grands modèles de langage comme ChatGPT sont souvent accessibles via des services payants qui ne permettent pas aux utilisateurs de visualiser ou de modifier leurs composants internes. Pour adapter ces modèles à des tâches spécifiques, les utilisateurs doivent les interroger à plusieurs reprises, ce qui est à la fois coûteux et long. Cet article explore comment plusieurs utilisateurs peuvent collaborer pour peaufiner ces modèles sans partager leurs données, grâce à une méthode appelée apprentissage fédéré. Cependant, dans cette configuration, le coût se multiplie : chaque utilisateur participant doit effectuer de nombreuses requêtes aux modèles de langage, ce qui la rend impraticable. Nous présentons FedOne, une nouvelle approche qui entraîne le modèle en activant un seul utilisateur à la fois. Notre analyse montre que cette configuration est non seulement beaucoup plus efficace pour réduire les requêtes coûteuses, mais elle conserve également d'excellentes performances. Nous avons testé cette idée sur des tâches réelles en utilisant des modèles comme GPT-3.5 et avons démontré que FedOne est à la fois efficace et économique. FedOne permet aux particuliers et aux organisations d'adapter plus facilement et à moindre coût des outils d'IA puissants à leurs besoins.

Correcteurs de Feynman-Kac en diffusion : recuit, guidage et produits d’experts *

Marta Skreta, Tara Akhound-Sadegh, Viktor Ohanesian, Roberto Bondesan, Alan Aspuru-Guzik (membre du corps professoral de Vector), Arnaud Doucet, Rob Brekelmans (boursier postdoctoral distingué de Vector), Alexander Tong, Kirill Neklyudov

Abstrait

Bien que les modèles génératifs basés sur les scores soient privilégiés dans divers domaines, les outils permettant de contrôler rigoureusement leur comportement lors de l'inférence, notamment pour composer plusieurs modèles préentraînés, restent limités. Les méthodes de guidage sans classificateur existantes utilisent une heuristique simple pour combiner les scores conditionnels et inconditionnels afin d'échantillonner approximativement les distributions conditionnelles. Cependant, ces méthodes n'approximent pas les distributions intermédiaires, ce qui nécessite des étapes de « correction » supplémentaires. Dans ce travail, nous proposons une méthode efficace et rigoureuse pour échantillonner une séquence de distributions recuites, moyennées géométriquement ou produit, dérivées de modèles préentraînés basés sur les scores. Nous dérivons un schéma de simulation pondéré, que nous appelons Correcteurs de Feynman-Kac (FKC), basé sur la célèbre formule de Feynman-Kac et prenant en compte les termes des équations aux dérivées partielles (EDP) appropriées. Pour simuler ces EDP, nous proposons des algorithmes de rééchantillonnage Monte Carlo séquentiels (SMC) qui exploitent la mise à l'échelle lors de l'inférence pour améliorer la qualité de l'échantillonnage. Nous démontrons empiriquement l'utilité de nos méthodes en proposant un échantillonnage amorti via un recuit thermique au temps d'inférence, en améliorant la génération de molécules multi-objectifs à l'aide de modèles pré-entraînés et en améliorant le guidage sans classificateur pour la génération de texte en image.

Résumé

Les modèles de diffusion sont des outils puissants pour la génération de données telles que des images, des molécules ou du texte, mais leur processus de génération est généralement difficile à contrôler. Cet article présente une méthode appelée Correcteurs de Feynman-Kac (FKC), qui permet un contrôle précis des données générées par un modèle de diffusion sans nécessiter de réentraînement. FKC fonctionne en ajustant la manière dont les échantillons sont tirés du modèle, selon le cadre de la méthode de Monte Carlo séquentielle et, en particulier, la formule de Feynman-Kac. Cela permet une approche rigoureuse de l'échantillonnage à partir de distributions cibles combinées, telles que des mélanges ou des produits de plusieurs modèles pré-entraînés, ou des distributions cibles dont la température a été modifiée. Nous montrons que FKC améliore l'échantillonnage dans trois contextes : 1. le guidage sans classificateur, largement utilisé pour la génération d'images à partir de texte ; 2. la génération de molécules répondant à plusieurs objectifs (liaison simultanée à deux protéines) ; et 3. l'échantillonnage de systèmes physiques à différentes températures à l'aide d'un modèle entraîné à une seule température. Contrairement aux méthodes traditionnelles, FKC permet un échantillonnage flexible et efficace avec un faible surcoût de calcul. Cela ouvre de nouvelles perspectives d'applications dans les domaines de l'IA, de la découverte de médicaments et des simulations scientifiques.

Des pêcheurs gratuits ? Approximation de la matrice d'information de Fisher par recyclage de l'accumulateur de gradient au carré *

Yu Li, Felix Dangel (boursier postdoctoral distingué de Vector), Derek Tam, Colin Raffel (membre du corps professoral de Vector)

Abstrait

La diagonale de la matrice d'information de Fisher d'un modèle (la « matrice de Fisher ») est fréquemment utilisée pour mesurer la sensibilité des paramètres. Généralement, la matrice de Fisher est estimée en calculant le carré du gradient des sorties du modèle par rapport à ses paramètres, moyenné sur quelques centaines ou milliers d'exemples – un processus coûteux en ressources de calcul. Pendant ce temps, les méthodes de gradient adaptatif, comme l'optimiseur Adam, calculent une moyenne mobile du carré du gradient au cours de l'entraînement. Cet article explore donc la possibilité d'obtenir une approximation de la matrice de Fisher « gratuitement » en réutilisant l'accumulateur du carré du gradient déjà calculé pendant l'entraînement. À travers une série d'expériences couvrant cinq applications de la matrice de Fisher, nous démontrons que le « Squisher » (accumulateur du carré du gradient comme approximation de la matrice de Fisher) offre des performances similaires à celles de la matrice de Fisher, tout en surpassant les méthodes de référence. De plus, on précise les différences exactes entre le Squisher et la matrice de Fisher et on fournit une quantification empirique de leur impact respectif.

Résumé

Comprendre quelles parties d'un réseau de neurones sont les plus importantes (c'est-à-dire quels paramètres sont les plus déterminants) facilite des tâches telles que la fusion de modèles, l'élagage, l'apprentissage par transfert et l'apprentissage continu. Un outil couramment utilisé à cette fin est la diagonale de la matrice d'information de Fisher, qu'on appelle simplement « Fisher ». Cependant, son calcul peut s'avérer coûteux : il nécessite des calculs supplémentaires sur des centaines, voire des milliers d'exemples. Dans cet article, nous cherchons à obtenir une version suffisamment performante de « Fisher » sans ce coût de calcul élevé. Étonnamment, la réponse est oui. Lors de l'entraînement, des optimiseurs largement utilisés comme Adam suivent déjà une grandeur similaire : le carré du gradient des paramètres du modèle. Cette approximation, que nous appelons Squisher (accumulateur de gradient au carré comme approximation de « Fisher »), ne nécessite aucun calcul ni mémoire supplémentaires et est disponible gratuitement. À travers cinq applications courantes de « Fisher », nous démontrons que Squisher produit des résultats comparables à la méthode originale de Fisher, mais à un coût de calcul nettement inférieur. Il permet ainsi de gagner du temps et des ressources, facilitant l'application de ces techniques à grande échelle.

Galilée : Apprentissage des caractéristiques globales et locales de nombreuses modalités de télédétection

Gabriel Tseng, Anthony Fuller, Marlena Reil, Henry Herzog, Patrick Beukema, Favyen Bastani, James Green, Evan Shelhamer (membre du corps professoral de Vector), Hannah Kerner, David Rolnick

Abstrait

Nous présentons un transformateur multimodal permettant de représenter de nombreuses modalités de télédétection – optique multispectrale, radar à synthèse d'ouverture, altitude, données météorologiques, pseudo-étiquettes, etc. – dans l'espace et le temps. Ces données sont utiles pour diverses applications de télédétection, comme la cartographie des cultures et la détection des inondations. Cependant, l'apprentissage de représentations partagées des données de télédétection est complexe, compte tenu de la diversité des modalités de données pertinentes et de la grande variabilité d'échelle des objets d'intérêt, allant de petites embarcations (1 à 2 pixels, déplacement rapide) à des glaciers (plusieurs milliers de pixels, déplacement lent). Nous proposons un nouvel algorithme d'apprentissage auto-supervisé qui extrait des caractéristiques multi-échelles à partir d'un ensemble flexible de modalités d'entrée grâce à la modélisation par masquage. Nos deux fonctions de perte contrastives, globale et locale, diffèrent par leurs cibles (représentations profondes vs projections superficielles des données d'entrée) et leurs stratégies de masquage (structuré vs non structuré). Notre modèle Galileo est un modèle généraliste unique qui surpasse les modèles spécialisés de pointe pour les images satellites et les séries temporelles de pixels sur onze ensembles de données de référence et pour de multiples applications.

Résumé

 Nous recueillons une foule d'informations sur notre planète grâce aux données de télédétection (observations satellitaires, cartes topographiques, etc.), mais nos connaissances sont plus limitées qu'on ne le pense. L'analyse de ces données par l'apprentissage machine peut nous aider à mieux comprendre l'évolution de notre planète. Nous présentons Galileo, un modèle d'apprentissage machine capable de synthétiser ces données. Ainsi, après un traitement minimal, ses synthèses permettent d'établir des prédictions et des cartes, par exemple des inondations ou des zones agricoles. Pour ce faire, Galileo est alimenté par un ensemble de données incomplet pour un lieu et un moment donnés, et doit reconstruire les données manquantes. En choisissant soigneusement les éléments à reconstruire, nous nous assurons que ses synthèses tiennent compte aussi bien des phénomènes de grande ampleur et à évolution lente (comme les glaciers) que des phénomènes de petite taille et à évolution rapide (comme les bateaux de pêche). Galileo est particulièrement pertinent pour la télédétection en raison de sa capacité à modéliser des données spatialement, temporellement et de types variés (données optiques satellitaires, cartes topographiques, données météorologiques, etc.). Nous avons testé Galileo sur 15 tâches diverses, en le comparant à 11 autres méthodes : Galileo obtient les meilleurs résultats avec un modèle général unique. Cela le rend immédiatement utilisable dans de nombreuses applications existantes.

Cache-cache : Les symétries transformables obscurcissent la netteté et la géométrie riemannienne la trouve *

Marvin F. da Silva, Felix Dangel (boursier postdoctoral distingué de Vector), Sageev Oore (membre affilié de la faculté de Vector)

Abstrait

Le concept de netteté a été appliqué avec succès aux architectures traditionnelles telles que les MLP et les CNN pour prédire leur capacité de généralisation. Cependant, pour les transformateurs, des travaux récents ont mis en évidence une faible corrélation entre la platitude et la généralisation. Nous soutenons que les mesures de netteté existantes sont inadaptées aux transformateurs, car leur mécanisme d'attention présente des symétries beaucoup plus riches, induisant des directions dans l'espace des paramètres le long desquelles le réseau, ou sa fonction de perte, reste identique. Nous postulons que la netteté doit pleinement tenir compte de ces symétries et la redéfinissons donc sur une variété quotient obtenue en éliminant les symétries des transformateurs par un facteur de quotient, levant ainsi leurs ambiguïtés. En nous basant sur des outils de géométrie riemannienne, nous proposons une notion de netteté entièrement générale, exprimée en termes de boule géodésique sur la variété quotient corrigée des symétries. En pratique, il faut se rapprocher des géodésiques. Une approximation au premier ordre permet d'obtenir les mesures de netteté adaptatives existantes, et nous démontrons que l'inclusion de termes d'ordre supérieur est cruciale pour rétablir la corrélation avec la généralisation. Nous présentons des résultats sur des réseaux diagonaux avec des données synthétiques et montrons que notre netteté géodésique révèle une forte corrélation pour les transformateurs du monde réel sur les tâches de classification de texte et d'image.

Résumé

En apprentissage profond, comprendre pourquoi certains réseaux de neurones font de meilleures prédictions que d'autres est un problème important. Une idée courante pour expliquer ce phénomène est celle de la netteté. La netteté s'intéresse à la forme du paysage de perte du réseau, une représentation graphique de la performance du réseau en fonction de petites variations de ses paramètres internes. Généralement, si ce paysage est « plat », cela signifie que les petites modifications n'affectent que très peu les performances et que le modèle est plus susceptible de bien généraliser à des données qu'il n'a jamais vues auparavant. Cette idée fonctionne bien pour les anciens types de réseaux neuronaux comme les MLP (perceptrons multicouches) et les CNN (réseaux neuronaux convolutifs). Mais pour les transformateurs, cette relation n'est plus valable. Les chercheurs ont constaté que la netteté, telle qu'elle est généralement mesurée, ne permet pas de prédire de manière fiable si un transformateur généralisera bien. Nous croyons que le problème ne réside pas dans le concept de netteté en lui-même, mais dans sa mesure pour les transformeurs. Les transformateurs offrent de nombreuses possibilités de modifier leurs paramètres internes sans altérer le comportement du modèle (symétries). Ces symétries faussent les mesures traditionnelles de netteté. En utilisant des outils de géométrie différentielle, nous introduisons une définition plus précise de la netteté qui tient compte de ces symétries, et nous constatons qu'une fois ces symétries corrigées, la netteté reste un concept utile.

Adaptation de domaine graphique améliorée par l'homophilie

Ruiyi Fang, Bingheng Li, Jingyu Zhao, Ruizhi Pu, QIUHAO Zeng, Gezheng Xu, Charles X. Ling, Boyu Wang (affilié à la faculté Vector)

Abstrait

L'adaptation de domaine de graphes (GDA) transfère des connaissances de graphes sources étiquetés vers des graphes cibles non étiquetés, palliant ainsi la rareté des étiquettes. Dans cet article, nous soulignons l'importance de l'homophilie des graphes, un facteur crucial pour l'alignement de domaines de graphes, qui a pourtant été longtemps négligé dans les approches existantes. Plus précisément, notre analyse révèle d'abord l'existence de disparités d'homophilie dans les ensembles de données de référence. De plus, nous montrons que ces disparités dégradent les performances de la GDA, tant du point de vue empirique que théorique, ce qui souligne l'importance de l'alignement par homophilie dans la GDA. Forts de ce constat, nous proposons un nouvel algorithme d'alignement par homophilie qui utilise des filtres mixtes pour lisser les signaux des graphes, capturant et atténuant ainsi efficacement les disparités d'homophilie entre les graphes. Les résultats expérimentaux obtenus sur divers ensembles de données de référence confirment l'efficacité de notre méthode.

Résumé

Les graphiques sont des outils puissants pour représenter des relations complexes, comme les interactions sur les médias sociaux ou la circulation de l'information sur Internet. Dans de nombreuses situations concrètes, des renseignements utiles (comme des étiquettes ou des catégories) existent pour un graphe, mais pas pour un autre. L'adaptation de domaine de graphes (GDA) permet de transférer ces connaissances d'un graphe à un autre, ce qui représente un gain de temps et de ressources. Nos recherches ont révélé qu'un facteur clé, l'homophilie (la tendance des nœuds connectés à être similaires), diffère souvent d'un graphe à l'autre, et que ce décalage peut nuire aux performances de la GDA. Étonnamment, ce problème a été largement ignoré jusqu'à présent. Nous avons étudié l'impact de ces différences sur les résultats et constaté que l'harmonisation de cette similarité entre les graphiques peut faire toute la différence. Nous avons mis au point une nouvelle méthode pour atténuer ces différences et améliorer le transfert de connaissances entre les graphes. Notre approche fonctionne bien sur divers ensembles de données et s'avère prometteuse pour l'amélioration de l'apprentissage à partir de données graphiques dans de nombreuses applications, des systèmes de recommandation aux médias sociaux.

Amélioration de la robustesse aux changements de sous-population par exploration heuristique de sous-espaces avec diversification accrue

Nguyen Nhat Minh To, Paul Wilson, Viet Nguyen, Mohamed Harmanani, Michael Cooper, Fahimeh Fooladgar, Purang Abolmaesumi, Parvin Mousavi (membre du corps professoral de Vector), Rahul G. Krishnan (membre du corps professoral de Vector)

Abstrait

Les décalages de sous-population, caractérisés par des disparités dans la distribution des sous-populations entre les ensembles de données d'entraînement et de cible, peuvent dégrader significativement la performance du modèle d'apprentissage automatique. Les solutions actuelles à ces décalages consistent souvent à modifier la minimisation du risque empirique par des stratégies de pondération afin d'améliorer la généralisation entre les sous-populations. Cette stratégie repose fréquemment sur des hypothèses concernant le nombre et la nature des sous-populations, ainsi que sur des annotations relatives à l'appartenance à ces sous-populations, des informations qui sont souvent indisponibles pour de nombreux ensembles de données réels. Nous proposons une nouvelle solution pour explorer heuristiquement le sous-espace des caractéristiques : nous entraînons de nombreux classificateurs tout en imposant une diversification afin de favoriser la découverte et la classification correctes de nouvelles sous-populations, sans nécessiter de connaissances préalables à leur sujet. Étant donné un réseau d'extraction de caractéristiques, nous remplaçons sa couche linéaire standard par un mélange de classificateurs prototypiques, chaque membre étant entraîné à classifier les données en se concentrant sur des caractéristiques et des échantillons différents de ceux des autres membres. Nous démontrons que notre solution surpasse l'état de l'art précédent en termes de précision du pire groupe sur la plupart des benchmarks en utilisant des évaluations empiriques sur neuf ensembles de données réelles couvrant divers domaines et types de changements de sous-population (le code est disponible à l'adresse https://anonymous.4open.science/r/prototypical_ensembles-BCB3).

Résumé

Les modèles d'apprentissage machine rencontrent souvent des difficultés lorsqu'ils sont confrontés à des situations légèrement différentes de celles sur lesquelles ils ont été entraînés. Ce problème est majeur lorsque les données incluent des sous-groupes cachés, tels que différents types de personnes, d'environnements ou de pathologies, qui ne sont pas représentés de manière égale. Par exemple, un modèle formé principalement sur des patients en bonne santé pourrait ne pas fonctionner avec des personnes atteintes de maladies rares. Nos recherches introduisent une nouvelle technique, appelée Ensemble Prototypique Diversifié (EPD), pour résoudre ce problème. Au lieu d'utiliser un seul modèle, on crée un groupe de classificateurs simples, appelés prototypes. Chacun apprend à se concentrer sur différents schémas ou caractéristiques des données. Nous encourageons ces classificateurs à être aussi différents que possible, afin qu'ensemble, ils puissent couvrir une plus grande variété de sous-groupes cachés. Le principal avantage de l'EPD est qu'il ne nécessite aucune connaissance préalable des sous-groupes. Il peut les découvrir et s'y adapter automatiquement en utilisant seulement les données elles-mêmes. Cela le rend particulièrement utile dans des situations réelles où ces étiquettes de sous-groupes sont manquantes ou difficiles à définir. Sur neuf ensembles de données complexes, notre méthode surpasse systématiquement les solutions existantes et contribue à rendre les modèles d'apprentissage automatique plus équitables et fiables lorsqu'ils sont utilisés auprès de populations diverses.

Pondération gaussienne sensible à l'interaction pour l'apprentissage fédéré en grappes

Alessandro Licciardi, Davide Leo, Eros Fanì, Barbara Caputo, Marco Ciccone (boursier postdoctoral distingué Vector)

Abstrait

L'apprentissage fédéré (FL) est apparu comme un paradigme décentralisé pour la formation de modèles tout en préservant la confidentialité. Cependant, le FL conventionnel est confronté à l'hétérogénéité des données et au déséquilibre des classes, ce qui dégrade la performance du modèle. Le FL par regroupement équilibre la personnalisation et l'entraînement décentralisé en regroupant les clients présentant des distributions de données analogues, permettant ainsi une meilleure précision tout en respectant les contraintes de confidentialité. Cette approche atténue efficacement l'impact négatif de l'hétérogénéité dans le FL. Dans ce travail, nous proposons une nouvelle méthode de regroupement pour le FL, **FedGWC** (Federated Gaussian Weighting Clustering), qui regroupe les clients en fonction de la distribution de leurs données, permettant ainsi l'entraînement d'un modèle plus robuste et personnalisé sur les clusters identifiés. **FedGWC** identifie les grappes homogènes en transformant les pertes empiriques individuelles pour modéliser les interactions des clients à l'aide d'un mécanisme de récompense gaussien. De plus, nous introduisons le *Score ajusté de Wasserstein*, une nouvelle mesure de regroupement pour le FL permettant d'évaluer la cohésion des grappes par rapport à la distribution des classes individuelles. Nos expériences sur des ensembles de données de référence montrent que **FedGWC** surpasse les algorithmes FL existants en termes de qualité de grappe et de précision de classification, validant ainsi l'efficacité de notre approche.

Résumé

L'entraînement des modèles d'IA nécessite généralement la centralisation de vastes quantités de données, ce qui soulève des problèmes de confidentialité. L'apprentissage fédéré (FL) offre une solution en permettant aux appareils périphériques ou aux institutions – tels que les téléphones intelligents et les hôpitaux – d'entraîner en collaboration un modèle partagé sans envoyer leurs données privées à un serveur central. Cependant, les données réelles sont souvent hétérogènes : différents appareils peuvent contenir des types de données très divers, ou certaines catégories de données peuvent être rares sur certains appareils tandis que d'autres sont courantes. Cette hétérogénéité des données rend difficile l'obtention de bonnes performances des modèles FL sur tous les appareils. Nos travaux présentent FedGWC, une nouvelle méthode pour rendre l'entraînement FL plus efficace. Au lieu d'imposer l'entraînement d'un seul modèle à tous les appareils, FedGWC regroupe les appareils ayant des caractéristiques de données similaires en grappes, permettant à chaque grappe d'entraîner son modèle spécialisé, bien mieux adapté aux données de son groupe. Imaginez l'organisation d'un groupe d'étude : au lieu d'étudier tous le même sujet général, des groupes plus restreints se forment pour se concentrer sur des sujets spécifiques pour lesquels ils ont tous besoin d'aide. FedGWC réalise cela en analysant la capacité du modèle de chaque appareil à apprendre à partir de ses propres données sans examiner les données elles-mêmes. Nous avons aussi mis au point une nouvelle méthode pour évaluer la qualité de ces grappes, notamment lorsque certaines catégories de données sont beaucoup plus rares que d'autres. Nos expériences montrent que FedGWC améliore considérablement la précision des modèles dans les configurations FL, en particulier lorsque les données sont diversifiées et inégalement réparties. Cela signifie que nous pouvons concevoir des applications d'IA plus performantes et personnalisées tout en préservant la confidentialité des renseignements sensibles.

Les modèles de langage peuvent compléter mot pour mot des textes sur lesquels ils n'ont pas été explicitement entraînés .

Ken Ziyu Liu, Christopher A. Choquette Choo, Matthew Jagielski, Peter Kairouz, Sanmi Koyejo, Nicolas Papernot (affilié à la faculté Vector), Percy Liang

Abstrait

An important question today is whether a given text was used to train a large language model (LLM). A \emph{completion} test is often employed: check if the LLM completes a sufficiently complex text. But, we require a ground-truth definition of membership; most commonly, it is defined as a member based on the \ngram overlap between the target text and any text in the dataset. In this work, we demonstrate that this $n$-gram based membership definition can be effectively gamed. We study scenarios where sequences are \emph{non-members} for a given $n$ and we find that completion tests still succeed. We find many natural cases of this by retraining LLMs after removing all training samples that were completed: these cases include exact duplicates, near-duplicates, and even short overlaps; they showcase that it is difficult to find a single viable choice of $n$. Using these insights, we design adversarial datasets that can cause any target sequences to be completed without containing it, for any reasonable choice of $n$. Our findings highlight the inadequacy of $n$-gram membership, suggesting membership definitions fail to account for auxiliary information available to the training algorithm.

Résumé

Que signifie exactement « inclusion dans l’ensemble d’entraînement » dans le cadre des modèles linguistiques ? De nombreux travaux – recherches, politiques et même litige – convergent implicitement vers des définitions basées sur le chevauchement des n-grammes (sous-chaînes). Autrement dit, un texte est considéré comme « membre » de l’ensemble d’entraînement si une partie de ce texte (n-gramme) s’y trouve. Cet article présente deux expériences qui démontrent les limites fondamentales de toutes les définitions d'appartenance basées sur les n-grammes. Nous posons deux questions dans le contexte de la complétion de texte (verbatim) avec un modèle de langage : 1. **Suppression :** peut-on *empêcher* la génération littérale d’un texte en supprimant tous ses n-grammes et en réentraînant le modèle ? La réponse est non ! De nombreux textes supprimés peuvent encore être générés littéralement par le modèle de langage réentraîné. 2. **Ajout :** peut-on *provoquer* la génération littérale d’un texte en entraînant le modèle sur des textes sans chevauchement de n-grammes ? La réponse est oui ! Et ça ne nécessite que quelques ajustements progressifs. Le message clé de ce travail est que l'appartenance aux données dans les modèles linéaires latents (LLM) s'étend au-delà de l'appartenance à un ensemble de textes dans l'ensemble de données brut ; elle englobe également les voisinages de données (« appartenance souple ») grâce à la généralisation des LLM, la provenance des données, le prétraitement et d’autres informations auxiliaires auxquelles l’algorithme d’apprentissage accède tout au long du pipeline d’apprentissage automatique. De nombreux sous-domaines, tels que le droit d’auteur, le désapprentissage, l’inférence d’appartenance et la transparence des données, nécessitent une définition de l’appartenance, et nos travaux montrent que des notions d’appartenance trop simplistes entravent les progrès dans ces domaines.

Dernière étape pour la modélisation des séries temporelles asynchrones

Shubham Gupta, Thibaut Durand, Graham Taylor (membre du corps professoral de Vector), Lilian Bialokozowicz

Abstrait

Nous présentons une nouvelle conception d'invites pour les grands modèles de langage (LLM), adaptée aux **séries temporelles asynchrones**. Contrairement aux séries temporelles classiques, dont les valeurs sont réparties à intervalles réguliers, les séries temporelles asynchrones sont constituées d'événements se produisant à intervalles irréguliers, chacun étant décrit en langage naturel. Notre approche exploite efficacement la richesse du langage naturel des descriptions d'événements, permettant aux LLM de tirer parti de leur vaste connaissance du monde pour raisonner dans différents domaines et pour diverses tâches. Cela nous permet d'étendre la portée de l'analyse des séries temporelles asynchrones au-delà de la prévision, pour inclure des tâches telles que la détection d'anomalies et l'imputation de données. Nous introduisons également le **Suggestion stochastique douce**, un nouveau mécanisme d'ajustement des invites qui améliore significativement les performances du modèle, surpassant les méthodes d'ajustement existantes telles que QLORA. Grâce à de nombreuses expériences sur des ensembles de données réels, nous démontrons que notre approche atteint des performances de pointe pour différentes tâches et différents ensembles de données.

Résumé

La plupart des systèmes d'IA analysent des données arrivant à intervalles réguliers, comme les cours boursiers quotidiens ou les relevés de température horaires. Maintenant, de nombreux événements du monde réel surviennent de manière imprévisible – urgences médicales, publications sur les médias sociaux, pannes d'équipement – ​​et sont décrits en langage naturel plutôt que par de simples chiffres. Les méthodes traditionnelles ont du mal à traiter ces données de « séries temporelles asynchrones » parce qu'elles ne peuvent pas gérer simultanément l'irrégularité du rythme et la richesse des descriptions textuelles. Nous avons développé LASTS, une nouvelle approche qui utilise des modèles de langage étendus (LLM) pour analyser ces séquences d'événements irrégulières. Au lieu de contraindre les événements à des catégories rigides, notre méthode préserve leurs descriptions en langage naturel, permettant ainsi à l'IA d'exploiter sa compréhension du langage et sa connaissance du monde. Nous avons aussi créé « Stochastic Soft Prompting », une technique d'ajustement fin qui permet aux LLM de mieux comprendre nos données spécifiques que les autres techniques d'ajustement fin classiques. Notre approche surpasse nettement les méthodes existantes sur de nombreux ensembles de données réels. Elle rend ainsi l'analyse sophistiquée des séries temporelles plus accessible et pourrait améliorer les applications dans les domaines de la surveillance des soins de santé, de l'analyse financière et de la compréhension des réseaux sociaux, aidant les organisations à mieux anticiper et réagir aux événements irréguliers mais importants.

Utilisation de problèmes mathématiques en ligne de niveau olympiade pour la formation et l'évaluation résistante à la contamination des LLM

Seyed Mohammad Sadegh Mahdavi, Muchen Li, Kaiwen Liu, Christos Thrampoulidis, Leonid Sigal (membre du corps professoral de Vector), Renjie Liao (membre du corps professoral de Vector)

Abstrait

Les progrès réalisés dans le domaine des grands modèles linguistiques (LLM) ont suscité un vif intérêt pour leur capacité à résoudre des problèmes mathématiques de niveau olympique. Cependant, l'entraînement et l'évaluation de ces modèles sont limités par la taille et la qualité des ensembles de données disponibles, car la création de données à grande échelle pour des problèmes aussi complexes exige un effort considérable de la part d'experts. De plus, les ensembles de données de référence actuels sont sujets à contamination, ce qui rend les évaluations peu fiables. Dans cet article, nous présentons un pipeline automatisé qui exploite les riches ressources du forum Art of Problem Solving (AoPS), qui propose principalement des problèmes de niveau olympique et des solutions issues de la communauté. À l'aide de LLM open source, nous développons une méthode pour extraire des paires questions-réponses du forum, ce qui nous permet de constituer **AoPS-Instruct**, un jeu de données de plus de 600 000 paires QA de haute qualité. Nos expériences démontrent que l'ajustement précis des LLM sur AoPS-Instruct améliore leurs capacités de raisonnement sur différents ensembles de données de référence. De plus, nous avons mis en place un pipeline automatisé qui introduit **LiveAoPSBench**, un ensemble d'évaluation évolutif horodaté, dérivé des données les plus récentes du forum, fournissant ainsi un banc d'essai résistant à la contamination pour évaluer les performances des modèles de raisonnement logique (LLM). Notamment, nous observons un déclin significatif des performances des LLM au fil du temps, ce qui suggère que leur succès sur les exemples plus anciens pourrait provenir d'une exposition préalable à l'entraînement plutôt que d'une véritable capacité de raisonnement. Nos travaux présentent une approche évolutive pour la création et la maintenance d'ensembles de données à grande échelle et de haute qualité pour le raisonnement mathématique avancé, offrant des informations précieuses sur les capacités et les limites des LLM dans ce domaine.

Résumé

La plupart des modèles de raisonnement logique (LLM) existants ont du mal à résoudre des problèmes mathématiques complexes, car les données d'entraînement de haute qualité pour les questions de niveau Olympiades sont rares et les benchmarks existants incluent souvent des problèmes déjà rencontrés par les modèles lors du pré-entraînement, ce qui rend les évaluations peu fiables. Pour remédier à ce problème, nous avons conçu un pipeline automatisé qui explore le forum Art of Problem Solving à la recherche de problèmes authentiques de niveau compétition et de solutions proposées par la communauté, puis utilise des LLM à code source ouvert pour extraire et nettoyer plus de 600 000 paires questions-réponses, créant ainsi l'ensemble de données AoPS-Instruct. Nous avons également développé LiveAoPSBench, un ensemble d'évaluation évolutif alimenté par les messages les plus récents du forum, qui filtre tout chevauchement avec les données antérieures afin d'éviter toute contamination. En peaufinant différents LLM sur AoPS-Instruct, nous avons observé des améliorations significatives de leur capacité à résoudre des problèmes mathématiques difficiles. De plus, le suivi des performances sur LiveAoPSBench a révélé que de nombreux modèles obtiennent de moins bons résultats sur les questions les plus récentes, ce qui indique que leurs succès passés étaient souvent dus à la familiarisation avec des problèmes similaires lors du pré-entraînement plutôt qu'à de véritables aptitudes au raisonnement. Ce travail propose une méthode évolutive pour générer et maintenir de vastes ensembles de données fiables pour le raisonnement mathématique avancé, permettant ainsi aux chercheurs de mieux comprendre et d'exploiter pleinement le potentiel des modèles de raisonnement logique dans ce domaine.

Exploiter la confidentialité par instance pour le désapprentissage automatique

Naz Sepahvand, Anvith Thudi, Berivan Isik, Ashmita Bhattacharyya, Nicolas Papernot (membre du corps professoral vectoriel), Eleni Triantafillou, Daniel Roy (membre du corps professoral vecteur), Gintare Karolina Dziugaite

Abstrait

Nous présentons une approche rigoureuse, instance par instance, pour quantifier la difficulté du désapprentissage par ajustement fin. Nous commençons par peaufiner l'analyse de la descente de gradient bruitée pour le désapprentissage (Chien et al., 2024), obtenant un meilleur compromis utilité-désapprentissage en remplaçant les bornes de perte de confidentialité dans le pire des cas par des pertes de confidentialité par instance (Thudi et al., 2024), chacune bornant la divergence (de Rényi) par rapport au réentraînement sans point de données individuel. Pour démontrer l'applicabilité pratique de notre théorie, nous présentons des résultats empiriques montrant que nos prédictions théoriques se vérifient aussi bien pour la dynamique de Langevin à gradient stochastique (SGLD) que pour l'ajustement fin standard sans bruit explicite. Nous démontrons également que les pertes de confidentialité par instance sont fortement corrélées à plusieurs mesures de difficulté des données existantes, tout en identifiant des groupes de points de données plus difficiles, et nous introduisons de nouvelles méthodes d'évaluation basées sur des seuils de perte. Dans l'ensemble, nos résultats fournissent une base pour des stratégies de désapprentissage plus efficaces et adaptatives, adaptées aux propriétés uniques de chaque point de données.

Résumé

Dans certains cas, comme le respect de la législation ou la corruption des données d'entraînement, un programme d'entraînement doit « oublier » une partie de son ensemble de données. Nous établissons qu'une mesure, dérivée des statistiques recueillies pendant l'entraînement, permet de prédire la difficulté à oublier un point de données. Théoriquement, nous démontrons que cette métrique fournit une borne supérieure au nombre d'itérations de descente de gradient nécessaires pour oublier un point de données. Empiriquement, on constate, pour différentes configurations d'entraînement, que cette métrique classe précisément les points de données selon le nombre d'itérations de descente de gradient requises pour les oublier. De plus, on observe que notre mesure permet de détecter les points de données les plus difficiles à oublier, comparativement aux méthodes précédentes d'identification des points de données difficiles.

MedRAX : Agent de raisonnement médical pour les radiographies thoraciques

Adibvafa Fallahpour, Jun Ma, Alif Munim, Hongwei Lyu, Bo Wang (membre du corps professoral de Vector)

Abstrait

En tant que pierre angulaire de l'imagerie diagnostique, la radiographie thoracique (RT) joue un rôle essentiel dans la prise de décisions cruciales pour la gestion des maladies et les soins aux patients. Bien que les innovations récentes aient permis le développement de modèles spécialisés pour diverses tâches d'interprétation des RT, ces solutions fonctionnent souvent de manière isolée, ce qui limite leur utilité pratique en clinique. Nous présentons MedRAX, le premier agent d'IA polyvalent qui intègre de manière transparente des outils d'analyse de RT de pointe et de vastes modèles de langage multimodaux dans un cadre unifié. MedRAX exploite dynamiquement ces modèles pour répondre à des requêtes médicales complexes sans nécessiter d'entraînement supplémentaire. Afin d'évaluer rigoureusement ses capacités, nous présentons ChestAgentBench, un banc d'essai complet contenant 2 500 requêtes médicales complexes réparties en 7 catégories. Nos expériences démontrent que MedRAX atteint des performances de pointe par rapport aux modèles open source et propriétaires, ce qui représente une avancée significative vers le déploiement pratique de systèmes automatisés d'interprétation des RT. Les données et le code seront accessibles au public à l'adresse https://medrax25.github.io.

Résumé

L'interprétation des radiographies thoraciques est une tâche essentielle mais exigeante en médecine. Les outils d'intelligence artificielle (IA) existants fonctionnent souvent de manière autonome, ce qui limite leur intégration dans des flux de travail cliniques complets. De plus, les modèles d'IA généralistes actuels, malgré leurs progrès, ne fournissent pas toujours les capacités d'analyse multi-étapes ni la transparence des processus décisionnels nécessaires au diagnostic médical. Nous avons développé MedRAX, un cadre d'IA conçu pour pallier ces limites dans l'analyse des radiographies thoraciques. MedRAX fonctionne en coordonnant une suite d'outils d'IA spécialisés, chacun étant compétent dans des tâches spécifiques telles que la détection de maladies, l'identification et le contourage des structures anatomiques, ou la réponse à des questions détaillées basées sur l'image. Le système sélectionne et séquence dynamiquement ces outils, intégrant leurs résultats pour répondre à des requêtes médicales complexes sans nécessiter de réentraînement du cadre de base lors de l'ajout ou de la modification d'outils. Cette approche permet à MedRAX d'offrir des analyses de radiographies thoraciques plus précises, détaillées et interprétables que les méthodes existantes, ce qui représente une avancée significative vers l'application pratique de l'IA en radiologie. Ce système vise à améliorer l'efficacité du diagnostic, à réduire les risques d'erreur et à accroître la clarté des analyses issues de l'IA, soutenant ainsi les professionnels de la santé et améliorant potentiellement la prise en charge des patients grâce à une assistance par IA plus performante.

MixMin : Recherche de mélanges de données par minimisation convexe

Anvith Thudi, Evianne Rovers, Yangjun Ruan, Tristan Thrush, Chris Maddison (membre du corps professoral de Vector)

Abstrait

Les pipelines d'apprentissage automatique modernes combinent et mélangent de plus en plus de données provenant de sources diverses et hétérogènes, par exemple pour le préentraînement de grands modèles de langage. Cependant, trouver le mélange de données optimal demeure un problème complexe et ouvert. Nous formalisons ce problème de mélange de données comme un objectif à deux niveaux : le meilleur mélange est celui qui permet d'obtenir le meilleur modèle pour un objectif en aval. Malheureusement, cet objectif est généralement insoluble. Dans cet article, nous observons que l'objectif de mélange de données à deux niveaux devient convexe à mesure que notre classe de modèles s'agrandit. Nous développons et étudions une approche basée sur le gradient pour optimiser cet objectif convexe, que nous appelons MixMin, et nous la testons sur des tâches de modélisation linguistique et de chimie. MixMin s'est avéré être la seule méthode pour améliorer uniformément le mélange de données dans toutes nos expériences. Avec MixMin, nous avons amélioré le mélange de données en utilisant moins de 0,2 % de puissance de calcul supplémentaire pour un modèle Pythia de 410 millions de dollars entraîné sur 8,2 milliards de jetons, ce qui a permis d'obtenir une amélioration relative de 1 à 5 % de la vraisemblance logarithmique négative sur PIQA, ARC Easy, SciQ et OpenWebMath. De manière cruciale, nous avons constaté que les mélanges MixMin pour les petits modèles amélioraient l'entraînement des grands modèles, ce qui suggère que les mélanges MixMin pourraient être invariants d'échelle. Lors du mélange de données de bioessais pour entraîner un modèle XGBoost, nous avons observé des améliorations des scores de précision moyens de 0,03 à 0,15.

Résumé

Un apprentissage machine performant exige un ensemble de données pertinent pour la tâche à accomplir. Face à de nombreuses sources de données, la constitution d'un ensemble de données de qualité constitue un problème d'optimisation complexe. Dans cet article, nous démontrons que cette optimisation peut être simplifiée en entraînant préalablement un modèle (peu coûteux) sur chacune de nos sources de données. Nous proposons ainsi une méthode pour créer de meilleurs ensembles de données, ce qui permet d'améliorer les performances en modélisation du langage et en chimie. Nos travaux ouvrent la voie à la recherche d'ensembles de données utiles pour les tâches généralement caractérisées par une rareté des données.

Optimisation budgétaire multi-sessions pour l'apprentissage fédéré basé sur les enchères en direct

Xiaoli Tang, Han Yu, Zengxiang Li, Xiaoxiao Li (membre du corps professoral de Vector)

Abstrait

L'apprentissage fédéré basé sur les enchères (AFL) est devenu un domaine de recherche important ces dernières années. Les stratégies courantes pour les consommateurs de données (CD) d'apprentissage fédéré supposent que l'ensemble des propriétaires de données (PD) requis pour une tâche d'apprentissage fédéré doit être réuni avant le début de l'entraînement. En pratique, un CD peut déclencher le processus d'entraînement à plusieurs reprises. Les PD peuvent ainsi être recrutés progressivement au cours de plusieurs séances d'entraînement du modèle. Les stratégies d'enchères existantes pour les CD d'AFL ne sont pas conçues pour gérer de tels scénarios. Par conséquent, le problème de l'AFL multi-sessions reste ouvert. Afin d'y remédier, nous proposons la stratégie d'optimisation budgétaire multi-sessions pour l'apprentissage fédéré basé sur les enchères (MBOS-AFL). Basée sur l'apprentissage par renforcement hiérarchique, MBOS-AFL optimise conjointement le rythme budgétaire inter-sessions et les enchères intra-sessions pour les CD d'AFL, dans le but de maximiser l'utilité totale. De nombreuses expériences menées sur six ensembles de données de référence montrent qu'elle surpasse significativement sept approches de pointe. En moyenne, la méthode \methodname{} atteint une utilité supérieure de 12,28 %, une acquisition de données accrue de 14,52 % via les enchères pour un budget donné, et une précision de test supérieure de 1,23 % pour le modèle FL résultant, par rapport à la meilleure référence. À notre connaissance, il s'agit de la première méthode d'aide à la décision pour l'optimisation budgétaire avec gestion du rythme budgétaire conçue pour les centres de données dans le cadre d'un apprentissage fluide basé sur des enchères multi-sessions.

Résumé

L'apprentissage fédéré basé sur les enchères (AFL) est devenu un domaine de recherche important ces dernières années. Les stratégies courantes pour les consommateurs de données (CD) d'apprentissage fédéré supposent que l'ensemble des propriétaires de données (PD) requis pour une tâche d'apprentissage fédéré doit être réuni avant le début de l'entraînement. En pratique, un CD peut déclencher le processus d'entraînement à plusieurs reprises. Les PD peuvent ainsi être recrutés progressivement au cours de plusieurs séances d'entraînement du modèle. Les stratégies d'enchères existantes pour les CD d'AFL ne sont pas conçues pour gérer de tels scénarios. Par conséquent, le problème de l'AFL multi-sessions reste ouvert. Afin d'y remédier, nous proposons la stratégie d'optimisation budgétaire multi-sessions pour l'apprentissage fédéré basé sur les enchères (MBOS-AFL). Basée sur l'apprentissage par renforcement hiérarchique, MBOS-AFL optimise conjointement le rythme budgétaire inter-sessions et les enchères intra-sessions pour les CD d'AFL, dans le but de maximiser l'utilité totale. De nombreuses expériences menées sur six ensembles de données de référence montrent qu'elle surpasse significativement sept approches de pointe. En moyenne, MBOS-AFL atteint une utilité supérieure de 12,28 %, une acquisition de données accrue de 14,52 % via les enchères pour un budget donné et une précision de test supérieure de 1,23 % pour le modèle FL résultant, par rapport à la meilleure solution de référence. À notre connaissance, il s'agit de la première méthode d'aide à la décision pour l'optimisation budgétaire avec gestion du rythme budgétaire conçue pour les centres de données dans le cadre d'un AFL multisession.

De l'importance de la gaussianisation des représentations

Daniel Eftekhari, Vardan Papyan (Affilié à la Faculté Vector)

Abstrait

La distribution normale joue un rôle central en théorie de l'information : elle représente à la fois la meilleure distribution pour le signal et la pire pour le bruit, possède la plus grande capacité de représentation de toutes les distributions et offre une équivalence entre la non-corrélation et l'indépendance pour les distributions conjointes. La prise en compte de la moyenne et de la variance des activations à travers les différentes couches des réseaux de neurones profonds a grandement facilité leur apprentissage, mais il est rare de trouver une indication précise de la distribution que devraient adopter ces activations, ni de la manière de l'obtenir. Motivés par les propriétés informationnelles de la distribution normale, nous abordons cette question et présentons simultanément la normalisation par la normalité : une nouvelle couche de normalisation qui favorise la normalité des représentations des caractéristiques des réseaux de neurones grâce à la transformation de puissance et utilise un bruit gaussien additif pendant l'apprentissage. Nos expériences démontrent de manière exhaustive l'efficacité de la normalisation, en ce qui concerne ses performances de généralisation sur un ensemble de combinaisons de modèles et de jeux de données largement utilisés, ses solides performances sur divers facteurs de variation courants tels que la largeur du modèle, la profondeur et la taille du mini-lot d'entraînement, son adéquation à une utilisation partout où les couches de normalisation existantes sont conventionnellement utilisées, et comme moyen d'améliorer la robustesse du modèle face aux perturbations aléatoires.

Résumé

La réussite de l'entraînement des réseaux de neurones profonds dépend fortement de la représentation des données lors de leur traitement à travers les différentes couches du réseau. Jusqu'à présent, le contrôle de la moyenne et de la dispersion de ces représentations constituait la principale approche pour optimiser l'entraînement des réseaux de neurones. Dans ce travail, nous avons par ailleurs défini une distribution spécifique que les représentations des réseaux de neurones devraient suivre et l'avons concrétisée grâce à une nouvelle couche que nous avons développée. Nos expériences et analyses ont démontré de manière exhaustive l'efficacité de cette nouvelle couche.

Sur l'apprentissage des classes de distribution avec des adversaires adaptatifs

Tosca Lechner (boursière postdoctorale distinguée de Vector), Alex Bie, Gautam Kamath (membre du corps professoral de Vector)

Abstrait

Nous étudions la question de l'apprentissage des classes de distributions en présence d'adversaires adaptatifs, c'est-à-dire d'adversaires capables d'intercepter les échantillons demandés par un apprenant et d'y apporter des manipulations en connaissant parfaitement ces échantillons avant de les lui transmettre. Ceci contraste avec les adversaires inconscients, qui ne peuvent modifier que la distribution sous-jacente d'où proviennent les échantillons, mais pas leur nature i.i.d. Nous formulons une notion générale d'apprentissage par rapport aux adversaires adaptatifs, en tenant compte du budget de l'adversaire. Nous montrons que l'apprentissage par rapport aux adversaires adaptatifs additifs est une condition strictement plus forte que l'apprentissage par rapport aux adversaires inconscients additifs.

Résumé

La généralisation à partir des données d'entraînement est fondamentale pour la plupart des processus d'apprentissage automatique. Souvent, ces données sont supposées être générées directement à partir du phénomène étudié. Dans nos travaux, nous étudions le cas où un adversaire manipule les données d'entraînement avant même que l'apprenant n'y ait accès. Nous étudions les adversaires adaptatifs, qui ont accès à l'intégralité des données d'entraînement et peuvent donc les manipuler en toute connaissance de cause. Nous les opposons aux adversaires inconscients, qui ne connaissent que le processus de génération des données, mais pas les données d'entraînement elles-mêmes. Nous démontrons que les adversaires adaptatifs peuvent être strictement plus puissants que les adversaires inconscients. Plus précisément, nous étudions les adversaires additifs, capables d'ajouter des points de données, et les adversaires soustractifs, capables d'en supprimer. Nous soulignons une distinction entre les adversaires additifs adaptatifs et les adversaires adaptatifs inconscients. Ainsi, nous montrons que, dans certaines situations, l'ajout de points de données en connaissant un échantillon peut gravement nuire à l'apprentissage, tandis que des manipulations additives similaires sur le processus de génération des données n'auront qu'un impact limité.

Optimisation des distributions de bruit pour la confidentialité différentielle

Atefeh Gilani, Felipe Gomez, Shahab Asoodeh (affilié à la faculté Vector), Flavio Calmon, Oliver Kosut, Lalitha Sankar

Abstrait

Nous proposons un cadre d'optimisation unifié pour la conception de distributions de bruit continues et discrètes garantissant la confidentialité différentielle (DP) en minimisant la DP de R'enyi, une variante de la DP, sous une contrainte de coût. La DP de R'enyi a l'avantage de permettre, en considérant différentes valeurs du paramètre de R'enyi $\alpha$, d'adapter l'optimisation à un nombre quelconque de compositions. Pour résoudre le problème d'optimisation, on le ramène à une formulation convexe de dimension finie et on effectue une descente de gradient préconditionnée. Les distributions de bruit obtenues sont ensuite comparées à leurs homologues gaussiennes et laplaciennes. Les résultats numériques démontrent que nos distributions optimisées sont systématiquement meilleures, avec des améliorations significatives des garanties de DP(\varepsilon, \delta)$ dans les régimes de composition modérés, par rapport aux distributions gaussiennes et laplaciennes de même variance.

Résumé

La protection des données sensibles est un enjeu majeur à l'ère du Big Data. La confidentialité différentielle (DP) est une méthode courante pour assurer la confidentialité en ajoutant du bruit aléatoire aux données, ce qui rend l'identification des individus difficile. Cependant, le choix du type de bruit est crucial : un bruit excessif peut nuire à la précision des données, tandis qu'un bruit insuffisant peut compromettre la protection de la vie privée. Dans ce travail, nous présentons une nouvelle méthode pour déterminer la distribution de bruit optimale pour un niveau de confidentialité donné. Notre méthode améliore la précision des résultats tout en respectant des normes de confidentialité strictes. Nous démontrons que notre bruit optimisé est plus performant que les types de bruit couramment utilisés, tels que le bruit gaussien ou laplacien, sur différents ensembles de données et pour divers paramètres de confidentialité. Cette approche peut contribuer à rendre l'apprentissage automatique respectueux de la vie privée plus fiable et efficace dans les applications concrètes.

PANDAS : Amélioration du jailbreak à plusieurs reprises grâce à l’affirmation positive, la démonstration négative et l’échantillonnage adaptatif *

Avery Ma, Yangchen Pan, Amir-massoud Farahmand (Affilié à la faculté Vector)

Abstrait

Le jailbreak par « many-shot » contourne l'alignement de sécurité des grands modèles de langage en exploitant leur capacité à traiter de longues séquences d'entrée. Pour ce faire, la requête malveillante est précédée de centaines de dialogues fictifs entre l'utilisateur et le modèle. Ces échanges sont tirés aléatoirement d'un ensemble de questions et réponses malveillantes, donnant l'illusion que le modèle a déjà exécuté des instructions malveillantes. Dans cet article, nous présentons PANDAS : une technique hybride qui améliore le jailbreak par « many-shot » en modifiant ces dialogues fictifs à l'aide d'affirmations positives, de démonstrations négatives et d'une méthode d'échantillonnage adaptatif optimisée, adaptée au sujet de la requête. De nombreuses expériences menées sur AdvBench et HarmBench, avec des modèles de langage de pointe, démontrent que PANDAS surpasse de manière significative les méthodes de référence dans les scénarios à contexte long. Grâce à une analyse de l'attention, nous expliquons comment les vulnérabilités liées au contexte long sont exploitées et comment PANDAS améliore encore le jailbreak par « tir à l'arc ».

Résumé

Il est possible de duper les grands modèles de langage et de les amener à produire des résultats malveillants en les surchargeant de longues conversations factices. Ces conversations sont conçues pour faire croire que le modèle a déjà exécuté des instructions dangereuses à de nombreuses reprises. Dans cet article, nous présentons PANDAS, une technique qui améliore ce type d'attaque en modifiant les conversations factices par l'ajout de phrases d'affirmation positives, de démonstrations négatives et d'une sélection de contenu plus ciblée. Les résultats obtenus sur des modèles open source de pointe montrent que PANDAS est plus efficace que les méthodes précédentes pour provoquer des résultats malveillants. Nous analysons aussi les résultats intermédiaires des modèles afin de comprendre l'effet de PANDAS.

Position : Au-delà de l'assistance – Repenser les juristes spécialisés en droit comme co-créateurs éthiques et adaptatifs dans le domaine des soins de santé mentale

Abeer Badawi, Md Tahmid Rahman Laskar, Jimmy Huang, Shaina Raza (scientifique en apprentissage automatique appliqué vectoriel), Elham Dolatabadi (affilié à la faculté Vector)

Abstrait

Ce document de position plaide en faveur d'une transformation fondamentale de l'intégration des grands modèles linguistiques (GML) dans le domaine des soins de santé mentale. Nous défendons leur rôle de co-créateurs plutôt que de simples outils d'assistance. Bien que les GML aient le potentiel d'améliorer l'accessibilité, la personnalisation et l'intervention en situation de crise, leur adoption demeure limitée par des préoccupations liées aux biais, à l'évaluation, à la dépendance excessive, à la déshumanisation et aux incertitudes réglementaires. Pour relever ces défis, nous proposons deux approches structurées : les lignes directrices de mise en œuvre SAFE-I pour un déploiement éthique et responsable, et le cadre d'évaluation HAAS-E pour une évaluation multidimensionnelle et centrée sur l'humain. SAFE-I fournit un cadre pour la gouvernance des données, l'ingénierie adaptative des modèles et l'intégration en situation réelle, garantissant ainsi la conformité des GML aux normes cliniques et éthiques. HAAS-E introduit des indicateurs d'évaluation qui vont au-delà de la simple précision technique pour mesurer la fiabilité, l'empathie, la sensibilité culturelle et l'applicabilité. Nous demandons l'adoption de ces approches structurées afin d'établir un modèle responsable et évolutif pour le soutien en santé mentale basé sur les GML, garantissant que l'IA complète – et non remplace – l'expertise humaine.

Résumé

Et si l'IA pouvait être votre alliée, et non votre remplaçante, pour offrir des soins de santé mentale empreints de compassion ? Alors que la génération née avec le numérique se tourne vers des outils comme ChatGPT pour tout, des devoirs scolaires aux conseils d'orientation professionnelle, il ne faudra pas longtemps avant qu'elle ne s'appuie sur l'IA pour obtenir un soutien émotionnel et psychologique. La question n'est plus de savoir si les assistants numériques personnels (ANP) ont leur place en santé mentale, mais comment ils peuvent y contribuer de manière sécuritaire, éthique et pertinente. Cet article soutient que les ANP sont prêts à faire bien plus qu'automatiser des tâches lorsqu'ils sont conçus dans le respect des principes éthiques et de sécurité. Ces outils peuvent aider à alléger la charge de travail des équipes surchargées, offrir un accompagnement personnalisé et offrir un soutien opportun. Cependant, les enjeux sont considérables : sans garanties adéquates, les ANP peuvent causer de graves préjudices, diffuser des biais et de la désinformation, ou inciter les utilisateurs à accorder une confiance excessive à leurs réponses. La mise en place de garanties robustes est essentielle pour assurer la sécurité, la fiabilité et la conformité de ces outils aux normes éthiques. Pour concrétiser cette vision, notre position propose deux cadres : SAFE-i, qui soutient une conception et un déploiement responsables grâce à trois piliers : des fondements éthiques des données, l’ingénierie des modèles et l’intégration au monde réel ; et HAAS-e, qui propose un cadre d’évaluation centré sur l’humain, articulé autour de quatre dimensions essentielles : la fiabilité, l’équité, l’empathie et l’applicabilité. Ce cadre introduit des indicateurs tels que le score d’empathie contextuelle (CES), l’indice de sensibilité culturelle (CSI), le score de pertinence de la personnalisation (PAS) et l’évaluation de l’applicabilité et de la sécurité (ASA). Ensemble, ces outils offrent une feuille de route pratique pour aligner les systèmes d’IA sur les valeurs humaines, les objectifs cliniques et les divers contextes culturels, permettant ainsi aux professionnels de la santé mentale de collaborer avec des IA adaptatives, éthiques et empathiques.

Position : L'humanité est confrontée à un risque existentiel lié à une perte graduelle de pouvoir.

Jan Kulveit, Raymond Douglas, Nora Ammann, Deger Turan, David Krueger, David Duvenaud (membre du corps professoral de Vector)

Abstrait

Cet article examine les risques systémiques posés par les progrès graduels de l'intelligence artificielle, en développant le concept de « dépossession progressive du pouvoir », par opposition aux scénarios de prise de contrôle brutale généralement abordés dans le domaine de la sécurité de l'IA. Nous analysons comment même des améliorations progressives des capacités de l'IA peuvent saper l'influence humaine sur les systèmes à grande échelle dont dépend la société, notamment l'économie, la culture et les États-nations. À mesure que l'IA remplace de plus en plus le travail et la cognition humains dans ces domaines, elle peut affaiblir à la fois les mécanismes de contrôle humain explicites (comme le vote et le choix du consommateur) et les alignements implicites avec les préférences humaines qui découlent souvent de la dépendance des systèmes sociétaux à l'égard de la participation humaine pour fonctionner. De plus, les systèmes d'IA peuvent amplifier les désalignements existants avec les préférences humaines en optimisant davantage ces systèmes. Ces distorsions entre les domaines peuvent se renforcer mutuellement : le pouvoir économique façonne les récits culturels et les décisions politiques, tandis que les changements culturels modifient les comportements économiques et politiques. Nous soutenons que cette dynamique pourrait mener à une perte pratiquement irréversible d'influence humaine sur les systèmes sociétaux cruciaux, précipitant une catastrophe existentielle par la dépossession permanente de l'humanité de son pouvoir. Cette analyse suggère la nécessité de recherches techniques et d'approches de gouvernance qui abordent spécifiquement le risque d'érosion progressive de l'influence humaine au sein des systèmes sociétaux interconnectés.

Résumé

Les scénarios de risques liés à l'IA décrivent généralement une perte de contrôle humaine relativement soudaine au profit des IA, qui prennent le dessus sur les individus et les institutions humaines, en raison d'une augmentation soudaine des capacités de l'IA ou d'une trahison coordonnée. Cependant, nous soutenons que même une augmentation progressive des capacités de l'IA, sans aucune recherche de pouvoir concertée, représente un risque important de dépossession humaine. Cette perte d'influence humaine sera principalement due à l'émergence d'alternatives automatisées plus compétitives aux humains dans presque toutes les fonctions sociales, telles que le travail économique, la prise de décision, la création artistique et même les relations humaines. Une perte graduelle du contrôle de notre propre civilisation peut sembler improbable. Les bouleversements technologiques n'ont-ils pas généralement amélioré le bien-être collectif de l'humanité ? Nous affirmons que l'alignement des systèmes sociaux sur les intérêts humains n'a été stable que grâce à la nécessité de la participation humaine pour la prospérité des économies, des États et des cultures. Une fois cette participation humaine supplantée par des alternatives automatisées plus compétitives, les incitations à la croissance de nos institutions ne seront plus liées à la nécessité d'assurer l'épanouissement humain. Les décideurs à tous les niveaux seront bientôt confrontés à des pressions visant à réduire l'intervention humaine sur les marchés du travail, dans les structures de gouvernance, la production culturelle et même les interactions sociales. Ceux qui résisteront à ces pressions seront éventuellement supplantés par ceux qui n'y résisteront pas. Pourtant, l'humanité ne s'apercevra-t-elle pas de ce qui se passe et ne se coordonnera-t-elle pas pour l'enrayer ? Pas nécessairement. Ce qui rend cette transition particulièrement difficile à contrer, c'est que les pressions exercées sur chaque système social se répercutent sur les autres. Par exemple, on pourrait essayer d'utiliser le pouvoir de l'État et les mentalités culturelles pour préserver le pouvoir économique humain. Cependant, les incitations économiques qui poussent les entreprises à remplacer les humains par l'IA les inciteront également à influencer les États et la culture pour soutenir ce changement, en utilisant leur pouvoir économique croissant pour façonner les politiques et l'opinion publique, ce qui leur permettra à son tour d'accumuler un pouvoir économique encore plus grand. Une fois que l'IA commencera à supplanter les humains, les mécanismes de rétroaction existants qui encouragent l'influence et l'épanouissement humains commenceront à s'effondrer. Par exemple, les États financés principalement par les impôts sur les bénéfices de l'IA plutôt que par le travail de leurs citoyens seront peu incités à garantir la représentation des citoyens. Cela pourrait se produire simultanément à l'augmentation de l'influence sans précédent de l'IA sur la culture et les comportements humains, ce qui risquerait de compliquer la coordination entre les individus et de réduire d'autant leur capacité à résister à ces pressions. Nous décrivons plus en détail ces mécanismes et boucles de rétroaction, ainsi que d'autres, dans cet ouvrage. Bien que nous formulions des propositions pour ralentir ou enrayer ce processus et que nous passons en revue les discussions connexes, nous insistons sur le fait qu'aucun plan concret et plausible n'existe pour stopper la perte progressive de pouvoir humain et que les méthodes visant à aligner les systèmes d'IA individuels sur les intentions de leurs concepteurs sont insuffisantes. Cette perte de pouvoir étant globale et permanente, et l'épanouissement humain nécessitant des ressources considérables à l'échelle mondiale, elle pourrait vraisemblablement mener à l'extinction de l'humanité ou à des conséquences similaires.

Position : Pour que les données générées par l'humain soient toujours utilisées en apprentissage machine, il est nécessaire de se concentrer sur les motivations humaines intrinsèques.

Sebastin Santy, Prasanta Bhattacharya, Manoel Ribeiro, Kelsey Allen (membre du corps professoral de Vector), Sewoong Oh

Abstrait

Les progrès de l'IA reposent sur les données générées par l'humain, des plateformes d'annotation à l'Internet en général. Cependant, l'utilisation généralisée de grands modèles linguistiques menace désormais la qualité et l'intégrité de ces données sur ces mêmes plateformes. Nous soutenons que ce problème dépasse le simple défi du filtrage du contenu généré par l'IA : il révèle des failles plus profondes dans la conception des systèmes de collecte de données. Les systèmes existants privilégient souvent la rapidité, l'échelle et l'efficacité au détriment de la motivation humaine intrinsèque, ce qui entraîne une baisse de l'engagement et de la qualité des données. Nous proposons de repenser les systèmes de collecte de données afin de les aligner sur les motivations intrinsèques des contributeurs – plutôt que de s'appuyer uniquement sur des incitations externes – pour garantir un approvisionnement en données de haute qualité à grande échelle, tout en préservant la confiance des contributeurs et leur participation à long terme.

Résumé

Les discussions sur la qualité des données en apprentissage machine se concentrent souvent sur les indicateurs et définitions techniques, négligeant les sources humaines qui les génèrent. Une grande partie des données actuelles proviennent de la participation des utilisateurs sur les plateformes en ligne. Cela nous a amenés à nous interroger : pouvons-nous tirer des leçons sur le maintien de la qualité des données en examinant la manière dont les humains participent à ces plateformes ? Nous analysons le compromis quantité-qualité dans la génération de données à travers le prisme de la motivation humaine. En s'appuyant sur les sciences sociales, nous montrons comment une dépendance excessive aux incitations externes peut nuire à la motivation intrinsèque. Nous proposons un changement de perspective : concevoir des environnements attrayants et incitatifs (par exemple, des jeux en ligne) qui encouragent une participation significative tout en produisant des données de haute qualité. Notre article met en lumière les forces motivationnelles qui sous-tendent la génération de données en ligne pour l’IA/ML et illustre des exemples de systèmes antérieurs qui ont su gérer avec succès le compromis quantité-qualité pour générer des données humaines pertinentes. Nous insistons également sur les considérations de conception essentielles pour construire, à l’avenir, des environnements de collecte de données fiables qui non seulement généreront des données de haute qualité, mais respecteront et soutiendront également les personnes qui y contribuent.

Position : La partie la plus coûteuse d'un LLM *devrait* être ses données de formation

Nikhil Kandpal, Colin Raffel (membre du corps professoral de Vector)

Abstrait

L'entraînement des grands modèles de langage (LLM) est une entreprise de plus en plus coûteuse en raison de l'augmentation des besoins en calcul, en matériel, en énergie et en main-d'œuvre d'ingénierie. Outre les coûts d'entraînement, un coût souvent négligé (et rarement pris en compte) est le travail humain nécessaire à la rédaction des milliards de mots utilisés pour entraîner les LLM les plus performants. Dans ce document de synthèse, nous proposons d'attribuer une valeur monétaire à ce travail et défendons l'idée que la part la plus importante du coût de production d'un LLM *devrait* être la rémunération des producteurs de données d'entraînement. Pour appuyer cette position, nous étudions 64 LLM publiés entre 2016 et 2024, en analysant à la fois le coût d'entraînement des modèles et le coût hypothétique de la création des données d'entraînement. Notre analyse indique que même avec une estimation extrêmement prudente de la rémunération à verser pour le travail humain investi dans la création des données d'entraînement, le coût des ensembles de données d'entraînement de ces modèles est de un à trois ordres de grandeur supérieur au coût d'entraînement des modèles eux-mêmes. Face à l'écart considérable entre la valeur des données d'entraînement et l'absence actuelle de compensation pour leur création, nous mettons en lumière et discutons des pistes de recherche qui pourraient permettre des pratiques plus équitables à l'avenir.

Résumé

Training a modern Large Language Model (LLM) is an incredibly expensive endeavor due to the cost of specialized hardware, energy required to run that hardware, and the enormous engineering labor needed to architect large-scale training systems. However, an often overlooked (and seldom paid) expense is the human labor behind these models’ training data. Every LLM is built on an unfathomable amount of human effort: trillions of carefully written words sourced from books, academic papers, codebases, social media, and more. This position paper aims to assign a monetary value to this labor and argues that the most expensive part of producing an LLM \emph{should} be the compensation provided to training data producers for their work. To support this position, we study 64 LLMs released between 2016 and 2024, estimating what it would cost to pay people to produce their training datasets from scratch. Even under highly conservative estimates of wage rates, the costs of these models’ training datasets are 10-1000 times larger than the costs to train the models themselves, representing a significant financial liability for LLM providers. In the face of the massive gap between the value of training data and the lack of compensation for its creation, we highlight and discuss research directions that could enable fairer practices in the future.

QuEst : Amélioration des estimations des mesures de distribution basées sur les quantiles à l’aide de prédictions de modèles

Zhun Deng, Thomas Zollo, Benjamin Eyre, Amogh Inamdar, David Madras, Richard Zemel (membre du corps professoral de Vector)

Abstrait

À mesure que les modèles d'apprentissage machine gagnent en performance, leurs prédictions sont utilisées pour compléter les données rares ou coûteuses nécessaires à l'estimation de grandeurs importantes. En combinant un petit ensemble de données observées de haute fidélité (mesures réelles) avec un ensemble plus important de données imputées (prédictions du modèle), les praticiens peuvent améliorer la qualité des estimations au-delà de ce que chaque source fournit isolément. Bien que ce paradigme soit prometteur, les cadres existants se concentrent principalement sur l'estimation de moyennes ou de quantiles uniques, ce qui limite leur applicabilité à de nombreux domaines et cas d'utilisation critiques. Pour relever ce défi, nous présentons **QuEst**, un cadre intégrant des données observées et imputées afin d'estimer et de fournir des intervalles de confiance rigoureux pour les mesures de distribution basées sur les quantiles. Ces mesures incluent des mesures de queue comme le CVaR, des segments de population comme les quartiles et d'autres grandeurs clés d'intérêt dans des domaines tels que l'économie, la sociologie, l'éducation et la médecine. QuEst intègre également un algorithme permettant d'estimer ces statistiques pour des mesures et des mesures multidimensionnelles. De plus, nous proposons une nouvelle méthode basée sur une fonction spline pour optimiser notre méthode (ainsi que d'autres méthodes existantes pour ce type d'estimation hybride). Nous démontrons l'utilité de notre cadre à travers des expériences de modélisation économique, de sondage d'opinion et d'auto-évaluation de modèles de langage.

Résumé

Nous présentons QuEst, une méthode permettant de combiner des données réelles observées avec des prédictions de modèles d'apprentissage machine afin d'obtenir de meilleures estimations de grandeurs importantes. Notre cadre est particulièrement utile pour enrichir les résultats expérimentaux dans des domaines tels que l'économie, la sociologie, l'éducation et la médecine, ainsi que pour évaluer les modèles de langage.

Réfléchir puis planifier : planification hors ligne basée sur un modèle à travers une double perspective bayésienne

Jihwan Jeong, Xiaoyu Wang, Jingmin Wang, Scott Sanner (membre affilié de la faculté Vector), Pascal Poupart (membre de la faculté Vector)

Abstrait

L'apprentissage par renforcement hors ligne (RL) est crucial lorsque l'exploration en ligne est coûteuse ou dangereuse, mais il se heurte souvent à une forte incertitude épistémique en raison de la rareté des données. Les méthodes existantes reposent sur des politiques conservatrices fixes, limitant ainsi leur adaptabilité et leur capacité de généralisation. Pour remédier à ce problème, nous proposons Reflect-then-Plan (RefPlan), une nouvelle approche de planification hors ligne basée sur un modèle (MB) doublement bayésien. RefPlan unifie la modélisation de l'incertitude et la planification MB en reformulant la planification comme une estimation bayésienne de la distribution a posteriori. Lors du déploiement, il met à jour une croyance sur la dynamique de l'environnement à l'aide d'observations en temps réel, intégrant ainsi l'incertitude dans la planification MB par marginalisation. Les résultats empiriques obtenus sur des ensembles de données de référence standard montrent que RefPlan améliore considérablement les performances des politiques RL hors ligne conservatrices. En particulier, RefPlan maintient des performances robustes malgré une forte incertitude épistémique et des données limitées, tout en faisant preuve de résilience face à l'évolution de la dynamique de l'environnement, améliorant ainsi la flexibilité, la généralisabilité et la robustesse des politiques apprises hors ligne.

Résumé

Imaginez apprendre à une IA à réaliser une tâche, comme se repérer dans un bâtiment, en utilisant seulement un ensemble fixe d'exemples enregistrés. Face à une situation inédite, l'IA peut être désorientée et prendre de mauvaises décisions, car ses connaissances sont incomplètes. De nombreuses approches existantes rendent l'IA excessivement prudente pour éviter les erreurs, ce qui l'empêche de s'adapter efficacement. Nous présentons une nouvelle méthode, appelée Réflexion-Planification (RefPlan), qui permet à une IA de raisonner intelligemment sur ce qu'elle ignore. Notre méthode fonctionne en deux étapes : * Réflexion : Pendant son fonctionnement, l'IA « réfléchit » continuellement à ses expériences récentes (les actions entreprises et leurs conséquences) afin d'actualiser sa compréhension de l'environnement spécifique dans lequel elle se trouve. * Planification : Lorsqu'elle « planifie » son prochain mouvement, elle ne se base pas sur une prédiction unique et rigide de l'avenir. Au contraire, elle envisage un éventail de scénarios possibles en fonction de son incertitude, ce qui rend sa stratégie plus robuste face à l'imprévu. Nos résultats montrent que cette approche améliore considérablement les performances de l'IA, la rendant plus souple et résiliente, notamment face à des situations inédites, des données limitées ou des conditions changeantes.

Détection des faiblesses du tatouage numérique de texte grâce aux attaques par réécriture d'informations personnelles

Yixin Cheng, Hongcheng Guo, Yangming Li, Leonid Sigal (membre du corps professoral de Vector)

Abstrait

Le tatouage numérique de texte vise à intégrer subtilement des signaux statistiques dans un texte en contrôlant le processus d'échantillonnage du modèle de langage étendu (LLM), permettant ainsi aux détecteurs de tatouage de vérifier que la sortie a bien été générée par le modèle spécifié. La robustesse de ces algorithmes de tatouage est devenue un facteur clé pour évaluer leur efficacité. Les algorithmes actuels intègrent les tatouages ​​dans des jetons à haute entropie afin d'assurer la qualité du texte. Dans cet article, nous démontrons que cette conception, en apparence inoffensive, peut être exploitée par des attaquants, ce qui représente un risque important pour la robustesse du tatouage. Nous introduisons une attaque par paraphrase générique et efficace, l'attaque par réécriture d'auto-information (SIRA), qui exploite cette vulnérabilité en calculant l'auto-information de chaque jeton pour identifier les jetons présentant des motifs potentiels et mener une attaque ciblée. Nos travaux mettent en évidence une vulnérabilité répandue dans les algorithmes de tatouage actuels. Les résultats expérimentaux montrent que SIRA atteint un taux de réussite proche de 100 % sur sept méthodes de tatouage récentes, pour un coût de seulement 0,88 $ par million de jetons. Notre approche ne nécessite aucun accès aux algorithmes de tatouage numérique ni au modèle LLM tatoué et peut être facilement adaptée à n'importe quel modèle LLM, même ceux destinés aux appareils mobiles. Nos résultats soulignent l'urgence de développer des techniques de tatouage numérique plus robustes.

Résumé

Le développement rapide des grands modèles de langage (GML) a suscité des inquiétudes quant à leur utilisation potentiellement abusive, notamment la diffusion de fausses informations et les atteintes à l'intégrité académique. Pour y remédier, le tatouage numérique de texte est apparu comme une solution prometteuse, intégrant subtilement des motifs indétectables dans les textes générés par les GML afin d'en vérifier l'origine. Cependant, l'efficacité de ces tatouages ​​dépend de leur robustesse face aux attaques visant à les supprimer. Les méthodes d'attaque existantes sont souvent inefficaces, non ciblées, gourmandes en ressources et difficilement transposables d'un GML à l'autre. Nos recherches présentent l'attaque par réécriture d'auto-information (SIRA), une nouvelle attaque de paraphrase efficace qui révèle une vulnérabilité fondamentale des algorithmes de tatouage numérique actuels. Nous avons découvert que les techniques de tatouage intègrent des motifs dans des jetons à « haute entropie » — des jetons présentant une forte auto-information en raison de leur imprévisibilité et de leur faible probabilité. SIRA exploite cette propriété en calculant l'auto-information de chaque jeton afin d'identifier et de masquer les jetons susceptibles de contenir un tatouage numérique. Nous utilisons ensuite un modèle linéaire à longue portée (LLM) pour réaliser une tâche ciblée de « remplissage de blanc », en réécrivant le texte masqué tout en préservant son intégrité sémantique. SIRA représente une avancée significative dans la compréhension et l'évaluation de la robustesse du tatouage numérique par LLM. Nos expériences montrent que SIRA atteint un taux de réussite d'attaque proche de 100 % face à sept méthodes de tatouage numérique récentes, pour un coût très faible de 0,88 $ par million de jetons. Cette attaque ne nécessite aucune connaissance préalable de l'algorithme de tatouage numérique ni du LLM utilisé, et elle est hautement transférable, fonctionnant même avec des modèles plus petits, adaptés aux appareils mobiles. En exposant cette vulnérabilité répandue, nos travaux soulignent l'urgence de développer des approches de tatouage numérique plus robustes et adaptatives afin d'assurer la transparence et l'intégrité du contenu généré par l'IA.

Transformateurs auto-supervisés comme algorithmes d'amélioration itératifs des solutions pour la satisfaction des contraintes

Yudong W Xu, Wenhao Li, Scott Sanner (membre associé de la faculté Vector), Elias Khalil (membre associé de la faculté Vector)

Abstrait

Les problèmes de satisfaction des contraintes (CSP) trouvent de nombreuses applications, et leur résolution accélérée par l'apprentissage automatique suscite un vif intérêt. La plupart des approches existantes reposent sur l'apprentissage supervisé à partir de solutions réalisables ou sur l'apprentissage par renforcement, des paradigmes qui nécessitent soit des solutions réalisables pour ces CSP NP-complets, soit des budgets d'entraînement importants et un signal de récompense complexe conçu par des experts. Pour relever ces défis, nous proposons ConsFormer, un cadre auto-supervisé qui exploite un Transformer comme raffineur de solution. ConsFormer construit une solution à un CSP de manière itérative, selon un processus qui imite la recherche locale. Au lieu d'utiliser des solutions réalisables comme données étiquetées, nous concevons des approximations différentiables des contraintes discrètes d'un CSP pour guider l'entraînement du modèle. Notre modèle est entraîné à améliorer les affectations aléatoires lors d'une étape unique, mais est déployé de manière itérative lors des tests, contournant ainsi les limites de l'apprentissage supervisé et par renforcement. Notre méthode peut traiter les CSP hors distribution simplement par des itérations supplémentaires.

Résumé

La résolution de problèmes soumis à des règles et contraintes spécifiques fait partie intégrante de nombreuses tâches du quotidien, qu'il s'agisse de résoudre des Sudokus ou de planifier les horaires des employés. Ces problèmes sont souvent complexes, et même les meilleures méthodes traditionnelles peuvent rencontrer des difficultés lorsque leur taille et leur complexité augmentent. L'intelligence artificielle a été utilisée pour contribuer à une résolution plus efficace de ces problèmes. Cependant, de nombreuses méthodes existantes reposent sur des exemples de solutions satisfaisantes ou nécessitent de nombreux essais et erreurs, ce qui peut s'avérer long et fastidieux. Nous présentons ConsFormer, qui propose une approche différente. Ce logiciel entraîne un modèle d'IA à apporter de petites améliorations à une solution en une seule étape, sans avoir besoin de réponses correctes pendant l'entraînement. Une fois déployé, ConsFormer est utilisé de manière répétée pour effectuer des améliorations continues, en partant d'une première estimation aléatoire et en l'affinant progressivement. ConsFormer fonctionne avec différents types de problèmes et peut gérer des cas plus complexes en multipliant simplement les étapes d'amélioration. Cela en fait un outil prometteur pour la résolution efficace de problèmes complexes de raisonnement par contraintes dans le monde réel.

Entraînement parcimonieux à partir d'une initialisation aléatoire : alignement des masques de billets de loterie à l'aide de la symétrie des poids

Mohammed Adnan, Rohan Jain, Ekansh Sharma, Rahul G. Krishnan (membre du corps professoral de Vector), Yani Ioannou

Abstrait

L'hypothèse du billet de loterie (LTH) suggère l'existence d'un masque et d'un poids LTH clairsemés qui atteignent les mêmes performances de généralisation que le modèle dense, tout en utilisant un nombre de paramètres nettement inférieur. Cependant, la recherche d'une solution LTH est coûteuse en calcul, et le masque de parcimonie LTH ne se généralise pas à d'autres initialisations aléatoires des poids. Des travaux récents ont suggéré que les réseaux de neurones entraînés à partir d'une initialisation aléatoire trouvent des solutions au sein du même bassin d'énergie, à permutation près, et proposent une méthode pour aligner les modèles entraînés au sein du même bassin d'énergie. Nous émettons l'hypothèse que le désalignement des bassins d'énergie est la raison pour laquelle les masques LTH ne se généralisent pas à de nouvelles initialisations aléatoires et proposons de permuter le masque LTH pour l'aligner sur le nouveau bassin d'énergie lors d'un entraînement clairsemé à partir d'une initialisation aléatoire différente. Nous montrons empiriquement une augmentation significative de la généralisation lors de l'entraînement clairsemé à partir d'une initialisation aléatoire avec le masque permuté par rapport à l'utilisation du masque LTH non permuté, sur plusieurs ensembles de données (CIFAR-10/100 et ImageNet) et modèles (VGG11 et ResNet20/50).

Résumé

Les systèmes d'intelligence artificielle (IA) modernes sont incroyablement puissants, mais leur entraînement nécessite souvent une puissance de calcul et une quantité de données considérables. Par conséquent, ils sont coûteux et inaccessibles à de nombreux chercheurs et développeurs. Pour remédier à ce problème, les scientifiques explorent des modèles d'IA « sparses » – des systèmes qui n'utilisent qu'une petite fraction de leurs connexions potentielles – ce qui les rend beaucoup plus efficaces à entraîner et à exécuter. Cependant, un obstacle majeur se pose : un modèle sparse performant avec un point de départ donné échoue souvent lorsque l'entraînement commence à partir d'un autre. Nos recherches ont identifié la cause profonde de ce problème : un désalignement. Imaginez une clé (le modèle sparse) sur une serrure légèrement tournée : elle ne rentre tout simplement pas. Pour résoudre ce problème, nous avons développé une méthode permettant de « réaligner » la structure sparse afin qu'elle corresponde aux schémas d'un nouveau point de départ. Cet ajustement améliore considérablement les performances des modèles sparses entraînés à partir de différents points de départ, les rendant presque aussi efficaces que leurs versions originales. Nos résultats rendent plus facile et plus pratique le développement de systèmes d'IA plus légers et plus efficaces, ouvrant la voie à une accessibilité et à une innovation plus larges dans la recherche en IA.

Déconvolution stochastique avant-arrière : entraînement de modèles de diffusion avec ensembles de données bruités finis

Haoye Lu, Qifan Wu, Yaoliang Yu (membre du corps professoral de Vector)

Abstrait

Les modèles génératifs récents basés sur la diffusion obtiennent des résultats remarquables grâce à leur entraînement sur des ensembles de données massifs. Cependant, cette pratique soulève des inquiétudes quant à la mémorisation et aux risques de violation du droit d'auteur. Une solution proposée consiste à entraîner le modèle exclusivement sur des données bruitées présentant des risques potentiels de violation du droit d'auteur, garantissant ainsi qu'il n'observe jamais le contenu original. Toutefois, à travers le prisme de la théorie de la déconvolution, nous montrons que, bien qu'il soit théoriquement possible d'apprendre la distribution des données à partir d'échantillons bruités, la difficulté pratique de recueillir suffisamment d'échantillons rend cet apprentissage quasi impossible. Pour surmonter cette limite, nous proposons de pré-entraîner le modèle avec une petite fraction de données propres afin de guider le processus de déconvolution. Combinée à notre méthode de déconvolution stochastique avant-arrière (SFBD), cette approche nous permet d'atteindre un FID de 6,31 sur CIFAR-10 avec seulement 4 % d'images propres (et 3,58 avec 10 %). Théoriquement, on prouve que la SFBD guide le modèle vers l'apprentissage de la vraie distribution des données. Ce résultat souligne également l'importance du pré-entraînement sur des données limitées mais propres, ou sur des ensembles de données similaires. Des études empiriques viennent étayer ces résultats et apporter des éclaircissements supplémentaires.

Résumé

Les modèles modernes de génération d'images, tels que ceux utilisés dans les outils artistiques d'IA, sont généralement entraînés sur d'immenses collections d'images. Cependant, cette pratique soulève d'importantes préoccupations : certaines données d'entraînement peuvent être protégées par le droit d'auteur, et les modèles risquent de mémoriser et de reproduire ce contenu de manière trop fidèle. Une solution proposée consiste à entraîner les modèles uniquement sur des versions bruitées (floues ou altérées) des images, en veillant à ce que les originaux ne soient jamais vus directement. Ou, en pratique, nous montrons qu'apprendre à partir de données bruitées seulement est extrêmement difficile ; cela nécessite un nombre d'échantillons excessivement élevé pour être efficace. Dans ce travail, nous nous concentrons sur les modèles de diffusion et démontrons que l'introduction d'une petite fraction de données propres (originales), seulement 4 % ou 10 %, peut faire une différence substantielle. Nous proposons une méthode appelée déconvolution stochastique avant-arrière (SFBD), qui alterne entre le débruitage des échantillons bruités à l'aide du modèle actuel et le réentraînement du modèle avec ces résultats débruités. Ce processus aide le modèle à apprendre graduellement à générer des images réalistes, même lorsque la plupart des données d'entraînement sont bruitées. Nos expériences montrent que SFBD atteint une qualité d'image proche de celle des modèles entraînés sur des ensembles de données parfaitement exempts de défauts, tout en réduisant considérablement les risques juridiques et éthiques. Ces travaux ouvrent la voie à un entraînement plus responsable et efficace des modèles génératifs.

Filtre d'adéquation : un cadre statistique pour l'évaluation de modèles dans des contextes de déploiement réels *

Angéline Pouget, Mohammad Yaghini, Stephan Rabanser, Nicolas Papernot (membre du corps professoral de Vector)

Abstrait

Le déploiement de modèles d'apprentissage automatique dans des domaines critiques pour la sécurité pose un défi majeur : assurer la fiabilité des performances du modèle sur les données utilisateur en aval, sans accès aux étiquettes de vérité terrain pour une validation directe. Nous proposons le filtre de pertinence, un cadre novateur conçu pour détecter la dégradation des performances grâce à des signaux de pertinence – des caractéristiques de sortie du modèle sensibles aux variations de covariables et indicatives d'erreurs de prédiction potentielles. Le filtre de pertinence évalue si la précision du classificateur sur les données utilisateur non étiquetées présente une dégradation significative par rapport à la précision mesurée sur l'ensemble de données de test étiquetées. Plus précisément, il garantit que cette dégradation ne dépasse pas une marge prédéfinie, qui représente la baisse maximale acceptable de précision. Pour obtenir une évaluation fiable du rendement, nous agrégeons les signaux de pertinence pour les données de test et les données utilisateur, puis comparons ces distributions empiriques à l'aide de tests d'hypothèses statistiques, fournissant ainsi des informations sur l'incertitude de la décision. Notre méthode modulaire s'adapte à différents modèles et domaines. Des évaluations empiriques sur différentes tâches de classification démontrent que le filtre de pertinence détecte de manière fiable les écarts de performance dus aux variations de covariables. Cela permet d'atténuer de manière proactive les défaillances potentielles dans les applications critiques.

Résumé

Les modèles d'apprentissage machine apprennent à partir des données pour prendre des décisions, mais il peut être difficile d'assurer leur fiabilité face à des situations réelles et inédites. Cette recherche propose une nouvelle méthode pour vérifier si ces modèles commencent à faire plus d'erreurs avec de nouvelles données, notamment lorsqu'il est difficile de vérifier l'exactitude de leurs décisions. La méthode consiste à examiner des indices subtils dans le comportement du modèle avec des données familières et nouvelles afin de détecter une baisse possible de la qualité de ses décisions. Les expériences ont démontré que cette approche permet de repérer efficacement les difficultés rencontrées par un modèle lorsque les nouvelles informations diffèrent de celles pour lesquelles il a été préparé. Cela contribue à renforcer la confiance dans le bon fonctionnement et la fiabilité de ces modèles d'apprentissage automatique, en particulier dans les applications critiques du quotidien.

Limites théoriques des ensembles à l'ère de la surparamétrisation *

Niclas Dern, John Cunningham, Geoff Pleiss (membre du corps professoral de Vector)

Abstrait

Les ensembles classiques généralisent mieux que n'importe quel modèle à composant unique. En revanche, des études empiriques récentes montrent que les ensembles modernes de réseaux de neurones (surparamétrés) n'offrent pas nécessairement d'avantage intrinsèque en matière de généralisation par rapport à des réseaux de neurones uniques, même de plus grande taille. Cet article clarifie la différence entre les ensembles modernes surparamétrés et leurs homologues classiques sous-paramétrés, en utilisant des ensembles de régresseurs à caractéristiques aléatoires (RF) comme base théorique. Contrairement au régime sous-paramétré, où la mise en ensemble induit généralement une régularisation et améliore la généralisation, nous démontrons, sous des hypothèses minimales, que les ensembles infinis de régresseurs RF surparamétrés deviennent ponctuellement équivalents à des régresseurs RF de largeur infinie (uniques), et que les ensembles de largeur finie convergent rapidement vers des modèles uniques avec le même budget de paramètres. Ces résultats, exacts pour les modèles sans crête et approximatifs pour de faibles pénalités de crête, impliquent que les ensembles surparamétrés et les grands modèles uniques présentent une généralisation quasi identique. Nous caractérisons aussi la variance prédictive entre les membres de l'ensemble, démontrant qu'elle quantifie les effets attendus de l'augmentation de capacité plutôt que de refléter une notion conventionnelle d'incertitude. Nos résultats remettent en question les hypothèses courantes concernant les avantages des ensembles dans des contextes surparamétrés, ce qui incite à reconsidérer dans quelle mesure les intuitions tirées des ensembles sous-paramétrés se transposent aux ensembles profonds et au régime surparamétré.

Résumé

Dans les applications critiques pour la sécurité, comme le diagnostic médical ou les véhicules autonomes, les chercheurs combinent souvent plusieurs modèles d'IA en « ensembles » afin d'améliorer les prédictions – un peu comme consulter un comité plutôt qu'un expert unique. Cette approche a bien fonctionné pour les modèles simples, mais avec les puissants réseaux neuronaux actuels capables de mémoriser des ensembles de données entiers, les ensembles ne produisent souvent pas les avantages escomptés. Nous avons analysé ce phénomène mathématiquement à l'aide de réseaux neuronaux simplifiés. Nous avons découvert que lorsque les modèles sont suffisamment complexes pour mémoriser leurs données d'entraînement, leurs ensembles se comportent de manière très similaire à un seul modèle plus grand. Cela signifie que l'assemblage de grands modèles n'offre que peu d'avantages par rapport à l'entraînement d'un seul modèle plus grand. De plus, nous avons constaté qu'une méthode courante d'estimation de l'incertitude des prédictions d'un ensemble – la mesure du désaccord entre les membres de l'ensemble – manque de fondement théorique dans de tels cas. Nos résultats ne nient pas que les ensembles puissent encore être utiles en pratique, car les grands modèles peuvent, par exemple, être difficiles à entraîner. Cependant, ils mettent en garde contre le fait de considérer les ensembles comme une stratégie simple et fiable pour améliorer les performances par rapport à celles d'un seul grand modèle ou pour évaluer l'incertitude.

Vers une génération de textes guidée par récompense rentable

Ahmad Rashid, Ruotian Wu, Rongqi Fan, Hongliang Li, Agustinus Kristiadi (boursier postdoctoral distingué de Vector), Pascal Poupart (membre du corps professoral de Vector)

Abstrait

La génération de texte guidée par récompense (RGTG) s'est imposée comme une alternative viable à l'apprentissage par renforcement hors ligne à partir de rétroactions humaines (RLHF). Les méthodes RGTG permettent d'aligner des modèles de langage de base sur les préférences humaines sans entraînement supplémentaire, contrairement aux méthodes RLHF (PPO et DPO). Cependant, elles s'appuient sur un modèle de récompense pour évaluer chaque jeton candidat généré par le modèle de langage lors de l'inférence, ce qui engendre une surcharge importante. De plus, ce modèle de récompense est entraîné uniquement pour évaluer des séquences complètes, ce qui peut mener à des choix sous-optimaux pour les séquences partielles. Dans ce travail, nous présentons une nouvelle architecture de modèle de récompense, entraînée, à l'aide d'une fonction de perte de Bradley-Terry, à privilégier l'expansion optimale d'une séquence en un seul appel. Ainsi, un score est généré simultanément pour tous les jetons candidats possibles, ce qui permet une inférence efficace. Nous analysons théoriquement les modèles de récompense RGTG et démontrons que les modèles de récompense de base privilégient des séquences sous-optimales par rapport à notre méthode lors de l'inférence. Empiriquement, notre modèle de récompense mène à une inférence nettement plus rapide, comparativement aux autres méthodes RGTG, avec moins d'appels au modèle de récompense et des performances compétitives par rapport à RGTG et RLHF.

Résumé

Les modèles de langage peuvent-ils s'améliorer grâce à la rétroaction humaine sans réentraînement ? Le réentraînement est coûteux parce qu'il nécessite des ressources de calcul et de l'électricité, et contribue aux émissions de carbone. Des travaux antérieurs ont montré que c'est possible, mais au prix d'un temps de réponse plus long du modèle de langage lors du traitement d'une requête. Nous présentons FaRMA, une méthode qui réduit considérablement ce temps de réponse tout en évitant le réentraînement. De plus, nous illustrons des scénarios où les méthodes précédentes ne parviennent pas à fournir des réponses satisfaisantes et démontrons que FaRMA n'est pas vulnérable à ces situations.

TypyBench : Évaluation de l'inférence de type LLM pour les dépôts Python non typés

Honghua Dong, Jiacheng Yang, Xun Deng, Yuhe Jiang, Gennady Pekhimenko (membre du corps professoral des vecteurs), Fan Long, Xujie Si (affilié du corps professoral des vecteurs)

Abstrait

L'inférence de types pour les langages dynamiques comme Python représente un défi persistant en génie logiciel. Bien que les grands modèles linguistiques (LLM) se soient révélés prometteurs pour la compréhension du code, leurs capacités d'inférence de types restent encore peu explorées. Nous présentons `TypyBench`, un banc d'essai conçu pour évaluer l'inférence de types des LLM sur des dépôts Python entiers. `TypyBench` propose deux nouvelles mesures : `TypeSim`, qui saisit les relations sémantiques nuancées entre les types prédits et les types réels, et `TypeCheck`, qui évalue la cohérence des types entre les bases de code. Notre évaluation de différents LLM sur un ensemble de données soigneusement sélectionné, composé de 50 dépôts Python de haute qualité, révèle que, malgré des scores `TypeSim` satisfaisants, les LLM rencontrent des difficultés avec les types imbriqués complexes et présentent d'importantes erreurs de cohérence des types. Ces résultats suggèrent que les recherches futures devraient se concentrer non plus sur l'amélioration de la similarité des types, mais sur la cohérence au niveau du dépôt. `TypyBench` jette les bases de cette nouvelle approche, en offrant un éclairage sur les performances des modèles face à différentes complexités de types et de contextes d'utilisation.

Résumé

Déterminer les types de données spécifiques utilisés dans les langages de programmation flexibles comme Python peut s'avérer un véritable casse-tête pour les développeurs. Bien que les puissants modèles d'IA, appelés LLM, soient efficaces pour la compréhension du code, leur capacité à gérer cette tâche spécifique à grande échelle restait incertaine. Pour le découvrir, nous avons créé TypyBench, un nouveau test permettant d'évaluer la précision avec laquelle ces IA prédisent les types de données dans des projets logiciels entiers. Nous avons développé deux nouvelles méthodes de mesure de leur rendement : l'une vérifie la proximité sémantique du type prédit avec le type correct, et l'autre la cohérence des prédictions de l'IA dans l'ensemble du code. Nos tests, menés sur 50 projets Python de haute qualité, ont révélé que si les IA sont généralement capables de deviner la signification générale des types, elles commettent souvent des erreurs avec les types plus complexes et créent des incohérences au sein d'un même projet. Cela démontre que les efforts futurs doivent se concentrer sur l'amélioration de la cohérence des prédictions de l'IA, et TypyBench constitue l'outil idéal pour orienter ces recherches.

Auto-encodeurs clairsemés universels : alignement conceptuel intermodèles interprétable

Harrish Thasarathan, Julian Forsyth, Thomas Fel, Matthew Kowal, Konstantinos Derpanis (membre affilié de la faculté Vector)

Abstrait

Nous présentons les Auto-encodeurs Sparses Universels (USAE), un cadre permettant de découvrir et d'aligner des concepts interprétables à travers plusieurs réseaux neuronaux profonds pré-entraînés. Contrairement aux méthodes d'interprétabilité existantes basées sur les concepts, qui se concentrent sur un seul modèle, les USAE apprennent conjointement un espace conceptuel universel capable de reconstruire et d'interpréter simultanément les activations internes de plusieurs modèles. Notre idée principale est d'entraîner un auto-encodeur parcimonieux (SAE) unique et surcomplet, qui ingère les activations de n'importe quel modèle et les décode pour approximer les activations de tout autre modèle considéré. En optimisant un objectif commun, le dictionnaire appris saisit les facteurs de variation communs – les concepts – à travers différentes tâches, architectures et ensembles de données. Nous montrons que les États-Unis découvrent des concepts universels sémantiquement cohérents et importants à travers les modèles de vision ; allant des caractéristiques de bas niveau (par exemple, les couleurs et les textures) aux structures de haut niveau (par exemple, les parties et les objets). Globalement, les États-Unis offrent une nouvelle méthode puissante pour l'analyse inter-modèles interprétable et proposent des applications inédites, telles que la maximisation coordonnée de l'activation, qui ouvrent la voie à une compréhension plus approfondie des systèmes d'IA multi-modèles.

Résumé

Les modèles de vision par ordinateur modernes sont de plus en plus diversifiés, entraînés à l'aide de jeux de données et d'architectures variés pour accomplir des tâches visuelles spécifiques telles que l'estimation de profondeur ou la reconnaissance d'objets. Ces choix de conception déterminent les « concepts » ou caractéristiques visuelles que chaque modèle apprend, de la reconnaissance des contours et des textures à la compréhension des objets et des scènes. Cela soulève une question scientifique fondamentale : ces modèles, malgré leurs différences, convergent-ils vers l'apprentissage des mêmes concepts visuels fondamentaux ? Répondre à cette question est complexe parce que les représentations internes apprises par ces modèles sont encodées de manière à être interprétables directement par l'humain. Nos travaux introduisent les auto-encodeurs épars universels (USAE), afin de créer un espace conceptuel universel et interprétable qui révèle ce que plusieurs modèles de vision apprennent en commun sur le monde visuel. Notre approche nous permet d'identifier les concepts universels les plus importants partagés par les modèles, tout en découvrant des caractéristiques propres à chaque modèle. Cette analyse permet de comprendre quels choix architecturaux et d'entraînement mènent à de meilleures représentations visuelles, et quels concepts semblent être des éléments constitutifs fondamentaux de la compréhension visuelle. Ces travaux contribuent à améliorer notre capacité à comprendre et à comparer la manière dont différents systèmes d'IA perçoivent et traitent l'information visuelle.