HyperAIHyperAI

Command Palette

Search for a command to run...

AutoMix : Révéler le Pouvoir du Mixup pour des Classificateurs Plus Puissants

Zicheng Liu Siyuan Li Di Wu Zihan Liu Zhiyuan Chen Lirong Wu Stan Z. Li

Résumé

L’augmentation par mélange de données s’est avérée efficace pour améliorer la capacité de généralisation des réseaux neuronaux profonds. Alors que les méthodes initiales mélangent les échantillons selon des politiques manuelles (par exemple, interpolation linéaire), les approches récentes exploitent des informations de saliency afin d’associer de manière précise les échantillons mélangés aux étiquettes via une optimisation hors ligne complexe. Toutefois, un compromis apparaît entre la précision des politiques de mélange et la complexité d’optimisation. Pour surmonter ce défi, nous proposons un nouveau cadre automatique de mixup, nommé AutoMix, dans lequel la politique de mixup est paramétrée et orientée directement vers l’objectif final de classification. Plus précisément, AutoMix reformule le problème de classification par mixup en deux sous-tâches (génération d’échantillons mélangés et classification par mixup), chacune étant traitée par un sous-réseau dédié, et résout ces deux sous-tâches dans un cadre d’optimisation bi-niveau. Pour la génération, un générateur de mixup léger et apprenable, appelé Mix Block, est conçu pour produire des échantillons mélangés en modélisant les relations entre patches sous la supervision directe des étiquettes correspondantes. Afin de prévenir la dégradation et l’instabilité liées à l’optimisation bi-niveau, nous introduisons également une pipeline à momentum pour entraîner AutoMix de manière end-to-end. Des expériences étendues sur neuf benchmarks d’images démontrent l’avantage d’AutoMix par rapport aux méthodes de pointe dans divers scénarios de classification et tâches descendantes.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp