{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2603-14392","title":"WestWorld: A Knowledge-Encoded Scalable Trajectory World Model for Diverse Robotic Systems","arxiv_id":"2603.14392","date":"2026-03-15","proceeding":"ICML","authors":["Yuchen Wang","Jiangtao Kong","Sizhe Wei","Xiaochang Li","Haohong Lin","Hongjue Zhao","Tianyi Zhou","Lu Gan","Huajie Shao"],"abstract":"Trajectory world models play a crucial role in robotic dynamics learning, planning, and control. While recent works have explored trajectory world models for diverse robotic systems, they struggle to scale to a large number of distinct system dynamics and overlook domain knowledge of physical structures. To address these limitations, we introduce WestWorld, a knoWledge-Encoded Scalable Trajectory World model for diverse robotic systems. To tackle the scalability challenge, we propose a novel system-aware Mixture-of-Experts (Sys-MoE) that dynamically combines and routes specialized experts for different robotic systems via a learnable system embedding. To further enhance zero-shot generalization, we incorporate domain knowledge of robot physical structures by introducing a structural embedding that aligns trajectory representations with morphological information. After pretraining on 89 complex environments spanning diverse morphologies across both simulation and real-world settings, WestWorld achieves significant improvements over competitive baselines in zero- and few-shot trajectory prediction. Additionally, it shows strong scalability across a wide range of robotic environments and significantly improves performance on downstream model-based control for different robots. Finally, we deploy our model on a real-world Unitree Go1, where it demonstrates stable locomotion performance. The code is available at https://github.com/511205787/WestWorld.","url_abs":"https://arxiv.org/abs/2603.14392","url_pdf":"https://arxiv.org/pdf/2603.14392","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":null,"atlas_url":"https://app.syntology.ai/?focus=2603.14392","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2603.14392"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/511205787/WestWorld","reach":null}],"summary":{"unverified":9},"by_repo_kind":{"found_in_text":{"samples":9,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":9,"samples":[{"code_sha256_prefix":"256d234570e4b1f8","entry":"build_shared_support_from_stats","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/westworld.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/westworld.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"256d234570e4b1f8"}},{"code_sha256_prefix":"771c65667dd313cc","entry":"sinkhorn","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/mamba_moe.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/mamba_moe.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"771c65667dd313cc"}},{"code_sha256_prefix":"bd83a230627fd57f","entry":"symexp_torch","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/westworld.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/westworld.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"bd83a230627fd57f"}},{"code_sha256_prefix":"e5912388dbacfd5c","entry":"symlog_torch","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/westworld.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/westworld.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"e5912388dbacfd5c"}},{"code_sha256_prefix":"55e9f36554f7f672","entry":"transform","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/westworld_utils.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/westworld_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"55e9f36554f7f672"}},{"code_sha256_prefix":"f01fc5b65f20bb9c","entry":"transform","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"MPPI/hopper/world_model_westworld/westworld_utils.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/MPPI/hopper/world_model_westworld/westworld_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"f01fc5b65f20bb9c"}},{"code_sha256_prefix":"ffb516bd8f2ed3e3","entry":"transform_to_onehot","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/westworld_utils.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/westworld_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"ffb516bd8f2ed3e3"}},{"code_sha256_prefix":"9342027103e44c25","entry":"transform_to_probs","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"models/WestWorld/westworld_utils.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/models/WestWorld/westworld_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"9342027103e44c25"}},{"code_sha256_prefix":"a9abc7424e907664","entry":"transform_to_probs","repo":"511205787/WestWorld","repo_kind":"found_in_text","path":"MPPI/hopper/world_model_westworld/westworld_utils.py","file_url":"https://github.com/511205787/WestWorld/blob/HEAD/MPPI/hopper/world_model_westworld/westworld_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"a9abc7424e907664"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}