כתבה
arXiv cs.LG ·
תכנון דליל במודלים עולמיים חזותיים
Sparse Planning in Visual World Models via Cost Gradients
COSTGRAD הוא מנגנון חדש לתכנון דליל במודלים עולמיים חזותיים. הוא מאפשר לדרג טוקנים על פי חשיבותם לתכנון, ובכך להאיץ את תהליך התכנון. COSTGRAD הוכח כיעיל במספר בנכות, והראה שיפור של עד 5 פעמים בביצועים.
תקציר מקורי באנגליתarXiv:2610.10274v1 Announce Type: new Abstract: Token-based world models enable fine-grained latent planning, but repeatedly processing large spatial token grids makes action search expensive. We introduce COSTGRAD, a training-free, goal-conditioned selector that ranks spatial tokens by the gradient norm of the planning cost with respect to each input token. By deriving importance from the downstream control objective, COSTGRAD targets tokens that matter for planning rather than merely for prediction. On AdaLN-conditioned predictors at $50\%$ sparsity, COSTGRAD matches or exceeds full-token planning on three of four continuous-control benchmarks, while giving a measured $2.6\times$ wall-clock speedup per environment planning step. Combining token sparsity with reduced CEM search increases
קרא במקור המקורי
arxiv.org
פתח כתבה מקורית