Google's Retrieve-for-Train cuts complex AI search from ~50s to sub-second

aigclink · x · 2026-09-16

Google Research unveiled Retrieve-for-Train, an ICML 2026 framework that moves AI search's "thinking" from inference time to offline training, cutting large-scale query latency from 50 seconds to sub-second—12-20x faster than autoregressive approaches. The method uses offline RL to compile expert retrieval behavior into a 53.9M-parameter diffusion model that generates an entire slate of retrieval targets in embedding space with a single parallel pass, bypassing expensive text CoT generation. It improves set-level quality (returning complementary slates like tent+sleeping bag+stove instead of ten near-identical tents) while removing the inference bottleneck that makes complex AI search impractical for real-time search boxes.

Related event: Google's Retrieve-for-Train Moves Search Thinking to Training Time for Sub-second Responses(6 posts)→

Original post →

More from Infra

Infra channel →