LeWM applied SIGReg per time-step, so nothing prevents temporal collapse. If your data has such slow features, simply apply SIGReg on the (N*T, D) tensor. We showed this solves your problem here:
And
@vlad_is_ai already spoked about JEPA+Slow features 4 years ago here
JEPA has a huge blind spot no one talks about: it’s starving for stable, slow features.
Add a SINGLE irrelevant patch, keep it fixed within each episode, and planning falls apart. Make it unpredictable, and performance almost returns to normal. Wild!
More in a blog post soon.
もっと見る