← Search

Yuchen Ma

7 accepted papers

2026

Foundation Models for Causal Inference via Prior-Data Fitted Networks

ICLR 2026poster

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 generated from a prespecified prior distribution and that enable Bayesian inference through in-context learning. In this p…

Cited by 30SourcecodeScholar
2026

ProbeLLM: Automating Principled Diagnosis of LLM Failures

ICML 2026poster

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 been enabled by dynamic test generation, existing approaches often discover isolated failure cases, lack principled con…

Cited by 0SourceScholar
2025

LLM-Driven Treatment Effect Estimation Under Inference Time Text Confounding

NeurIPS 2025poster

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 estimating treatment effects are typically trained on well-structured medical datasets that contain detailed patient informat…

Cited by 0SourceScholar
2024

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

NeurIPS 2024poster

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 causal inference. Existing methods are largely limited to point estimates of potential outcomes with no uncertain quantific…

Cited by 1SourcePDFScholar
2020

BorderDet: Border Feature for Dense Object Detection

ECCV 2020poster

Dense object detectors rely on the sliding-window paradigm that predicts the object over a regular grid of image. Meanwhile, the feature maps on the point of the grid are adopted to generate the bounding box predictions. The point feature is convenient to use but may lack the explicit border informa…