Apprentissage sans simulation des dynamiques de population via résidus lagrangiens de Wasserstein

Les auteurs proposent Double-Stitch, une méthode sans simulation pour apprendre la mécanique lagrangienne dans l'espace de Wasserstein, en pénalisant le résidu de l'équation du mouvement le long d'un chemin de population appris. Dérivée d'un principe variationnel de Clebsch, la méthode égale ou surpasse les approches de gradient-flow et la méthode WLM basée sur simulation, tout en entraînant 4 à 14 fois plus vite. Testée sur des données synthétiques, de cellules uniques et de tourbillons océaniques, avec implémentation JAX disponible.