From Fake to Real: Pretraining on Balanced Synthetic Images to Prevent Spurious Correlations in Image Recognition
Le papier propose « From Fake to Real » (FFR), un pipeline d'entraînement en deux étapes qui pré-entraîne les modèles sur des données synthétiques équilibrées pour apprendre des représentations de sous-groupes robustes avant de procéder à un ajustement fin sur des données réelles, évitant ainsi les corrélations spécieuses causées par le mélange de distributions réelles et synthétiques et améliorant considérablement la précision du pire groupe.
Article original sous licence CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Ceci est une explication générée par l'IA de l'article ci-dessous. Elle n'a pas été rédigée ni approuvée par les auteurs. Pour une précision technique, consultez l'article original. Lire la clause de non-responsabilité complète
Imaginez que vous enseigniez à un robot à reconnaître les animaux. Vous lui montrez des milliers de photos, mais il y a un problème sournois : presque chaque photo d'un « Gros Chien » a été prise à l'intérieur d'un salon douillet, tandis que presque chaque photo d'un « Petit Chien » a été prise à l'extérieur, dans un parc. Le robot est intelligent, mais il est aussi un peu efficace. Au lieu d'apprendre à quoi ressemble réellement un chien, il apprend un raccourci : « Si je vois un canapé, c'est un Gros Chien. Si je vois de l'herbe, c'est un Petit Chien. » Dans le monde de l'informatique, on appelle cela une « corrélation spécieuse ». Le robot a appris à se fier à l'arrière-plan plutôt qu'au sujet. C'est un problème majeur car, lorsqu'il rencontre un Gros Chien à l'extérieur, il est confus et échoue. Les scientifiques essaient de corriger cela pour que l'IA puisse voir le monde clairement, et non pas seulement les motifs dans lesquels elle a été piégée.
Pour corriger cela, les chercheurs ont essayé une astuce ingénieuse : ils utilisent l'« IA générative » (comme un artiste numérique) pour créer de nouvelles photos truquées afin d'équilibrer les comptes. S'il n'y a pas assez de photos de Gros Chiens à l'extérieur, l'ordinateur en dessine quelques-unes. L'idée était qu'en mélangeant ces nouvelles photos truquées avec les vraies, le robot finirait par apprendre à ignorer l'arrière-plan. Mais, comme on peut s'y attendre, cette approche de mélange et d'association avait un défaut caché. Le nouvel article, « From Fake to Real », suggère que le simple fait de mélanger des photos réelles et truquées crée en réalité un nouveau type de dépendance. Le robot commence à remarquer que les photos truquées sont légèrement différentes des vraies photos — peut-être que les chiens truqués ont une texture étrange ou que la lumière est un tout petit peu différente. Ainsi, le robot apprend un nouveau raccourci, encore plus sournois : « Si la photo ressemble à une vraie photo, c'est un Petit Chien. Si la photo ressemble à un dessin truqué, c'est un Gros Chien. » Il ne regarde pas du tout le chien ; il regarde si l'image est réelle ou synthétique.
Les auteurs de cet article, Maan Qraitem, Kate Saenko et Bryan A. Plummer, ont réalisé que c'est le fait d'essayer d'enseigner au robot avec des photos réelles et truquées en même temps qui posait problème. Ils ont proposé une nouvelle méthode d'entraînement en deux étapes appelée « From Fake to Real » (FFR). Voyez cela comme l'enseignement d'un examen de mathématiques à un élève. D'abord, vous lui donnez une pile de problèmes d'entraînement parfaitement équilibrés et faciles à comprendre, afin qu'il apprenne les concepts fondamentaux sans aucune confusion. C'est l'étape « Fake » (Faux) : le robot s'entraîne uniquement sur les images générées par ordinateur, qui sont parfaitement équilibrées. Il apprend à quoi ressemble un chien sans les distractions de l'arrière-plan. Une fois que le robot possède une base solide, vous passez à la seconde étape : l'étape « Real » (Réel). Maintenant, vous lui montrez les photos réelles et désordonnées du monde réel. Parce que le robot sait déjà ce qu'il doit chercher grâce à la première étape, il ne se laisse pas piéger par l'arrière-plan ou par le fait que les images soient réelles. Il applique simplement ce qu'il a appris.
Les chercheurs ont testé cette idée sur trois ensembles de données différents, incluant des photos de visages et d'animaux, avec des niveaux de biais allant de modéré à extrêmement sévère (jusqu'à 99,9 % d'un groupe se trouvant dans un cadre spécifique). Ils ont découvert que leur méthode en deux étapes changeait la donne. Alors que les méthodes précédentes, qui mélangeaient les données réelles et truquées, peinaient à mesure que le biais s'accentuait, l'approche « From Fake to Real » restait solide. En fait, elle a amélioré la précision du robot sur les cas les plus difficiles (le « pire groupe ») de près de 20 % par rapport aux meilleures méthodes existantes. En examinant de près ce à quoi le robot prêtait attention, ils ont constaté que les anciennes méthodes étaient toujours distraites par l'arrière-plan ou par le côté « faux » des images, tandis que la nouvelle méthode se concentrait exactement sur le chien ou le visage de la personne.
L'article a également lancé des simulations pour prouver pourquoi l'ancienne méthode ne fonctionnait pas. Ils ont montré que tant que l'on mélange des données réelles et synthétiques, le robot trouvera presque toujours un moyen d'utiliser la différence entre elles comme facteur de distinction. En séparant l'entraînement en deux étapes distinctes, ils ont effectivement empêché le robot de voir la « différence » entre le réel et le faux comme un indice. Les auteurs notent que cette méthode fonctionne bien même lorsqu'elle est combinée à d'autres techniques, mais ils soulignent également une limite : elle repose sur la capacité de l'artiste informatique à dessiner un ensemble d'images truquées parfaitement équilibrées. Si l'artiste se trompe dans l'équilibre lors de la première étape, tout le système en pâtit. Cependant, pour l'instant, cette stratégie « Fake puis Real » offre un moyen étonnamment simple et efficace d'empêcher l'IA d'apprendre les mauvaises leçons, garantissant que lorsqu'elle regarde un chien, elle voit un chien, et non la pièce dans laquelle il se trouve.
Noyé(e) sous les articles dans votre domaine ?
Recevez des digests quotidiens des articles les plus récents correspondant à vos mots-clés de recherche — avec des résumés techniques, dans votre langue.