{"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/can-transformers-be-strong-treatment-effect","title":"Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimation","arxiv_id":"2202.01336","date":"2022-02-02","proceeding":null,"authors":["Yi-Fan Zhang","HANLIN ZHANG","Zachary C. Lipton","Li Erran Li","Eric P. Xing"],"abstract":"Previous works on Treatment Effect Estimation (TEE) are not in widespread use because they are predominantly theoretical, where strong parametric assumptions are made but untractable for practical application. Recent work uses multilayer perceptron (MLP) for modeling casual relationships, however, MLPs lag far behind recent advances in ML methodology, which limits their applicability and generalizability. To extend beyond the single domain formulation and towards more realistic learning scenarios, we explore model design spaces beyond MLPs, i.e., transformer backbones, which provide flexibility where attention layers govern interactions among treatments and covariates to exploit structural similarities of potential outcomes for confounding control. Through careful model design, Transformers as Treatment Effect Estimators (TransTEE) is proposed. We show empirically that TransTEE can: (1) serve as a general purpose treatment effect estimator that significantly outperforms competitive baselines in a variety of challenging TEE problems (e.g., discrete, continuous, structured, or dosage-associated treatments) and is applicable to both when covariates are tabular and when they consist of structural data (e.g., texts, graphs); (2) yield multiple advantages: compatibility with propensity score modeling, parameter efficiency, robustness to continuous treatment value distribution shifts, explainable in covariate adjustment, and real-world utility in auditing pre-trained language models","url_abs":"https://arxiv.org/abs/2202.01336v5","url_pdf":"https://arxiv.org/pdf/2202.01336v5.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":"can-transformers-be-strong-treatment-effect","repo_url":"https://github.com/hlzhang109/transtee","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"heterogeneous-treatment-effect-estimation","task_name":"Heterogeneous Treatment Effect Estimation"},{"task_slug":"pos","task_name":"POS"},{"task_slug":"selection-bias","task_name":"Selection bias"}],"methods":[{"method_slug":"absolute-position-encodings","method_name":"Absolute Position Encodings"},{"method_slug":"adam","method_name":"Adam"},{"method_slug":"attention","method_name":"Attention"},{"method_slug":"bpe","method_name":"BPE"},{"method_slug":"dense-connections","method_name":"Dense Connections"},{"method_slug":"dropout","method_name":"Dropout"},{"method_slug":"label-smoothing","method_name":"Label Smoothing"},{"method_slug":"layer-normalization","method_name":"Layer Normalization"},{"method_slug":"linear-layer","method_name":"Linear Layer"},{"method_slug":"multi-head-attention","method_name":"Multi-Head Attention"},{"method_slug":"position-wise-feed-forward-layer","method_name":"Position-Wise Feed-Forward Layer"},{"method_slug":"residual-connection","method_name":"Residual Connection"},{"method_slug":"softmax","method_name":"Softmax"},{"method_slug":"transformer","method_name":"Transformer"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2202.01336","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2202.01336"}},"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/hlzhang109/transtee","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran_draft_wrong":4},"by_repo_kind":{"official":{"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":0,"samples":[{"code_sha256_prefix":"e0e067c9a043710c","entry":"comp_grid","repo":"hlzhang109/transtee","repo_kind":"official","path":"CausalM/POMS_GendeRace/pipeline/TransTEE/dynamic_net.py","file_url":"https://github.com/hlzhang109/transtee/blob/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/dynamic_net.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"e0e067c9a043710c"}},{"code_sha256_prefix":"2fb9f4386193338d","entry":"create_batch_norm_1d_layers","repo":"hlzhang109/transtee","repo_kind":"official","path":"CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py","file_url":"https://github.com/hlzhang109/transtee/blob/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"2fb9f4386193338d"}},{"code_sha256_prefix":"fe1c0d1da4fbf989","entry":"get_activation","repo":"hlzhang109/transtee","repo_kind":"official","path":"CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py","file_url":"https://github.com/hlzhang109/transtee/blob/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"fe1c0d1da4fbf989"}},{"code_sha256_prefix":"bc6b389c01bb19ef","entry":"get_initialiser","repo":"hlzhang109/transtee","repo_kind":"official","path":"CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py","file_url":"https://github.com/hlzhang109/transtee/blob/HEAD/CausalM/POMS_GendeRace/pipeline/TransTEE/DisCri.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"bc6b389c01bb19ef"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}