← Search

Yujia Zheng

32 accepted papers

2026

A2ASecBench: A Protocol-Aware Security Benchmark for Agent-to-Agent Multi-Agent Systems

ICLR 2026poster

Multi-agent systems (MAS) built on large language models (LLMs) increasingly rely on agent-to-agent (A2A) protocols to enable capability discovery, task orchestration, and artifact exchange across heterogeneous stacks. While these protocols promise interoperability, they also introduce new vulnerabi…

Cited by 0SourcecodeScholar
2026

Ada-Diffuser: Latent-Aware Adaptive Diffusion for Decision-Making

ICLR 2026poster

Recent work has framed decision-making as a sequence modeling problem using generative models such as diffusion models. Although promising, these approaches often overlook latent factors that exhibit evolving dynamics, elements that are fundamental to environment transitions, reward structures, and…

Cited by 0SourceScholar
2026

Learning General Causal Structures with Hidden Dynamic Process for Climate Analysis

ICML 2026poster

Understanding climate dynamics requires going beyond correlations in observational data to uncover their underlying causal process. Latent drivers, such as atmospheric processes, play a critical role in temporal dynamics, while direct causal influences also exist among geographically proximate obser…

Cited by 0SourceScholar
2026

Learning Task-Sufficient World Models by Synergizing Agentic Exploration and Structured Modeling

ICML 2026poster

Learning and planning in imagination using world models provides an effective paradigm for training agents for decision-making. However, existing approaches often rely on high-dimensional latent spaces or generic visual embeddings that retain many factors irrelevant to control, limiting efficiency a…

Cited by 0SourceScholar
2026

ParamMem: Augmenting Language Agents with Parametric Reflective Memory

ICML 2026poster

Self-reflection enables language agents to iteratively refine solutions, yet often produces repetitive outputs that limit reasoning performance. Recent studies have attempted to address this limitation through various approaches, among which increasing reflective diversity has shown promise. Our emp…

Cited by 0SourceScholar
2025

A General Representation-Based Approach to Multi-Source Domain Adaptation

ICML 2025poster

A central problem in unsupervised domain adaptation is determining what to transfer from labeled source domains to an unlabeled target domain. To handle high-dimensional observations (e.g., images), a line of approaches use deep learning to learn latent representations of the observations, which fac…

Cited by 0SourcePDFScholar
2025

Butterfly Effects in Toolchains: A Comprehensive Analysis of Failed Parameter Filling in LLM Tool-Agent Systems

EMNLP 2025

The emergence of the tool agent paradigm has broadened the capability boundaries of the Large Language Model (LLM), enabling it to complete more complex tasks. However, the effectiveness of this paradigm is limited due to the issue of parameter failure during its execution. To explore this phenomeno

2025

Causal Representation Learning from Multimodal Biomedical Observations

ICLR 2025poster

Prevalent in biomedical applications (e.g., human phenotype research), multimodal datasets can provide valuable insights into the underlying physiological mechanisms. However, current machine learning (ML) models designed to analyze these datasets often lack interpretability and identifiability guar…

Cited by 0SourcePDFScholar
2025

LLM Interpretability with Identifiable Temporal-Instantaneous Representation

NeurIPS 2025poster

Despite Large Language Models' remarkable capabilities, understanding their internal representations remains challenging. Mechanistic interpretability tools such as sparse autoencoders (SAEs) were developed to extract interpretable features from LLMs but lack temporal dependency modeling, instantane…

Cited by 0SourceScholar
2025

Learning Vision and Language Concepts for Controllable Image Generation

ICML 2025poster

Concept learning seeks to extract semantic and interpretable representations of atomic concepts from high-dimensional data such as images and text, which can be instrumental to a variety of downstream tasks (e.g., image generation/editing). Despite its importance, the theoretical foundations for lea…

Cited by 0SourcePDFScholar
2025

SmartCLIP: Modular Vision-language Alignment with Identification Guarantees

CVPR 2025highlight

Contrastive Language-Image Pre-training (CLIP) \citep radford2021learning has emerged as a pivotal model in computer vision and multimodal learning, achieving state-of-the-art performance at aligning visual and textual representations through contrastive learning. However, CLIP struggles with poten…

2025

