← Search

Zhengzhang Chen

9 accepted papers

2026

MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery

AAAI 2026technical

Uncovering causal structures from observational data is crucial for understanding complex systems and making informed decisions. While reinforcement learning (RL) has shown promise in identifying these structures in the form of a directed acyclic graph (DAG), existing methods often lack efficiency,

Cited by 0SourcePDFScholar
2025

Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusion

ICML 2025spotlight

Symmetry in the parameter space of deep neural networks (DNNs) has proven beneficial for various deep learning applications. A well-known example is the permutation symmetry in Multi-Layer Perceptrons (MLPs), where permuting the rows of weight matrices in one layer and applying the inverse permutati…

2025

Exploring Multi-Modal Data with Tool-Augmented LLM Agents for Precise Causal Discovery

ACL 2025finding

Causal discovery is an imperative foundation for decision-making across domains, such as smart health, AI for drug discovery and AIOps. Traditional statistical causal discovery methods, while well-established, predominantly rely on observational data and often overlook the semantic cues inherent in…

2025

MixLLM: Dynamic Routing in Mixed Large Language Models

NAACL 2025long

Large Language Models (LLMs) exhibit potential artificial generic intelligence recently, however, their usage is costly with high response latency. Given mixed LLMs with their own strengths and weaknesses, LLM routing aims to identify the most suitable model for each query in the stream to maximize…

Cited by 4SourcePDFScholar
2025

SolverLLM: Leveraging Test-Time Scaling for Optimization Problem via LLM-Guided Search

NeurIPS 2025poster

Large Language Models (LLMs) offer promising capabilities for tackling complex reasoning tasks, including optimization problems. However, existing methods either rely on prompt engineering, which leads to poor generalization across problem types, or require costly supervised training. We introduce S…

Cited by 0SourceScholar
2023

Hierarchical Gaussian Mixture based Task Generative Model for Robust Meta-Learning

NeurIPS 2023poster

Meta-learning enables quick adaptation of machine learning models to new tasks with limited data. While tasks could come from varying distributions in reality, most of the existing meta-learning methods consider both training and testing tasks as from the same uni-component distribution, overlooking…

Cited by 1SourcePDFScholar
2022

Superclass-Conditional Gaussian Mixture Model For Learning Fine-Grained Embeddings

ICLR 2022spotlight

Learning fine-grained embeddings is essential for extending the generalizability of models pre-trained on "coarse" labels (e.g., animals). It is crucial to fields for which fine-grained labeling (e.g., breeds of animals) is expensive, but fine-grained prediction is desirable, such as medicine. The d…

2021

Dynamic Gaussian Mixture based Deep Generative Model For Robust Forecasting on Sparse Multivariate Time Series

AAAI 2021technical

Forecasting on sparse multivariate time series (MTS) aims to model the predictors of future values of time series given their incomplete past, which is important for many emerging applications. However, most existing methods process MTS’s individually, and do not leverage the dynamic distributions u…

2021

FaceSec: A Fine-Grained Robustness Evaluation Framework for Face Recognition Systems

CVPR 2021poster

We present FACESEC, a framework for fine-grained robustness evaluation of face recognition systems. FACESEC evaluation is performed along four dimensions of adversarial modeling: the nature of perturbation (e.g., pixel-level or face accessories), the attacker's system knowledge (about training data…

Cited by 27PDFcodeScholar