{"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/ode-discovery-for-longitudinal-heterogeneous","title":"ODE Discovery for Longitudinal Heterogeneous Treatment Effects Inference","arxiv_id":"2403.10766","date":"2024-03-16","proceeding":null,"authors":["Krzysztof Kacprzyk","Samuel Holt","Jeroen Berrevoets","Zhaozhi Qian","Mihaela van der Schaar"],"abstract":"Inferring unbiased treatment effects has received widespread attention in the machine learning community. In recent years, our community has proposed numerous solutions in standard settings, high-dimensional treatment settings, and even longitudinal settings. While very diverse, the solution has mostly relied on neural networks for inference and simultaneous correction of assignment bias. New approaches typically build on top of previous approaches by proposing new (or refined) architectures and learning algorithms. However, the end result -- a neural-network-based inference machine -- remains unchallenged. In this paper, we introduce a different type of solution in the longitudinal setting: a closed-form ordinary differential equation (ODE). While we still rely on continuous optimization to learn an ODE, the resulting inference machine is no longer a neural network. Doing so yields several advantages such as interpretability, irregular sampling, and a different set of identification assumptions. Above all, we consider the introduction of a completely new type of solution to be our most important contribution as it may spark entirely new innovations in treatment effects in general. We facilitate this by formulating our contribution as a framework that can transform any ODE discovery method into a treatment effects method.","url_abs":"https://arxiv.org/abs/2403.10766v1","url_pdf":"https://arxiv.org/pdf/2403.10766v1.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":"ode-discovery-for-longitudinal-heterogeneous","repo_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"jax","reach":{"status":"ok","spdx":"MIT"}},{"paper_slug":"ode-discovery-for-longitudinal-heterogeneous","repo_url":"https://github.com/vanderschaarlab/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"jax","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[],"methods":[{"method_slug":"set","method_name":"SET"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2403.10766","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2403.10766"}},"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/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","reach":{"status":"ok","spdx":"MIT"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/vanderschaarlab/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran":7,"unverified":2},"by_repo_kind":{"official":{"samples":9,"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":0,"samples":[{"code_sha256_prefix":"d023278409398a8b","entry":"config_to_dict","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/exp_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/exp_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"d023278409398a8b"}},{"code_sha256_prefix":"91d117b031324779","entry":"create_logger_in_process","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/logging_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/logging_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"91d117b031324779"}},{"code_sha256_prefix":"a855121467fdc9f9","entry":"dict_to_config","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/exp_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/exp_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a855121467fdc9f9"}},{"code_sha256_prefix":"417eade76c4290df","entry":"file_path_from_parent_directory","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/results_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/results_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"417eade76c4290df"}},{"code_sha256_prefix":"4f5266a8e125a9f4","entry":"get_dataset","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"libs_m/ct/runnables/run_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/libs_m/ct/runnables/run_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"4f5266a8e125a9f4"}},{"code_sha256_prefix":"05a7c2acf6129723","entry":"load_metrics_from_tensorboard_file","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/results_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/results_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"05a7c2acf6129723"}},{"code_sha256_prefix":"dfb9d0f97fc99892","entry":"moving_average","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/results_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/results_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"dfb9d0f97fc99892"}},{"code_sha256_prefix":"ac56469c6e01c50b","entry":"generate_log_file_path","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/logging_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/logging_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"ac56469c6e01c50b"}},{"code_sha256_prefix":"416afab47aa44669","entry":"load_dataset","repo":"samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference","repo_kind":"official","path":"utils/exp_utils.py","file_url":"https://github.com/samholt/ode-discovery-for-longitudinal-heterogeneous-treatment-effects-inference/blob/HEAD/utils/exp_utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"416afab47aa44669"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}