Stage - Bac+5 - Apprentissage par Renforcement et Modèles de Monde pour les maillages - H/F

En considérant un problème plus simple de maillage, à savoir le lissage qui consiste à déplacer les sommets d’un maillage pour améliorer la qualité géométrique des cellules composant le maillage, l’étudiant(e) aura pour objectif de développer et évaluer une architecture hybride où un agent RL apprend à déplacer les sommets du maillage en s’appuyant sur les prédictions abstraites de Graph-JEPA. L’idée sous-jacente est d’apprendre des représentations latentes relatives à des opérations de maillage sur des cartes combinatoires de dimension 2 et 3.

Les grandes étapes du travail proposé seront :

• Assimiler les concepts suffisants en maillage et comprendre le formalisme des cartes combinatoire.
• Concevoir une architecture JEPA adaptée. Le modèle devra apprendre à prédire l’impact latent d’un déplacement de sommet sans reconstruire le maillage complet.
• Implémenter un agent de RL avec une approche acteur-critique et une fonction de récompense qui sera directement corrélée à des métriques de qualité de mailles (ex : rapport de forme, jacobien des éléments, régularité des angles) en utilisant l’architecture précédemment définie.
• Comparer les résultats obtenus avec des algorithmes de lissage usuels, aussi bien en termes de qualité finale du maillage et de temps de convergence.

Conformément aux engagements pris par le CEA en faveur de l'intégration des personnes handicapées, cet emploi est ouvert à toutes et à tous. Le CEA propose des aménagements et/ou des possibilités d'organisation pour l’inclusion des travailleurs handicapés.

Participant à la protection nationale, une enquête administrative est réalisée pour tous les collaborateurs du CEA afin d'assurer l'intégrité et la sécurité de la nation.

Profil du candidat

Python, Apprentissage par renforcement, apprentissage profond, pytorch

Bac+5

Localisation du poste Site

DAM Île-de-France

Localisation du poste

France, Ile-de-France, Essonne (91)

Ville

Bruyères-le-Châtel