{"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/mico-learning-improved-representations-via","title":"MICo: Improved representations via sampling-based state similarity for Markov decision processes","arxiv_id":"2106.08229","date":"2021-06-03","proceeding":"NeurIPS 2021 12","authors":["Pablo Samuel Castro","Tyler Kastner","Prakash Panangaden","Mark Rowland"],"abstract":"We present a new behavioural distance over the state space of a Markov decision process, and demonstrate the use of this distance as an effective means of shaping the learnt representations of deep reinforcement learning agents. While existing notions of state similarity are typically difficult to learn at scale due to high computational cost and lack of sample-based algorithms, our newly-proposed distance addresses both of these issues. In addition to providing detailed theoretical analysis, we provide empirical evidence that learning this distance alongside the value function yields structured and informative representations, including strong results on the Arcade Learning Environment benchmark.","url_abs":"https://arxiv.org/abs/2106.08229v4","url_pdf":"https://arxiv.org/pdf/2106.08229v4.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"mico-learning-improved-representations-via","repo_url":"https://github.com/google-research/google-research/tree/master/mico","is_official":1,"mentioned_in_paper":0,"mentioned_in_github":0,"framework":"jax","reach":null},{"paper_slug":"mico-learning-improved-representations-via","repo_url":"https://github.com/zanghyu/offline_bisimulation","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[{"task_slug":"atari-games","task_name":"Atari Games"},{"task_slug":"deep-reinforcement-learning","task_name":"Deep Reinforcement Learning"},{"task_slug":"reinforcement-learning-1","task_name":"Reinforcement Learning (RL)"},{"task_slug":"reinforcement-learning-2","task_name":"reinforcement-learning"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":null,"atlas_url":"https://app.syntology.ai/?focus=2106.08229","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2106.08229"}},"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":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/google-research/google-research/tree/master/mico","reach":null},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/zanghyu/offline_bisimulation","reach":{"status":"ok"}}],"summary":{"ran":1,"ran_violates":1,"ran_honours":1,"ran_draft_wrong":1},"by_repo_kind":{"listed":{"samples":4,"ran":4,"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":4,"samples":[{"code_sha256_prefix":"94b0cae252af4dba","entry":"VMFProbabilisticTransitionModel","repo":"zanghyu/offline_bisimulation","repo_kind":"listed","path":"transition_model.py","file_url":"https://github.com/zanghyu/offline_bisimulation/blob/HEAD/transition_model.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"94b0cae252af4dba"}},{"code_sha256_prefix":"cdb052faeb90751e","entry":"_sample_orthonormal_to","repo":"zanghyu/offline_bisimulation","repo_kind":"listed","path":"transition_model.py","file_url":"https://github.com/zanghyu/offline_bisimulation/blob/HEAD/transition_model.py","link_basis":"first_harvest_node","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"cdb052faeb90751e"}},{"code_sha256_prefix":"8959efc52d7369e1","entry":"_sample_weight","repo":"zanghyu/offline_bisimulation","repo_kind":"listed","path":"transition_model.py","file_url":"https://github.com/zanghyu/offline_bisimulation/blob/HEAD/transition_model.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"8959efc52d7369e1"}},{"code_sha256_prefix":"8b3662abcb2a2524","entry":"sample_vMF","repo":"zanghyu/offline_bisimulation","repo_kind":"listed","path":"transition_model.py","file_url":"https://github.com/zanghyu/offline_bisimulation/blob/HEAD/transition_model.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"8b3662abcb2a2524"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}