HyperAIHyperAI

Command Palette

Search for a command to run...

Normalisation par lots activée sur place pour une formation optimisée en mémoire des DNNs

Samuel Rota Bulò Lorenzo Porzi Peter Kontschieder

Résumé

Dans cette étude, nous présentons la Normalisation par Lots Activée sur Place (InPlace-ABN) - une nouvelle approche visant à réduire drastiquement l'empreinte mémoire d'entraînement des réseaux neuronaux profonds modernes de manière calculatoirement efficace. Notre solution remplace la succession habituellement utilisée de couches de Normalisation par Lots (BatchNorm) + Activation par une seule couche plugin, évitant ainsi des modifications invasives du cadre tout en offrant une applicabilité simple pour les frameworks de deep learning existants. Nous obtenons des économies de mémoire allant jusqu'à 50 % en supprimant les résultats intermédiaires et en récupérant les informations nécessaires lors du passage arrière grâce à l'inversion des résultats stockés lors du passage avant, avec une augmentation mineure (0,8-2 %) du temps de calcul. De plus, nous démontrons comment les approches de checkpointing fréquemment utilisées peuvent être rendues aussi efficaces sur le plan computationnel que l'InPlace-ABN. Dans nos expériences sur la classification d'images, nous montrons des résultats comparables à ceux des approches les plus avancées sur ImageNet-1k. Pour la tâche exigeante en mémoire qu'est la segmentation sémantique, nous rapportons des résultats pour COCO-Stuff, Cityscapes et Mapillary Vistas, obtenant de nouveaux résultats d'état de l'art pour ce dernier sans données d'entraînement supplémentaires mais dans un scénario mono-échelle et mono-modèle. Le code est disponible à l'adresse suivante : https://github.com/mapillary/inplace_abn .


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
Normalisation par lots activée sur place pour une formation optimisée en mémoire des DNNs | Articles | HyperAI