activity
20232026
most citedDiffPO: A causal diffusion model for learning distributions of potential outcomes

1 citations · 2 across the 15 of their papers we have counts for

collaborators

16 papers

cs.MA2026

EAIR: Interaction Risks in Enterprise LLM Multi-Agent Systems

Yuchen Ma, Swadheen Shukla, Paulo Salem +5

LLM agents increasingly divide enterprise work across roles with different goals, authority, access, and information. Because each agent often sees only a local snapshot of the ent…

cs.AI2026

MemoHarness: Agent Harnesses That Learn from Experience

Yue Huang, Wenjie Wang, Han Bao +7

An agent harness is the external control layer that turns a base LLM into an executable agent by managing context, tools, orchestration, memory, decoding, and output handling. Whil…

cs.AI2026

UXBench: Measuring the Actionability of LLM-Generated UX Critiques

Wenjie Wang, Yue Huang, Zipeng Ling +11

Large language models (LLMs) are increasingly deployed as UX judges that inspect interfaces, diagnose usability problems, and propose repairs. Yet no controlled benchmark measures…

cs.AI2026

SpecAlign: Efficient Specification-Grounded Alignment of Large Language Models via Synthetic Data

Wenjie Wang, Yue Huang, Zhengqing Yuan +6

As large language models (LLMs) are increasingly deployed in real-world applications, alignment is no longer governed by a single universal notion of safety or helpfulness, but ins…

cs.LG2026

Causal methods for LLM development and evaluation

Dennis Frauen, Marie Brockschmidt, Konstantin Hess +10

Large language model (LLM) development is currently driven by large-scale empirical iteration over data mixtures, reward models, routing strategies, and evaluation pipelines. Here,…

cs.CR2026

AgentTrap: Measuring Runtime Trust Failures in Third-Party Agent Skills

Haomin Zhuang, Hanwen Xing, Yujun Zhou +5

Third-party skills are becoming the package ecosystem for LLM agents. They package natural-language instructions, helper scripts, templates, documents, and service configuration in…

cs.LG2026

SkillGen: Verified Inference-Time Agent Skill Synthesis

Yuchen Ma, Yue Huang, Han Bao +5

Skills are a promising way to improve LLM agent capabilities without retraining, while keeping the added procedure reusable and controllable. However, high-quality skills are still…

cs.CL2026

NARRA-Gym for Evaluating Interactive Narrative Agents

Yue Huang, Yuchen Ma, Jiayi Ye +14

Interactive narrative tasks require LLMs to sustain a coherent, evolving story while adapting to a user over multiple turns. However, suitable benchmarks for this setting are limit…

cs.CL2026

ProbeLLM: Automating Principled Diagnosis of LLM Failures

Yue Huang, Zhengzhe Jiang, Yuchen Ma +8

Understanding how and why large language models (LLMs) fail is becoming a central challenge as models rapidly evolve and static evaluations fall behind. While automated probing has…

cs.LG2026

Synthetic Interaction Data for Scalable Personalization in Large Language Models

Yuchen Ma, Yue Huang, Wenjie Wang +3

Personalized prompting offers large opportunities for deploying large language models (LLMs) to diverse users, yet existing prompt optimization methods primarily focus on task-leve…

stat.ML2026

Targeted Synthetic Control Method

Yuxin Wang, Dennis Frauen, Emil Javurek +3

The synthetic control method (SCM) estimates causal effects in panel data with a single-treated unit by constructing a counterfactual outcome as a weighted combination of untreated…

cs.LG2025

LLM-Driven Treatment Effect Estimation Under Inference Time Text Confounding

Yuchen Ma, Dennis Frauen, Jonas Schweisthal +1

Estimating treatment effects is crucial for personalized decision-making in medicine, but this task faces unique challenges in clinical practice. At training time, models for estim…

cs.LG2025

A Diffusion-Based Method for Learning the Multi-Outcome Distribution of Medical Treatments

Yuchen Ma, Jonas Schweisthal, Hengrui Zhang +1

In medicine, treatments often influence multiple, interdependent outcomes, such as primary endpoints, complications, adverse events, or other secondary endpoints. Hence, to make op…

cs.LG2025

Foundation Models for Causal Inference via Prior-Data Fitted Networks

Yuchen Ma, Dennis Frauen, Emil Javurek +1

Prior-data fitted networks (PFNs) have recently been proposed as a promising way to train tabular foundation models. PFNs are transformers that are pre-trained on synthetic data ge…

cs.LG2024★ 1 cited

DiffPO: A causal diffusion model for learning distributions of potential outcomes

Yuchen Ma, Valentyn Melnychuk, Jonas Schweisthal +1

Predicting potential outcomes of interventions from observational data is crucial for decision-making in medicine, but the task is challenging due to the fundamental problem of cau…

cs.LG2023★ 1 cited

Consistent End-to-End Estimation for Counterfactual Fairness

Yuchen Ma, Valentyn Melnychuk, Dennis Frauen +1

Fairness in predictions is of direct importance in practice due to legal, ethical, and societal reasons. This is often accomplished through counterfactual fairness, which ensures t…