{"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/contrastive-balancing-representation-learning","title":"Contrastive Balancing Representation Learning for Heterogeneous Dose-Response Curves Estimation","arxiv_id":"2403.14232","date":"2024-03-21","proceeding":null,"authors":["Minqin Zhu","Anpeng Wu","Haoxuan Li","Ruoxuan Xiong","Bo Li","Xiaoqing Yang","Xuan Qin","Peng Zhen","Jiecheng Guo","Fei Wu","Kun Kuang"],"abstract":"Estimating the individuals' potential response to varying treatment doses is crucial for decision-making in areas such as precision medicine and management science. Most recent studies predict counterfactual outcomes by learning a covariate representation that is independent of the treatment variable. However, such independence constraints neglect much of the covariate information that is useful for counterfactual prediction, especially when the treatment variables are continuous. To tackle the above issue, in this paper, we first theoretically demonstrate the importance of the balancing and prognostic representations for unbiased estimation of the heterogeneous dose-response curves, that is, the learned representations are constrained to satisfy the conditional independence between the covariates and both of the treatment variables and the potential responses. Based on this, we propose a novel Contrastive balancing Representation learning Network using a partial distance measure, called CRNet, for estimating the heterogeneous dose-response curves without losing the continuity of treatments. Extensive experiments are conducted on synthetic and real-world datasets demonstrating that our proposal significantly outperforms previous methods.","url_abs":"https://arxiv.org/abs/2403.14232v1","url_pdf":"https://arxiv.org/pdf/2403.14232v1.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":"contrastive-balancing-representation-learning","repo_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet","is_official":1,"mentioned_in_paper":0,"mentioned_in_github":0,"framework":"pytorch","reach":null}],"tasks":[{"task_slug":"decision-making","task_name":"Decision Making"},{"task_slug":"management","task_name":"Management"},{"task_slug":"representation-learning","task_name":"Representation Learning"},{"task_slug":null,"task_name":"counterfactual"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2403.14232","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2403.14232"}},"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/euzmin/Contrastive-Balancing-Representation-Network-CRNet","reach":null}],"summary":{"ran":4,"ran_honours":3},"by_repo_kind":{"official":{"samples":7,"ran":7,"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":"67289c78034d5de8","entry":"Backbone_module","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"67289c78034d5de8"}},{"code_sha256_prefix":"95dd03e55083611f","entry":"CRNet","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"95dd03e55083611f"}},{"code_sha256_prefix":"c9b04389808da25c","entry":"Partial_DC","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.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":"c9b04389808da25c"}},{"code_sha256_prefix":"06559cc5802d5a2a","entry":"T_module","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"06559cc5802d5a2a"}},{"code_sha256_prefix":"16f33524446b9301","entry":"U_distance_matrix","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.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":"16f33524446b9301"}},{"code_sha256_prefix":"4d06c0b6546fde57","entry":"U_product","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.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":"4d06c0b6546fde57"}},{"code_sha256_prefix":"f85d570105a0b89b","entry":"X_module","repo":"euzmin/Contrastive-Balancing-Representation-Network-CRNet","repo_kind":"official","path":"models/crnet.py","file_url":"https://github.com/euzmin/Contrastive-Balancing-Representation-Network-CRNet/blob/HEAD/models/crnet.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f85d570105a0b89b"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}