Search papers, labs, and topics across Lattice.
This paper introduces Evolving Cache Schedules (EVO), an innovative framework that optimizes the inference process of diffusion policies by employing evolutionary search to schedule cache refreshes. By intelligently reusing intermediate activations and addressing the heterogeneous redundancy across computation blocks, EVO significantly enhances the performance-efficiency trade-off during real-time deployment. The method achieves up to an 8.05x speedup in action generation while maintaining nearly full performance, demonstrating its effectiveness in reducing computational demands in visuomotor control tasks.
Evolving Cache Schedules can slash action-generation time by over 8x without sacrificing performance, revolutionizing real-time deployment of diffusion policies.
Diffusion policies achieve strong visuomotor control by iteratively denoising action chunks, but repeated denoising makes real-time deployment computationally demanding. Cache-based methods reduce inference cost by reusing intermediate activations, but existing training-free schedules typically allocate computation uniformly across blocks, ignoring heterogeneous redundancy across blocks and leading to a suboptimal performance-efficiency trade-off. To bridge this gap, we introduce Evolving Cache Schedules (EVO), a training-free acceleration framework that globally schedules cache refreshes via evolutionary search. EVO represents each candidate as a complete schedule over the block-timestep lattice. Thus, redundant transformer computations during iterative denoising can be skipped through cache reuse while preserving closed-loop rollout performance. To make the search practical, EVO introduces redundancy-aware initialization, which seeds the population with promising schedules, and target-conditioned early stopping, which verifies and terminates once a desired performance target is reached. The offline-optimized schedule can be directly plugged into pretrained diffusion policies without retraining. Extensive manipulation benchmarks show that EVO preserves near-full performance while substantially reducing computation, achieving up to 8.05x action-generation speedup and reducing FLOPs from 15.77G to as low as 1.96G. Source code is available at https://github.com/pillom/EVO.