Géométrie des données en diffusion par masquage : des schedules certifiés-optimaux via la complexité de croissance du démasquage
Un nouveau cadre théorique optimise les schedules de diffusion discrète en s'appuyant sur la géométrie des données, avec des gains prouvés en grande dimension.
arXiv cs.AI · cs.LG · cs.CL·Martin J. Wainwright·13 août 2026

Image · Source originale
Les auteurs introduisent la « unmasking growth complexity » (UGC), une mesure de géométrie des données qui contrôle l'erreur de discrétisation KL dans les schémas de diffusion par masquage. Cette approche permet de construire des samplers certifiés-optimaux, atteignant une erreur KL prescrite avec une complexité proche de l'optimal théorique. Les exemples montrent des gains significatifs, de l'ordre de racine(d), par rapport aux schedules classiques non adaptatifs.