{"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-2601-19336","title":"From Observations to Events: Event-Aware World Model for Reinforcement Learning","arxiv_id":"2601.19336","date":"2026-01-27","proceeding":null,"authors":["Zhao-Han Peng","Shaohui Li","Zhi Li","Shulan Ruan","Yu Liu","You He"],"abstract":"While model-based reinforcement learning (MBRL) improves sample efficiency by learning world models from raw observations, existing methods struggle to generalize across structurally similar scenes and remain vulnerable to spurious variations such as textures or color shifts. From a cognitive science perspective, humans segment continuous sensory streams into discrete events and rely on these key events for decision-making. Motivated by this principle, we propose the Event-Aware World Model (EAWM), a general framework that learns event-aware representations to streamline policy learning without requiring handcrafted labels. EAWM employs an automated event generator to derive events from raw observations and introduces a Generic Event Segmentor (GES) to identify event boundaries, which mark the start and end time of event segments. Through event prediction, the representation space is shaped to capture meaningful spatio-temporal transitions. Beyond this, we present a unified formulation of seemingly distinct world model architectures and show the broad applicability of our methods. Experiments on Atari 100K, Craftax 1M, and DeepMind Control 500K, DMC-GB2 500K demonstrate that EAWM consistently boosts the performance of strong MBRL baselines by 10%-45%, setting new state-of-the-art results across benchmarks. Our code is released at https://github.com/MarquisDarwin/EAWM.","url_abs":"https://arxiv.org/abs/2601.19336","url_pdf":"https://arxiv.org/pdf/2601.19336","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":{"atlas_url":null,"mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2601.19336"}},"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/MarquisDarwin/EAWM","reach":null}],"summary":{"ran_honours":1,"ran_draft_wrong":2,"unverified":4},"by_repo_kind":{"found_in_text":{"samples":7,"ran":3,"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":7,"samples":[{"code_sha256_prefix":"b7ff9ef9697d35f1","entry":"count_steps","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/dreamer.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/dreamer.py","link_basis":"plan_row","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"b7ff9ef9697d35f1"}},{"code_sha256_prefix":"fd421307a7e4cede","entry":"symexp","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/tools.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/tools.py","link_basis":"plan_row","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"fd421307a7e4cede"}},{"code_sha256_prefix":"3e346b9b00ea8169","entry":"symlog","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/tools.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/tools.py","link_basis":"plan_row","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"3e346b9b00ea8169"}},{"code_sha256_prefix":"d5bbb08ea8eb8767","entry":"calc_hns","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/tbcurve.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/tbcurve.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"d5bbb08ea8eb8767"}},{"code_sha256_prefix":"e2d69d87d575b649","entry":"calc_same_pad","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/networks.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/networks.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"e2d69d87d575b649"}},{"code_sha256_prefix":"91c5e45adc663e88","entry":"get_aug","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/augmentations.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/augmentations.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"91c5e45adc663e88"}},{"code_sha256_prefix":"97697619e5f42da7","entry":"simulate","repo":"MarquisDarwin/EAWM","repo_kind":"found_in_text","path":"EADream/tools.py","file_url":"https://github.com/MarquisDarwin/EAWM/blob/HEAD/EADream/tools.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"GPL-3.0","inline_ok":false,"mcp_get_code":{"code_sha256":"97697619e5f42da7"}}]},"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}