{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,8,18]],"date-time":"2026-08-18T01:46:41Z","timestamp":1787017601945,"version":"build-2736575974"},"publisher-location":"New York, NY, USA","reference-count":56,"publisher":"ACM","funder":[{"name":"German Federal Ministry of Education and Research","award":["01IS24082"],"award-info":[{"award-number":["01IS24082"]}]}],"content-domain":{"domain":["dl.acm.org"],"crossmark-restriction":true},"short-container-title":[],"published-print":{"date-parts":[[2025,8,3]]},"DOI":"10.1145\/3711896.3736819","type":"proceedings-article","created":{"date-parts":[[2025,8,3]],"date-time":"2025-08-03T20:54:17Z","timestamp":1754254457000},"page":"2066-2077","update-policy":"https:\/\/doi.org\/10.1145\/crossmark-policy","source":"Crossref","is-referenced-by-count":1,"title":["A Diffusion-Based Method for Learning the Multi-Outcome Distribution of Medical Treatments"],"prefix":"10.1145","author":[{"ORCID":"https:\/\/orcid.org\/0009-0003-9900-2822","authenticated-orcid":false,"given":"Yuchen","family":"Ma","sequence":"first","affiliation":[{"name":"Munich Center for Machine Learning & LMU Munich, Munich, Germany"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0000-0003-3725-3821","authenticated-orcid":false,"given":"Jonas","family":"Schweisthal","sequence":"additional","affiliation":[{"name":"Munich Center for Machine Learning & LMU Munich, Munich, Germany"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0009-0006-1330-0899","authenticated-orcid":false,"given":"Hengrui","family":"Zhang","sequence":"additional","affiliation":[{"name":"University of Illinois, Chicago, Chicago, IL, USA"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0000-0001-7856-8729","authenticated-orcid":false,"given":"Stefan","family":"Feuerriegel","sequence":"additional","affiliation":[{"name":"Munich Center for Machine Learning & LMU Munich, Munich, Germany"}],"role":[{"vocabulary":"crossref","role":"author"}]}],"member":"320","published-online":{"date-parts":[[2025,8,3]]},"reference":[{"key":"e_1_3_2_2_1_1","doi-asserted-by":"publisher","DOI":"10.1038\/s41591-023-02562-7"},{"key":"e_1_3_2_2_2_1","volume-title":"Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction. In International Conference on Machine Learning.","author":"Byambadalai Undral","year":"2024","unstructured":"Undral Byambadalai, Tatsushi Oka, and Shota Yasui. 2024. Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction. In International Conference on Machine Learning."},{"key":"e_1_3_2_2_3_1","doi-asserted-by":"publisher","DOI":"10.1214\/09-AOAS285"},{"key":"e_1_3_2_2_4_1","volume-title":"International Conference on Artificial Intelligence and Statistics.","author":"Curth Alicia","unstructured":"Alicia Curth and Mihaela van der Schaar. 2021a. Nonparametric estimation of heterogeneous treatment effects: From theory to learning algorithms. In International Conference on Artificial Intelligence and Statistics."},{"key":"e_1_3_2_2_5_1","unstructured":"Alicia Curth and Mihaela van der Schaar. 2021b. On inductive biases for heterogeneous treatment effect estimation. Advances in Neural Information Processing Systems(2021)."},{"key":"e_1_3_2_2_6_1","doi-asserted-by":"publisher","DOI":"10.1371\/journal.pdig.0000721"},{"key":"e_1_3_2_2_7_1","doi-asserted-by":"publisher","DOI":"10.1007\/s12094-019-02167-y"},{"key":"e_1_3_2_2_8_1","doi-asserted-by":"publisher","DOI":"10.1038\/s41591-024-02902-1"},{"key":"e_1_3_2_2_9_1","doi-asserted-by":"publisher","DOI":"10.1016\/0002-9343(80)90472-6"},{"key":"e_1_3_2_2_10_1","doi-asserted-by":"publisher","DOI":"10.2307\/2971729"},{"key":"e_1_3_2_2_11_1","volume-title":"International Conference on Learning Representations.","author":"Hess Konstantin","year":"2023","unstructured":"Konstantin Hess, Valentyn Melnychuk, Dennis Frauen, and Stefan Feuerriegel. 2023. Bayesian neural controlled differential equations for treatment effect estimation. In International Conference on Learning Representations."},{"key":"e_1_3_2_2_12_1","doi-asserted-by":"publisher","DOI":"10.1198\/jcgs.2010.08162"},{"key":"e_1_3_2_2_13_1","unstructured":"Jonathan Ho Ajay Jain and Pieter Abbeel. 2020. Denoising diffusion probabilistic models. In Advances in Neural Information Processing Systems."},{"key":"e_1_3_2_2_14_1","doi-asserted-by":"publisher","DOI":"10.1080\/01621459.1986.10478354"},{"key":"e_1_3_2_2_15_1","unstructured":"Fredrik D. Johansson Nathan Kallus Uri Shalit and David Sontag. 2018. Learning weighted representations for generalization across designs. arXiv preprint(2018)."},{"key":"e_1_3_2_2_16_1","volume-title":"International Conference on Machine Learning.","author":"Johansson Fredrik D.","year":"2016","unstructured":"Fredrik D. Johansson, Uri Shalit, and David Sontag. 2016. Learning representations for counterfactual inference. In International Conference on Machine Learning."},{"key":"e_1_3_2_2_17_1","doi-asserted-by":"publisher","DOI":"10.1038\/sdata.2016.35"},{"key":"e_1_3_2_2_18_1","doi-asserted-by":"publisher","DOI":"10.1093\/ajh\/hpt196"},{"key":"e_1_3_2_2_19_1","doi-asserted-by":"publisher","DOI":"10.1214\/23-EJS2157"},{"key":"e_1_3_2_2_20_1","doi-asserted-by":"publisher","DOI":"10.1177\/0962280217747130"},{"key":"e_1_3_2_2_21_1","unstructured":"Christoph Kern Unai Fischer-Abaigar Jonas Schweisthal Dennis Frauen Rayid Ghani Stefan Feuerriegel Mihaela van der Schaar and Frauke Kreuter. 2025. Algorithms for reliable decision-making need causal reasoning. Nature Computational Science(2025)."},{"key":"e_1_3_2_2_22_1","doi-asserted-by":"publisher","DOI":"10.1016\/j.ecosta.2021.07.006"},{"key":"e_1_3_2_2_23_1","doi-asserted-by":"publisher","DOI":"10.1073\/pnas.1804597116"},{"key":"e_1_3_2_2_24_1","volume-title":"International Conference on Machine Learning.","author":"Lee Chaejeong","year":"2023","unstructured":"Chaejeong Lee, Jayoung Kim, and Noseong Park. 2023. CoDi: Co-evolving contrastive diffusion models for mixed-type tabular synthesis. In International Conference on Machine Learning."},{"key":"e_1_3_2_2_25_1","doi-asserted-by":"publisher","DOI":"10.1111\/j.0006-341X.2000.00593.x"},{"key":"e_1_3_2_2_26_1","unstructured":"Yuchen Ma Valentyn Melnychuk Jonas Schweisthal and Stefan Feuerriegel. 2024. DiffPO: A causal diffusion model for learning distributions of potential outcomes. In Advances in Neural Information Processing Systems."},{"key":"e_1_3_2_2_27_1","doi-asserted-by":"publisher","DOI":"10.1145\/3450439.3451871"},{"key":"e_1_3_2_2_28_1","first-page":"1","volume-title":"Monthly Vital Statistics Report","volume":"46","author":"MacDorman M.F.","year":"1998","unstructured":"M.F. MacDorman and J.O. Atkinson. 1998. Infant mortality statistics from the linked birth\/infant death data set-1995 period data. Monthly Vital Statistics Report, Vol. 46, 6 Suppl 2 (1998), 1-22."},{"key":"e_1_3_2_2_29_1","unstructured":"Valentyn Melnychuk Stefan Feuerriegel and Mihaela van der Schaar. 2024. Quantifying aleatoric uncertainty of the treatment effect: a novel orthogonal learner. In Advances in Neural Information Processing Systems."},{"key":"e_1_3_2_2_30_1","volume-title":"International Conference on Machine Learning.","author":"Melnychuk Valentyn","year":"2023","unstructured":"Valentyn Melnychuk, Dennis Frauen, and Stefan Feuerriegel. 2023. Normalizing flows for interventional density estimation. In International Conference on Machine Learning."},{"key":"e_1_3_2_2_31_1","doi-asserted-by":"crossref","unstructured":"Christof Naumzik Stefan Feuerriegel and Anne Molgaard Nielsen. 2023. Data-driven dynamic treatment planning for chronic diseases. European Journal of Operational Research(2023).","DOI":"10.1016\/j.ejor.2022.06.016"},{"key":"e_1_3_2_2_32_1","volume-title":"Proceedings of the Conference on Health, Inference, and Learning (CHIL).","author":"Naumzik Christof","year":"2024","unstructured":"Christof Naumzik, Alice Kongsted, Werner Vach, and Stefan Feuerriegel. 2024. Data-driven subgrouping of patient trajectories with chronic diseases: Evidence from low back pain. In Proceedings of the Conference on Health, Inference, and Learning (CHIL)."},{"key":"e_1_3_2_2_33_1","doi-asserted-by":"publisher","DOI":"10.1093\/biomet\/asaa076"},{"key":"e_1_3_2_2_34_1","doi-asserted-by":"publisher","DOI":"10.1161\/CIR.0000000000000426"},{"key":"e_1_3_2_2_35_1","volume-title":"Elements of Causal Inference: Foundations and Learning Algorithms","author":"Peters Jonas","unstructured":"Jonas Peters, Dominik Janzing, and Bernhard Sch\u00f6lkopf. 2017. Elements of Causal Inference: Foundations and Learning Algorithms. MIT Press, Cambridge, MA."},{"key":"e_1_3_2_2_36_1","doi-asserted-by":"publisher","DOI":"10.1093\/biostatistics\/4.3.371"},{"key":"e_1_3_2_2_37_1","doi-asserted-by":"publisher","DOI":"10.1198\/016214504000001880"},{"key":"e_1_3_2_2_38_1","doi-asserted-by":"publisher","DOI":"10.1186\/1745-6215-12-101"},{"key":"e_1_3_2_2_39_1","volume-title":"International Conference on Machine Learning.","author":"Shalit Uri","year":"2017","unstructured":"Uri Shalit, Fredrik D. Johansson, and David Sontag. 2017. Estimating individual treatment effect: Generalization bounds and algorithms. In International Conference on Machine Learning."},{"key":"e_1_3_2_2_40_1","volume-title":"International Conference on Machine Learning.","author":"Sohl-Dickstein Jascha","year":"2015","unstructured":"Jascha Sohl-Dickstein, Eric Weiss, Niru Maheswaranathan, and Surya Ganguli. 2015. Deep unsupervised learning using nonequilibrium thermodynamics. In International Conference on Machine Learning."},{"key":"e_1_3_2_2_41_1","unstructured":"Yang Song and Stefano Ermon. 2019. Generative modeling by estimating gradients of the data distribution. In Advances in Neural Information Processing Systems."},{"key":"e_1_3_2_2_42_1","volume-title":"International Conference on Learning Representations.","author":"Song Yang","year":"2020","unstructured":"Yang Song, Jascha Sohl-Dickstein, Diederik P Kingma, Abhishek Kumar, Stefano Ermon, and Ben Poole. 2020. Score-based generative modeling through stochastic differential equations. In International Conference on Learning Representations."},{"key":"e_1_3_2_2_43_1","doi-asserted-by":"publisher","DOI":"10.1146\/annurev-statistics-010814-020148"},{"key":"e_1_3_2_2_44_1","doi-asserted-by":"publisher","DOI":"10.1161\/CIRCOUTCOMES.111.961581"},{"key":"e_1_3_2_2_45_1","doi-asserted-by":"publisher","DOI":"10.1016\/S0140-6736(21)00312-3"},{"key":"e_1_3_2_2_46_1","doi-asserted-by":"publisher","DOI":"10.1098\/rsos.181870"},{"key":"e_1_3_2_2_47_1","doi-asserted-by":"publisher","DOI":"10.1080\/01621459.2017.1319839"},{"key":"e_1_3_2_2_48_1","doi-asserted-by":"publisher","DOI":"10.1145\/3368555.3384469"},{"key":"e_1_3_2_2_49_1","volume-title":"Kehl","author":"Weberpals Janick","year":"2025","unstructured":"Janick Weberpals, Stefan Feuerriegel, Mihaela van der Schaar, and Kenneth L. Kehl. 2025. Opportunities for causal machine learning in precision oncology. NEJM AI(2025)."},{"key":"e_1_3_2_2_50_1","unstructured":"Yong Wu Mingzhou Liu Jing Yan Yanwei Fu Shouyan Wang Yizhou Wang and Xinwei Sun. 2023. The blessings of multiple treatments and outcomes in treatment effect estimation. arXiv preprint(2023)."},{"key":"e_1_3_2_2_51_1","unstructured":"Leon Yao Caroline Lo Israel Nir Sarah Tan Ariel Evnine Adam Lerer and Alex Peysakhovich. 2022. Efficient heterogeneous treatment effect estimation with multiple experiments and multiple outcomes. arXiv preprint arXiv:2206.04907(2022)."},{"key":"e_1_3_2_2_52_1","volume-title":"Alternative methods for testing treatment effects on the basis of multiple outcomes: simulation and case study. Statistics in medicine","author":"Yoon Frank B","year":"2011","unstructured":"Frank B Yoon, Garrett M Fitzmaurice, Stuart R Lipsitz, Nicholas J Horton, Nan M Laird, and Sharon-Lise T Normand. 2011. Alternative methods for testing treatment effects on the basis of multiple outcomes: simulation and case study. Statistics in medicine, Vol. 30, 16 (2011), 1917-1932."},{"key":"e_1_3_2_2_53_1","volume-title":"International Conference on Learning Representations.","author":"Yoon Jinsung","unstructured":"Jinsung Yoon, James Jordon, and Mihaela van der Schaar. 2018. GANITE: Estimation of individualized treatment effects using generative adversarial nets. In International Conference on Learning Representations."},{"key":"e_1_3_2_2_54_1","doi-asserted-by":"publisher","DOI":"10.1164\/rccm.202006-2381CP"},{"key":"e_1_3_2_2_55_1","unstructured":"Hengrui Zhang Liancheng Fang Qitian Wu and Philip S Yu. 2024a. Diffusion-nested auto-regressive synthesis of heterogeneous tabular data. arXiv preprint arXiv:2410.21523(2024)."},{"key":"e_1_3_2_2_56_1","volume-title":"International Conference on Learning Representations.","author":"Zhang Hengrui","year":"2024","unstructured":"Hengrui Zhang, Jiani Zhang, Balasubramaniam Srinivasan, Zhengyuan Shen, Xiao Qin, Christos Faloutsos, Huzefa Rangwala, and George Karypis. 2024b. Mixed-type tabular data synthesis with score-based diffusion in latent space. In International Conference on Learning Representations."}],"event":{"name":"KDD '25: The 31st ACM SIGKDD Conference on Knowledge Discovery and Data Mining","location":"Toronto ON Canada","acronym":"KDD '25","sponsor":["SIGKDD ACM Special Interest Group on Knowledge Discovery in Data","SIGMOD ACM Special Interest Group on Management of Data"]},"container-title":["Proceedings of the 31st ACM SIGKDD Conference on Knowledge Discovery and Data Mining V.2"],"original-title":[],"link":[{"URL":"https:\/\/dl.acm.org\/doi\/pdf\/10.1145\/3711896.3736819","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2026,4,30]],"date-time":"2026-04-30T18:10:15Z","timestamp":1777572615000},"score":1,"resource":{"primary":{"URL":"https:\/\/dl.acm.org\/doi\/10.1145\/3711896.3736819"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2025,8,3]]},"references-count":56,"alternative-id":["10.1145\/3711896.3736819","10.1145\/3711896"],"URL":"https:\/\/doi.org\/10.1145\/3711896.3736819","relation":{},"subject":[],"published":{"date-parts":[[2025,8,3]]},"assertion":[{"value":"2025-08-03","order":3,"name":"published","label":"Published","group":{"name":"publication_history","label":"Publication History"}}]}}