HanoiWorld : A Joint Embedding Predictive Architecture BasedWorld Model for Autonomous Vehicle Controller
About
Current attempts of Reinforcement Learning for Autonomous Controller are data-demanding while the results are under-performed, unstable, and unable to grasp and anchor on the concept of safety, and over-concentrating on noise features due to the nature of pixel reconstruction. While current Self-Supervised Learningapproachs that learning on high-dimensional representations by leveraging the JointEmbedding Predictive Architecture (JEPA) are interesting and an effective alternative, as the idea mimics the natural ability of the human brain in acquiring new skill usingimagination and minimal samples of observations. This study introduces Hanoi-World, a JEPA-based world model that using recurrent neural network (RNN) formaking longterm horizontal planning with effective inference time. Experimentsconducted on the Highway-Env package with difference enviroment showcase the effective capability of making a driving plan while safety-awareness, with considerablecollision rate in comparison with SOTA baselines
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Autonomous Driving | Highway-Env v0 (100 evaluation episodes) | Collision Rate20 | 3 | |
| Autonomous Driving | Highway-Env roundabout v0 (100 evaluation episodes) | Collision Rate34 | 3 | |
| Autonomous Driving Planning | roundabout v0 (test) | Average Episode Reward9.818 | 3 | |
| Autonomous Driving Planning | highway v0 (test) | Avg Episode Reward13.163 | 3 | |
| Autonomous Driving | Highway-Env merge v0 (100 evaluation episodes) | Collision Rate0.97 | 3 | |
| Autonomous Driving Planning | merge v0 (test) | Avg Episode Reward13.48 | 3 |