Google parallélise la recherche complexe avec Retrieve-for-Train

Google Research présente Retrieve-for-Train, un cadre conçu pour accélérer la recherche complexe par récupération. Pengcheng Jiang, étudiant chercheur chez Google Research, et Judith Yue Li, ingénieure de recherche senior chez Google Research, sont associés à ces travaux. Leur article « Efficient, Property-Aligned Fan-Out Retrieval via RL-Compiled Diffusion » a été présenté le 15 septembre 2026, lors de l’ICML 2026. Le cadre utilise l’apprentissage par renforcement pour entraîner un modèle de diffusion léger. Il évite ainsi le raisonnement autorégressif pendant l’inférence. La présentation est détaillée dans l’article de Google Research.

Les modèles génériques rencontrent, selon la description des travaux, deux difficultés pour décomposer des requêtes adaptées aux bases de données. La première est l’effondrement paraphrastique, qui produit des formulations similaires. La seconde est la latence autorégressive, liée au traitement séquentiel des étapes de raisonnement. Retrieve-for-Train s’attaque à ces deux points avec une approche entraînée par apprentissage par renforcement. Le cadre vise une génération efficace de plusieurs requêtes à partir d’une requête initiale. Cette opération correspond à un éventail de requêtes exécuté en un seul passage pendant l’inférence. Les travaux sont également référencés sur arXiv.

Retrieve-for-Train prépare plusieurs requêtes

Le modèle de diffusion compte 53,9 millions de paramètres. Il est présenté comme léger dans le cadre Retrieve-for-Train. Son entraînement repose sur l’apprentissage par renforcement, plutôt que sur une exécution autorégressive pendant l’inférence. Cette organisation permet un éventail de requêtes en un seul passage. Le cadre évite également la surcharge liée aux jetons de réflexion utilisés au moment du test. La description associe donc trois éléments précis : une diffusion entraînée par renforcement, une génération en passage unique et l’absence de cette surcharge. L’objectif technique porte sur la récupération et la décomposition de requêtes adaptées aux propriétés recherchées.

L’inférence sans séquence constitue le mécanisme central décrit pour Retrieve-for-Train. Un modèle autorégressif produit habituellement ses sorties étape après étape. Le cadre présenté par Google Research contourne cette phase de raisonnement autorégressif lors de l’inférence. Il permet ainsi un éventail de requêtes en un seul passage. Cette propriété est directement reliée à la réduction de la surcharge associée aux jetons de réflexion au moment du test. Le modèle de diffusion conserve une taille de 53,9 millions de paramètres. Pengcheng Jiang et Judith Yue Li figurent parmi les chercheurs de Google Research associés à cette présentation de l’ICML 2026.

Google Research présente une architecture compacte

La publication de septembre situe ces travaux dans le programme de l’ICML 2026. Le 15 septembre 2026, l’article « Efficient, Property-Aligned Fan-Out Retrieval via RL-Compiled Diffusion » a été présenté. Son titre met en avant une récupération en éventail alignée sur les propriétés, compilée par apprentissage par renforcement. Le cadre Retrieve-for-Train reprend cette logique pour traiter des recherches complexes. Google Research décrit une méthode fondée sur un modèle de diffusion de 53,9 millions de paramètres. L’inférence s’effectue en un seul passage pour générer plusieurs requêtes. La page de présentation fournit le détail de cette approche et son articulation avec la récupération.