Synergy Between Sufficient Changes and Sparse Mixing Procedure for Disentangled Representation Learning

ICLR 2025poster

Disentangled representation learning aims to uncover the latent variables underlying observed data, yet identifying these variables under mild assumptions remains challenging. Some methods rely on sufficient changes in the distribution of latent variables indicated by auxiliary variables, such as do…

2025

Thought Communication in Multiagent Collaboration

NeurIPS 2025spotlight

Natural language has long enabled human cooperation, but its lossy, ambiguous, and indirect nature limits the potential of collective intelligence. While machines are not subject to these constraints, most LLM-based multi-agent systems still rely solely on natural language, exchanging tokens or thei…

Cited by 0SourceScholar
2025

Type Information-Assisted Self-Supervised Knowledge Graph Denoising

AISTATS 2025poster

Knowledge graphs serve as critical resources supporting intelligent systems, but they can be noisy due to imperfect automatic generation processes. Existing approaches to noise detection often rely on external facts, logical rule constraints, or structural embeddings. These methods are often challen…

Cited by 0SourcecodeScholar
2024

A Versatile Causal Discovery Framework to Allow Causally-Related Hidden Variables

ICLR 2024poster

Most existing causal discovery methods rely on the assumption of no latent confounders, limiting their applicability in solving real-life problems. In this paper, we introduce a novel, versatile framework for causal discovery that accommodates the presence of causally-related hidden variables almost…

Cited by 17SourcePDFScholar
2024

Causal Representation Learning from Multiple Distributions: A General Setting

ICML 2024poster

In many problems, the measured variables (e.g., image pixels) are just mathematical functions of the latent causal variables (e.g., the underlying concepts or objects). For the purpose of making predictions in changing environments or making proper changes to the system, it is helpful to recover the…

Cited by 23SourcePDFScholar
2024

Causal Temporal Representation Learning with Nonstationary Sparse Transition

NeurIPS 2024poster

Causal Temporal Representation Learning (Ctrl) methods aim to identify the temporal causal dynamics of complex nonstationary temporal sequences. Despite the success of existing Ctrl methods, they require either directly observing the domain variables or assuming a Markov prior on them. Such requirem…

2024

Detecting and Identifying Selection Structure in Sequential Data

ICML 2024poster

We argue that the selective inclusion of data points based on latent objectives is common in practical situations, such as music sequences. Since this selection process often distorts statistical analysis, previous work primarily views it as a bias to be corrected and proposes various methods to mit…

Cited by 4SourcePDFScholar
2024

Local Causal Discovery with Linear non-Gaussian Cyclic Models

AISTATS 2024poster

Local causal discovery is of great practical significance, as there are often situations where the discovery of the global causal structure is unnecessary, and the interest lies solely on a single target variable. Most existing local methods utilize conditional independence relations, providing only…

2023

Generalized Precision Matrix for Scalable Estimation of Nonparametric Markov Networks

ICLR 2023poster

A Markov network characterizes the conditional independence structure, or Markov property, among a set of random variables. Existing work focuses on specific families of distributions (e.g., exponential families) and/or certain structures of graphs, and most of them can only handle variables of a si…

Cited by 5SourcePDFScholar
2023

On the Identifiability of Sparse ICA without Assuming Non-Gaussianity

NeurIPS 2023poster

Independent component analysis (ICA) is a fundamental statistical tool used to reveal hidden generative processes from observed data. However, traditional ICA approaches struggle with the rotational invariance inherent in Gaussian distributions, often necessitating the assumption of non-Gaussianity…

Cited by 6SourcePDFScholar
2022

Partial disentanglement for domain adaptation

ICML 2022spotlight

Unsupervised domain adaptation is critical to many real-world applications where label information is unavailable in the target domain. In general, without further assumptions, the joint distribution of the features and the label is not identifiable in the target domain. To address this issue, we re…

Cited by 81SourcePDFScholar
2021

Reliable Causal Discovery with Improved Exact Search and Weaker Assumptions

NeurIPS 2021poster

Many of the causal discovery methods rely on the faithfulness assumption to guarantee asymptotic correctness. However, the assumption can be approximately violated in many ways, leading to sub-optimal solutions. Although there is a line of research in Bayesian network structure learning that focuses…