About Me
Hello everyone, my name is Yuchen Ma. I am a PhD student in Computer Science at LMU Munich, supervised by Professor Stefan Feuerriegel.
I’m very happy to connect and discuss potential collaborations.
Research Focus
- LLM Alignment and Agentic AI: Develop alignment methods and agentic AI for large language models, with a focus on efficient data synthesis, personalization, prompt optimization, reinforcement learning, and applications of agentic workflows.
- Causal Foundation Models: Develop scalable transformer-based foundation models for Bayesian causal inference, enabling zero-shot, in-context learning of treatment effects with uncertainty quantification.
- Generative Models (Diffusion Models / LLMs) with Causal Inference: Explore diffusion models and large language models as generative tools for causal inference, with an emphasis on modeling distributions and handling text-based confounding.
News
Publications
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
Education
Present
LMU Munich, GermanySep. 2022
Heidelberg University, GermanyJun. 2019
Shandong University, ChinaExperience
Sep. 2026
Microsoft ResearchFeb. 2022
Max Planck Institute