Yuchen Ma | 马羽宸

Yuchen Ma 马羽宸

Email · Scholar · GitHub · LinkedIn · X

I am a PhD student in Computer Science at LMU Munich, supervised by Prof. Stefan Feuerriegel. My research interests are in agentic AI, large language models, and causal inference. I'm happy to connect and discuss potential collaborations.

News

Jun 2026 Joining Microsoft Research as a research intern this summer. I'll be in Seattle this summer, happy to meet for a coffee chat!
May 2026 Research stay at University of Notre Dame (May – Jun 2026).
May 2026 One paper accepted at KDD 2026. See you in Jeju!
May 2026 One paper accepted at ICML 2026. See you in Seoul!
Apr 2026 Organizing the RelSciFM @ KDD 2026 workshop on Reliable Scientific Foundation Models.
Jan 2026 One paper accepted at ICLR 2026. See you in Rio de Janeiro!
Jan 2026 Released the CausalFM toolkit — docs.
Sep 2025 One paper accepted at NeurIPS 2025. See you in San Diego!
May 2025 One paper accepted at KDD 2025. See you in Toronto!
Sep 2024 One paper accepted at NeurIPS 2024. See you in Vancouver!

Publications

The full publication list can be found on my Google Scholar profile.

Synthetic Interaction Data for Scalable Personalization in Large Language Models
Yuchen Ma, Yue Huang, Wenjie Wang, Xiaonan Luo, Xiangliang Zhang, Stefan Feuerriegel
KDD 2026
ProbeLLM: Automating Principled Diagnosis of LLM Failures
Yue Huang, Zhengzhe Jiang, Yuchen Ma, Yu Jiang, Xiangqi Wang, Yujun Zhou, Yuexing Hao, Kehan Guo, Pin-Yu Chen, Stefan Feuerriegel, Xiangliang Zhang
ICML 2026
Foundation Models for Causal Inference via Prior-Data Fitted Networks
Yuchen Ma, Dennis Frauen, Emil Javurek, Stefan Feuerriegel
ICLR 2026
LLM-based Treatment Effect Estimation under Inference Time Text Confounding
Yuchen Ma, Dennis Frauen, Jonas Schweisthal, Stefan Feuerriegel
NeurIPS 2025
A Diffusion-Based Method for Learning the Multi-Outcome Distribution of Medical Treatments
Yuchen Ma, Jonas Schweisthal, Hengrui Zhang, Stefan Feuerriegel
KDD 2025
DiffPO: A Causal Diffusion Model for Learning Distributions of Potential Outcomes
Yuchen Ma, Valentyn Melnychuk, Jonas Schweisthal, Stefan Feuerriegel
NeurIPS 2024
Distilling Knowledge from Self-Supervised Teacher by Embedding Graph Alignment
Yuchen Ma, Yanbei Chen, Zeynep Akata
BMVC 2022

Experience

Jun — Sep 2026
Research Intern
Microsoft Research · Advisors: Swadheen Shukla, Michel Galley
Jun 2021 — Feb 2022
Research Assistant
Max Planck Institute · Advisor: Prof. Zeynep Akata

Education

2022 — Present
Ph.D. in Computer Science
LMU Munich, Germany · Advisor: Prof. Stefan Feuerriegel
2019 — 2022
M.Sc. in Mathematics & Computer Vision
Heidelberg University, Germany · Advisor: Prof. Zeynep Akata
2015 — 2019
B.Sc. in Mathematics
Shandong University, China · Advisor: Prof. Guanghui Wang