← Search

Kun Zhang

236 accepted papers

2026

AISPO: Enhancing Depth Reliability for Robotic Manipulation of Non-Lambertian Objects via Affine-Invariant Shape Prior

RA-L 2026

Reliable depth perception is critical for robotic manipulation, especially for non-Lambertian objects such as transparent or highly specular surfaces, where raw depth measurements are often corrupted or missing. These failures frequently propagate to motion planning, resulting in invalid grasp poses

Cited by 0SourceScholar
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

Beyond DAGs: A Latent Partial Causal Model for Multimodal Learning

ICLR 2026poster

Directed Acyclic Graphs (DAGs) are a standard tool in causal modeling, but their suitability for capturing the complexity of large-scale multimodal data is questionable. In practice, real-world multimodal datasets are often collected from heterogeneous generative processes that do not conform to a s…

Cited by 0SourceScholar
2026

CausalGame: Benchmarking Causal Thinking of LLM Agents in Games

ICML 2026oral

Recently, it has received growing attention in building AI Scientist agents with Large Language Models (LLMs). Since scientific discovery fundamentally relies on uncovering causal relationships from observations, the capability of causal thinking that distinguish causation from correlation and hidde…

Cited by 0SourceScholar
2026

Characterization and Learning of Causal Graphs with Latent Confounders and Post-treatment Selection from Interventional Data

ICLR 2026poster

Interventional causal discovery seeks to identify causal relations by leveraging distributional changes introduced by interventions, even in the presence of latent confounders. Beyond the spurious dependencies induced by latent confounders, we highlight a common yet often overlooked challenge in the…

Cited by 0SourcecodeScholar
2026

ChemKGL: Bridging Knowledge Graphs and Large Language Models for Chemical Multi-Step Reaction Pathway Inference

IJCAI 2026

Large language models have shown promising potential in chemistry, with prior work exploring molecular recognition, classification, and property prediction. Despite the achieved progress, LLMs are still far from satisfactory when dealing with complex chemical multi-step reaction pathway inference ta

Cited by 0Scholar
2026

Concept Bottleneck Models for Explainable Decision Making: A Survey of Progress, Taxonomy, and Future Directions

IJCAI 2026

Deep neural networks deliver strong performance but remain opaque, limiting their use in high-stakes domains that require transparency and human oversight. Concept Bottleneck Models (CBMs) address this gap by introducing a human-interpretable concept layer that mediates inputs and decisions, enablin

Cited by 0Scholar
2026

Concept Concentration for Faithful Representation Intervention

ICML 2026poster

Representation intervention aims to locate and modify the representations that encode the underlying concepts in Large Language Models (LLMs) to elicit the aligned and expected behaviors. Despite the empirical success, it has never been examined whether one could locate the faithful concepts for int…

Cited by 0SourceScholar
2026

Conditional Independent Component Analysis For Estimating Causal Structure with Latent Variables

ICLR 2026poster

Identifying latent variables and their induced causal structure is fundamental in various scientific fields. Existing approaches often rely on restrictive structural assumptions (e.g., purity) and may become invalid when these assumptions are violated. We introduce Conditional Independent Component…

Cited by 0SourceScholar
2026

Controllable Video Generation with Provable Disentanglement

ICLR 2026poster

Controllable video generation remains a significant challenge, despite recent advances in generating high-quality and consistent videos. Most existing methods for controlling video generation treat the video as a whole, neglecting intricate fine-grained spatiotemporal relationships, which limits bot…

Cited by 0SourceScholar
2026

Distributional Equivalence in Linear Non-Gaussian Latent-Variable Cyclic Causal Models: Characterization and Learning

ICLR 2026oral

Causal discovery with latent variables is a fundamental task. Yet most existing methods rely on strong structural assumptions, such as enforcing specific indicator patterns for latents or restricting how they can interact with others. We argue that a core obstacle to a general, structural-assumption…

Cited by 0SourcecodeScholar
2026

From Reasoning Traces to Reusable Modules: Reinforcement Learning for Compositional Generalization in Language Model Reasoning

ICML 2026poster

Reinforcement learning (RL) has emerged as a key mechanism for transforming LLMs into robust reasoners. While supervised fine-tuning (SFT) often limits models to the distribution of observed reasoning traces, RL post-training significantly improves performance on out-of-distribution (OOD) tasks that…

Cited by 0SourceScholar
2026

Horizontal and Vertical Federated Causal Structure Learning via Higher-order Cumulants

AAAI 2026technical

Federated causal discovery aims to uncover causal relationships while protecting data privacy, with significant real-world applications. Existing methods focus on horizontal federated settings where clients share the same variables but have different samples. However, in practice, clients may have d

Cited by 0SourcePDFScholar
2026

Identifying Partially Observed Causal Models from Heterogeneous/Nonstationary Data

ICML 2026poster

Estimating causal structure in the presence of latent variables is an important yet challenging problem. Recent works have shown that distributional constraints, such as rank deficiency constraints of the covariance matrices, can be exploited to recover the underlying causal structure involving late…

Cited by 0SourceScholar
2026

Independence Test for Linear Non-Gaussian Data and Applications in Causal Discovery

ICLR 2026poster

Independence testing involves determining whether two variables are independent based on observed samples, which is a fundamental problem in statistics and machine learning. Existing testing methods, such as HSIC, can theoretically detect broad forms of dependence, but may sacrifice statistical powe…

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

Learning by Analogy: A Causal Framework for Compositional Generalization

CVPR 2026

Compositional generalization -- the ability to understand and generate novel combinations of learned concepts -- enables models to extend their capabilities beyond limited experiences. While effective, the data structures and principles that enable this crucial capability remain poorly understood. W

Cited by 0SourceScholar
2026

Local Covariate Selection for Average Causal Effect Estimation without Pretreatment and Causal Sufficiency Assumptions

ICML 2026spotlight

Causal effect estimation is a fundamental task in many scientific fields. Selecting appropriate covariates for adjustment is crucial for obtaining unbiased causal effects. However, most existing methods either rely on learning the global causal structure, assume the absence of latent variables, or i…

Cited by 0SourceScholar
2026

On the Thinking-Language Modeling Gap in Large Language Models

ICLR 2026poster

Large Language Models (LLMs) demonstrate remarkable capabilities in solving complicated reasoning tasks by imitating the human thinking process from human languages. However, even the most capable LLMs can still fail in tasks that are simple for humans. To understand the gap, we construct structural…

Cited by 0SourcecodeScholar
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
2026

PersonaX: Multimodal Datasets with LLM-Inferred Behavior Traits

ICLR 2026poster

Understanding human behavior traits is central to applications in human-computer interaction, computational social science, and personalized AI systems. Such understanding often requires integrating multiple modalities to capture nuanced patterns and relationships. However, existing resources rarely…

Cited by 0SourcecodeScholar
2026

Physics from Video: Identifiability of Time-Invariant Second-Order ODEs under Minimal Trajectory Conditions

ICML 2026poster

Bridging the gap between visual realism and physical understanding is a core challenge for video-based world models. We study the structural identifiability of continuous-time physical laws from raw pixels, focusing on whether an encoder-only pipeline can uniquely recover the parameters of second-or…

Cited by 0SourceScholar
2026

Position: Beyond Sensitive Attributes, ML Fairness Should Quantify Structural Injustice via Social Determinants

ICML 2026poster

Algorithmic fairness research has largely framed _unfairness as discrimination_ along _sensitive attributes_. However, this approach limits visibility into _unfairness as structural injustice_ instantiated through _social determinants_, which are contextual variables that shape attributes and outcom…

Cited by 0SourceScholar
2026

Revisiting Differentiable Structure Learning: Inconsistency of L1 Penalty and Beyond

AAAI 2026technical

Recent advances in differentiable structure learning have framed the combinatorial problem of learning directed acyclic graphs as a continuous optimization problem. Various aspects, including data standardization, have been studied to identify factors that influence the empirical performance of thes

Cited by 0SourcePDFScholar
2026

Scaling the Long Video Understanding of Multimodal Large Language Models via Visual Memory Mechanism

CVPR 2026

Long video understanding is a key challenge that plagues the advancement of Multimodal Large language Models (MLLMs). In this paper, we study this problem from the perspective of visual memory mechanism, and proposed a novel and training-free approach, termed Flexible Memory (FlexMem). In principle,

Cited by 0SourcecodeScholar
2026

Score-based Greedy Search for Structure Identification of Partially Observed Linear Causal Models

ICLR 2026poster

Identifying the structure of a partially observed causal system is essential to various scientific fields. Recent advances have focused on constraint-based causal discovery to solve this problem, and yet in practice these methods often face challenges related to multiple testing and error propagatio…

Cited by 0SourceScholar
2026

Selection, Reflection and Self-Refinement: Revisit Reasoning Tasks via a Causal Lens

ICLR 2026poster

Due to their inherent complexity, reasoning tasks have long been regarded as rigorous benchmarks for assessing the capabilities of machine learning models, especially large language models (LLMs). Although humans can solve these tasks with ease, existing models, even after extensive pre-training and…

Cited by 0SourcecodeScholar
2026

Socrates or Smartypants: Testing Logic Reasoning Capabilities of Large Language Models with Logic Programming-Based Test Oracles

AAAI 2026technical

Large Language Models (LLMs) have achieved significant progress in language understanding and reasoning. Evaluating and analyzing their logical reasoning abilities has therefore become essential. However, existing datasets and benchmarks are often limited to overly simplistic, unnatural, or contextu

Cited by 0SourcePDFScholar
2026

TRACE: Trajectory Recovery for Continuous Mechanism Evolution in Causal Representation Learning

ICML 2026poster

Temporal causal representation learning methods assume that causal mechanisms switch instantaneously between discrete domains, yet real-world systems often exhibit *continuous* mechanism transitions. For example, a vehicle's dynamics evolve gradually through a turning maneuver, and human gait shifts…

Cited by 0SourceScholar
2026

Treatment Responder Classification with Abstention

ICML 2026spotlight

Treatment responder classification seeks to learn a rule to classify individuals who will benefit from the treatment. This paper studies a new scenario in treatment responder classification when abstention is allowed, i.e., practitioners can opt out of making uncertain classification on some individ…

Cited by 0SourceScholar
2025

A Conditional Independence Test in the Presence of Discretization

ICLR 2025poster

Testing conditional independence (CI) has many important applications, such as Bayesian network learning and causal discovery. Although several approaches have been developed for learning CI structures for observed variables, those existing methods generally fail to work when the variables of intere…

2025

A General Knowledge Injection Framework for ICD Coding

ACL 2025finding

ICD Coding aims to assign a wide range of medical codes to a medical text document, which is a popular and challenging task in the healthcare domain. To alleviate the problems of long-tail distribution and the lack of annotations of code-specific evidence, many previous works have proposed incorpora…

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

A Robust Method to Discover Causal or Anticausal Relation

ICLR 2025poster

Understanding whether the data generative process follows causal or anticausal relations is important for many applications. Existing causal discovery methods struggle with high-dimensional perceptual data such as images. Moreover, they require well-labeled data, which may not be feasible due to mea…

Cited by 0SourcePDFScholar
2025

A Sample Efficient Conditional Independence Test in the Presence of Discretization

ICML 2025poster

Conditional independence (CI) test is a fundamental concept in statistics. In many real-world scenarios, some variables may be difficult to measure accurately, often leading to data being represented as discretized values. Applying CI tests directly to discretized data, however, can lead to incorrec…

2025

A Skewness-Based Criterion for Addressing Heteroscedastic Noise in Causal Discovery

ICLR 2025poster

Real-world data often violates the equal-variance assumption (homoscedasticity), making it essential to account for heteroscedastic noise in causal discovery. In this work, we explore heteroscedastic symmetric noise models (HSNMs), where the effect $Y$ is modeled as $Y = f(X) + \sigma(X)N$, with $X$…

Cited by 0SourcePDFScholar
2025

Analytic DAG Constraints for Differentiable DAG Learning

ICLR 2025poster

Recovering the underlying Directed Acyclic Graph (DAG) structures from observational data presents a formidable challenge, partly due to the combinatorial nature of the DAG-constrained optimization problem. Recently, researchers have identified gradient vanishing as one of the primary obstac…

2025

Causal Graph Transformer for Treatment Effect Estimation Under Unknown Interference

ICLR 2025poster

Networked interference, also known as the peer effect in social science and spillover effect in economics, has drawn increasing interest across various domains. This phenomenon arises when a unit’s treatment and outcome are influenced by the actions of its peers, posing significant challenges to cau…

2025

Causal Representation Learning from General Environments under Nonparametric Mixing

AISTATS 2025poster

Causal representation learning aims to recover the latent causal variables and their causal relations, typically represented by directed acyclic graphs (DAGs), from low-level observations such as image pixels. A prevailing line of research exploits multiple environments, which assume how data distri…

Cited by 0SourceScholar
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

CausalVerse: Benchmarking Causal Representation Learning with Configurable High-Fidelity Simulations

NeurIPS 2025spotlight

Causal Representation Learning (CRL) aims to uncover the data-generating process and identify the underlying causal variables and relations, whose evaluation remains inherently challenging due to the requirement of known ground-truth causal variables and causal structure. Existing evaluations often…

Cited by 0SourcecodeScholar
2025

Continual Unsupervised Generative Modelling via Online Optimal Transport

AAAI 2025technical

Lately, deep generative models have achieved excellent results after learning pre-defined and static data distribution. Meanwhile, their performance on continual learning suffers from degeneration, caused by catastrophic forgetting. In this paper, we study the unsupervised generative modelling in a…

2025

DH-Set: Improving Vision-Language Alignment with Diverse and Hybrid Set-Embeddings Learning

CVPR 2025poster

Vision-Language (VL) alignment across image and text modalities is a challenging task due to the inherent semantic ambiguity of data with multiple possible meanings. Existing methods typically solve it by learning multiple sub-representation spaces to encode each input data as a set of embeddings, a…

Cited by 0SourcePDFScholar
2025

Detecting Generated Images by Fitting Natural Image Distributions

NeurIPS 2025spotlight

The increasing realism of generated images has raised significant concerns about their potential misuse, necessitating robust detection methods. Current approaches mainly rely on training binary classifiers, which depend heavily on the quantity and quality of available generated images. In this work…

Cited by 0SourceScholar
2025

Differentiable Causal Discovery for Latent Hierarchical Causal Models

ICLR 2025poster

Discovering causal structures with latent variables from observational data is a fundamental challenge in causal discovery. Existing methods often rely on constraint-based, iterative discrete searches, limiting their scalability for large numbers of variables. Moreover, these methods frequently assu…

Cited by 0SourcePDFScholar
2025

Dynamic Expansion Diffusion Learning for Lifelong Generative Modelling

AAAI 2025technical

The diffusion model has lately been shown to achieve remarkable performances through its ability of generating high quality images. However, current diffusion model studies consider only learning from a single data distribution, resulting in catastrophic forgetting when attempting to learn new data.…

2025

Empowering LLMs with Logical Reasoning: A Comprehensive Survey

IJCAI 2025

Large language models (LLMs) have achieved remarkable successes on various tasks. However, recent studies have found that there are still significant challenges to the logical reasoning abilities of LLMs, which can be categorized into the following two aspects: (1) Logical question answering: LLMs o

Cited by 0SourcePDFScholar
2025

Equivalence of Closed Chains to Open Chains: Virtual Decomposition Control Combined With Adaptive RBF Neural Network for Hydraulic Robot Legs

RA-L 2025

The joints of the hydraulic robot, driven by linear cylinders, form triangular closed-chain structures composed of the cylinders and passive rotational joints. This configuration complicates the complete dynamic modeling and increases the system's nonlinearity. To simplify the modeling process, conv

Cited by 1SourceScholar
2025

Extracting Rare Dependence Patterns via Adaptive Sample Reweighting

ICML 2025poster

Discovering dependence patterns between variables from observational data is a fundamental issue in data analysis. However, existing testing methods often fail to detect subtle yet critical patterns that occur within small regions of the data distribution--patterns we term rare dependence. These rar…

Cited by 0SourcePDFScholar
2025

Fairness on Principal Stratum: A New Perspective on Counterfactual Fairness

ICML 2025poster

Fairness in human and algorithmic decision-making is crucial in areas such as criminal justice, education, and social welfare. Recently, counterfactual fairness has drawn increasing research interest, suggesting that decision-making for individuals should remain the same when intervening with differ…

Cited by 0SourcePDFScholar
2025

Flow: Modularized Agentic Workflow Automation

ICLR 2025poster

Multi-agent frameworks powered by large language models (LLMs) have demonstrated great success in automated planning and task execution. However, the effective adjustment of agentic workflows during execution has not been well studied. An effective workflow adjustment is crucial in real-world scenar…

2025

Gene Regulatory Network Inference in the Presence of Selection Bias and Latent Confounders

NeurIPS 2025poster

Gene regulatory network inference (GRNI) aims to discover how genes causally regulate each other from gene expression data. It is well-known that statistical dependencies in observed data do not necessarily imply causation, as spurious dependencies may arise from *latent confounders*, such as non-co…

Cited by 0SourceScholar
2025

Hierarchy-Aware Pseudo Word Learning with Text Adaptation for Zero-Shot Composed Image Retrieval

ICCV 2025poster

Zero-Shot Composed Image Retrieval (ZS-CIR) aims to retrieve the target image based on a reference image and a text describing the user's intention without training on the triplet datasets. The key to this task is to make specified changes to specific objects in the reference image based on the text…

Cited by 0SourcePDFScholar
2025

Identification of Latent Confounders via Investigating the Tensor Ranks of the Nonlinear Observations

ICML 2025poster

We study the problem of learning discrete latent variable causal structures from mixed-type observational data. Traditional methods, such as those based on the tensor rank condition, are designed to identify discrete latent structure models and provide robust identification bounds for discrete causa…

Cited by 0SourcePDFScholar
2025

Infant Cry Detection Using Causal Temporal Representation

ICASSP 2025accepted

This paper addresses a major challenge in acoustic event detection, in particular infant cry detection in the presence of other sounds and background noises: the lack of precise annotated data. We present two contributions for supervised and unsupervised infant cry detection. The first is an annotat…

Cited by 0SourceScholar
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

Latent Variable Causal Discovery under Selection Bias

ICML 2025poster

Addressing selection bias in latent variable causal discovery is important yet underexplored, largely due to a lack of suitable statistical tools: While various tools beyond basic conditional independencies have been developed to handle latent variables, none have been adapted for selection bias. We…

Cited by 0SourcePDFScholar
2025

Learning Counterfactual Outcomes Under Rank Preservation

NeurIPS 2025poster

Counterfactual inference aims to estimate the counterfactual outcome at the individual level given knowledge of an observed treatment and the factual outcome, with broad applications in fields such as epidemiology, econometrics, and management science. Previous methods rely on a known structural cau…

Cited by 0SourceScholar
2025

Learning Graph Invariance by Harnessing Spuriosity

ICLR 2025poster

Recently, graph invariant learning has become the _de facto_ approach to tackle the Out-of-Distribution (OOD) generalization failure in graph representation learning. They generically follow the framework of invariant risk minimization to capture the invariance of graph data from different environme…

Cited by 0SourcePDFScholar
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

MVP-CBM: Multi-layer Visual Preference-enhanced Concept Bottleneck Model for Explainable Medical Image Classification

IJCAI 2025

The concept bottleneck model (CBM), as a technique improving interpretability via linking predictions to human-understandable concepts, makes high-risk and life-critical medical image classification credible. Typically, existing CBM methods associate the final layer of visual encoders with concepts

2025

MoRE: A Mixture of Low-Rank Experts for Adaptive Multi-Task Learning

ACL 2025finding

With the rapid development of Large Language Models (LLMs), Parameter-Efficient Fine-Tuning (PEFT) methods have gained significant attention, which aims to achieve efficient fine-tuning of LLMs with fewer parameters. As a representative PEFT method, Low-Rank Adaptation (LoRA) introduces low-rank mat…

2025

Noisy Test-Time Adaptation in Vision-Language Models

ICLR 2025poster

Test-time adaptation (TTA) aims to address distribution shifts between source and target data by relying solely on target data during testing. In open-world scenarios, models often encounter noisy samples, i.e., samples outside the in-distribution (ID) label space. Leveraging the zero-shot capabilit…

2025

OCRT: Boosting Foundation Models in the Open World with Object-Concept-Relation Triad

CVPR 2025poster

Although foundation models (FMs) claim to be powerful, their generalization ability significantly decreases when faced with distribution shifts, weak supervision, or malicious attacks in the open world. On the other hand, most domain generalization or adversarial fine-tuning methods are task-related…

2025

On the Identification of Temporal Causal Representation with Instantaneous Dependence

ICLR 2025oral

Temporally causal representation learning aims to identify the latent causal process from time series observations, but most methods require the assumption that the latent causal processes do not have instantaneous relations. Although some recent methods achieve identifiability in the instantaneous…

Cited by 6SourcePDFScholar
2025

Online Time Series Forecasting with Theoretical Guarantees

NeurIPS 2025poster

This paper is concerned with online time series forecasting, where unknown distribution shifts occur over time, i.e., latent variables influence the mapping from historical to future observations. To develop an automated way of online time series forecasting, we propose a Theoretical framework for O…

Cited by 0SourceScholar
2025

Permutation-based Rank Test in the Presence of Discretization and Application in Causal Discovery with Mixed Data

ICML 2025poster

Recent advances have shown that statistical tests for the rank of cross-covariance matrices play an important role in causal discovery. These rank tests include partial correlation tests as special cases and provide further graphical information about latent variables. Existing rank tests typically…

2025

PerturBench: Benchmarking Machine Learning Models for Cellular Perturbation Analysis

NeurIPS 2025poster

We introduce a comprehensive framework for modeling single cell transcriptomic responses to perturbations, aimed at standardizing benchmarking in this rapidly evolving field. Our approach includes a modular and user-friendly model development and evaluation platform, a collection of diverse perturba…

Cited by 0SourcecodeScholar
2025

Practical Kernel Selection for Kernel-based Conditional Independence Test

NeurIPS 2025poster

Conditional independence (CI) testing is a fundamental yet challenging task in modern statistics and machine learning. One pivotal class of methods for assessing conditional independence encompasses kernel-based approaches, known for assessing CI by detecting general conditional dependence without…

Cited by 0SourceScholar
2025

Prompting Fairness: Integrating Causality to Debias Large Language Models

ICLR 2025poster

Large language models (LLMs), despite their remarkable capabilities, are susceptible to generating biased and discriminatory responses. As LLMs increasingly influence high-stakes decision-making (e.g., hiring and healthcare), mitigating these biases becomes critical. In this work, we propose a causa…

Cited by 0SourcePDFScholar
2025

Reflection-Window Decoding: Text Generation with Selective Refinement

ICML 2025poster

The autoregressive decoding for text generation in large language models (LLMs), while widely used, is inherently suboptimal due to the lack of a built-in mechanism to perform refinement and/or correction of the generated content. In this paper, we consider optimality in terms of the joint probabili…

Cited by 2SourcePDFScholar
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

Structured Discourse Representation for Factual Consistency Verification

ACL 2025finding

Analysing the differences in how events are represented across texts, or verifying whether the language model generations hallucinate, requires the ability to systematically compare their content. To support such comparison, structured representation that captures fine-grained information plays a vi…

Cited by 0SourcePDFScholar
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

TIU-Bench: A Benchmark for Evaluating Large Multimodal Models on Text-rich Image Understanding

EMNLP 2025

Text-rich images are ubiquitous in real-world applications, serving as a critical medium for conveying complex information and facilitating accessibility.Despite recent advances driven by Multimodal Large Language Models (MLLMs), existing benchmarks suffer from limited scale, fragmented scenarios, a

Cited by 0SourcePDFScholar
2025

The third pillar of causal analysis? A measurement perspective on causal representations

NeurIPS 2025poster

Causal reasoning and discovery, two fundamental tasks of causal analysis, often face challenges in applications due to the complexity, noisiness, and high-dimensionality of real-world data. Despite recent progress in identifying latent causal structures using causal representation learning (CRL), wh…

Cited by 0SourceScholar
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

Towards Accurate Time Series Forecasting via Implicit Decoding

NeurIPS 2025poster

Recent booming time series models have demonstrated remarkable forecasting performance. However, these methods often place greater focus on more effectively modelling the historical series, largely neglecting the forecasting phase, which generates long-term forecasts by separately predicting multipl…

Cited by 0SourcecodeScholar
2025

Towards Identifiability of Hierarchical Temporal Causal Representation Learning

NeurIPS 2025poster

Modeling hierarchical latent dynamics behind time series data is critical for capturing temporal dependencies across multiple levels of abstraction in real-world tasks. However, existing temporal causal representation learning methods fail to capture such dynamics, as they fail to recover the joint…

Cited by 0SourceScholar
2025

Towards Self-Refinement of Vision-Language Models with Triangular Consistency

NeurIPS 2025poster

Vision-Language Models (VLMs) integrate visual knowledge with the analytical capabilities of Large Language Models (LLMs) through supervised visual instruction tuning, using image-question-answer triplets. However, the potential of VLMs trained without supervised instruction remains largely unexplor…

Cited by 0SourcecodeScholar
2025

Transformer-Based Multi-Agent Reinforcement Learning Method With Credit-Oriented Strategy Differentiation

IROS 2025

The problem of Multi-Agent Reinforcement Learning (MARL) shows a high level of both complexity in the environment and coordination between agents. In order to scale the algorithm to large-scale agent scenarios, neural networks designed for MARL are typically implemented with parameter sharing. These

Cited by 0SourcecodeScholar
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
2025

Unmasking Style Sensitivity: A Causal Analysis of Bias Evaluation Instability in Large Language Models

ACL 2025long

Natural language processing applications are increasingly prevalent, but social biases in their outputs remain a critical challenge. While various bias evaluation methods have been proposed, these assessments show unexpected instability when input texts undergo minor stylistic changes. This paper co…

Cited by 0SourcePDFScholar
2025

When Selection Meets Intervention: Additional Complexities in Causal Discovery

ICLR 2025oral

We address the common yet often-overlooked selection bias in interventional studies, where subjects are selectively enrolled into experiments. For instance, participants in a drug trial are usually patients of the relevant disease; A/B tests on mobile applications target existing users only, and gen…

2024

A Local Method for Satisfying Interventional Fairness with Partially Known Causal Graphs

NeurIPS 2024poster

Developing fair automated machine learning algorithms is critical in making safe and trustworthy decisions. Many causality-based fairness notions have been proposed to address the above issues by quantifying the causal connections between sensitive attributes and decisions, and when the true causal…

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

ACAMDA: Improving Data Efficiency in Reinforcement Learning through Guided Counterfactual Data Augmentation

AAAI 2024technical

Data augmentation plays a crucial role in improving the data efficiency of reinforcement learning (RL). However, the generation of high-quality augmented data remains a significant challenge. To overcome this, we introduce ACAMDA (Adversarial Causal Modeling for Data Augmentation), a novel framework…

Cited by 6SourcePDFScholar
2024

CaRiNG: Learning Temporal Causal Representation under Non-Invertible Generation Process

ICML 2024poster

Identifying the underlying time-delayed latent causal processes in sequential data is vital for grasping temporal dynamics and making downstream reasoning. While some recent methods can robustly identify these latent causal variables, they rely on strict assumptions about the invertible generation p…

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 Structure Recovery with Latent Variables under Milder Distributional and Graphical Assumptions

ICLR 2024poster

Traditional causal discovery approaches typically assume the absence of latent variables, a simplification that often does not align with real-world situations. Recently, there has been a surge of causal discovery methods that explicitly consider latent variables. While some works aim to reveal caus…

Cited by 6SourcePDFScholar
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

Discovery of the Hidden World with Large Language Models

NeurIPS 2024poster

Revealing the underlying causal mechanisms in the real world is the key to the development of science. Despite the progress in the past decades, traditional causal discovery approaches (CDs) mainly rely on high-quality measured variables, usually given by human experts, to find causal relations. The…

Cited by 0SourcePDFScholar
2024

Empowering Graph Invariance Learning with Deep Spurious Infomax

ICML 2024poster

Recently, there has been a surge of interest in developing graph neural networks that utilize the invariance principle on graphs to generalize the out-of-distribution (OOD) data. Due to the limited knowledge about OOD data, existing approaches often pose assumptions about the correlation strengths o…

2024

Federated Causal Discovery from Heterogeneous Data

ICLR 2024poster

Conventional causal discovery methods rely on centralized data, which is inconsistent with the decentralized nature of data in many real-world situations. This discrepancy has motivated the development of federated causal discovery (FCD) approaches. However, existing FCD methods may be limited by th…

2024

Gene Regulatory Network Inference in the Presence of Dropouts: a Causal View

ICLR 2024oral

Gene regulatory network inference (GRNI) is a challenging problem, particularly owing to the presence of zeros in single-cell RNA sequencing data: some are biological zeros representing no gene expression, while some others are technical zeros arising from the sequencing procedure (aka dropouts), wh…

2024

Identifiable Latent Polynomial Causal Models through the Lens of Change

ICLR 2024poster

Causal representation learning aims to unveil latent high-level causal representations from observed low-level data. One of its primary tasks is to provide reliable assurance of identifying these latent causal models, known as \textit{identifiability}. A recent breakthrough explores identifiability…

Cited by 15SourcePDFScholar
2024

Identification of Causal Structure with Latent Variables Based on Higher Order Cumulants

AAAI 2024technical

Causal discovery with latent variables is a crucial but challenging task. Despite the emergence of numerous methods aimed at addressing this challenge, they are not fully identified to the structure that two observed variables are influenced by one latent variable and there might be a directed edge…

Cited by 4SourcePDFScholar
2024

Identification of Necessary Semantic Undertakers in the Causal View for Image-Text Matching

AAAI 2024technical

Image-text matching bridges vision and language, which is a fundamental task in multimodal intelligence. Its key challenge lies in how to capture visual-semantic relevance. Fine-grained semantic interactions come from fragment alignments between image regions and text words. However, not all fragmen…

2024

Identifying Latent State-Transition Processes for Individualized Reinforcement Learning

NeurIPS 2024poster

The application of reinforcement learning (RL) involving interactions with individuals has grown significantly in recent years. These interactions, influenced by factors such as personal preferences and physiological differences, causally influence state transitions, ranging from health conditions i…

Cited by 3SourcePDFScholar
2024

LLCP: Learning Latent Causal Processes for Reasoning-based Video Question Answer

ICLR 2024poster

Current approaches to Video Question Answering (VideoQA) primarily focus on cross-modality matching, which is limited by the requirement for extensive data annotations and the insufficient capacity for causal reasoning (e.g. attributing accidents). To address these challenges, we introduce a causal…

Cited by 2SourcePDFScholar
2024

Learning Discrete Concepts in Latent Hierarchical Models

NeurIPS 2024poster

Learning concepts from natural high-dimensional data (e.g., images) holds potential in building human-aligned and interpretable machine learning models. Despite its encouraging prospect, formalization and theoretical insights into this crucial task are still lacking. In this work, we formali…

Cited by 2SourcePDFScholar
2024

Learning Discrete Latent Variable Structures with Tensor Rank Conditions

NeurIPS 2024poster

Unobserved discrete data are ubiquitous in many scientific disciplines, and how to learn the causal structure of these latent variables is crucial for uncovering data patterns. Most studies focus on the linear latent variable model or impose strict constraints on latent structures, which fail to add…

Cited by 0SourcePDFScholar
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…

2024

Natural Counterfactuals With Necessary Backtracking

NeurIPS 2024poster

Counterfactual reasoning is pivotal in human cognition and especially important for providing explanations and making decisions. While Judea Pearl's influential approach is theoretically elegant, its generation of a counterfactual scenario often requires too much deviation from the observed scenario…

2024

Neural Collapse Inspired Feature Alignment for Out-of-Distribution Generalization

NeurIPS 2024poster

The spurious correlation between the background features of the image and its label arises due to that the samples labeled with the same class in the training set often co-occurs with a specific background, which will cause the encoder to extract non-semantic features for classification, resulting i…

Cited by 1SourcePDFScholar
2024

On Causal Discovery in the Presence of Deterministic Relations

NeurIPS 2024poster

Many causal discovery methods typically rely on the assumption of independent noise, yet real-life situations often involve deterministic relationships. In these cases, observed variables are represented as deterministic functions of their parental variables without noise. When determinism is presen…

Cited by 1SourcePDFScholar
2024

On the Parameter Identifiability of Partially Observed Linear Causal Models

NeurIPS 2024poster

Linear causal models are important tools for modeling causal dependencies and yet in practice, only a subset of the variables can be observed. In this paper, we examine the parameter identifiability of these models by investigating whether the edge coefficients can be recovered given the causal str…

2024

On the Recoverability of Causal Relations from Temporally Aggregated I.I.D. Data

ICML 2024poster

We consider the effect of temporal aggregation on instantaneous (non-temporal) causal discovery in general setting. This is motivated by the observation that the true causal time lag is often considerably shorter than the observational interval. This discrepancy leads to high aggregation, causing ti…

Cited by 1SourcePDFScholar
2024

Optimal Kernel Choice for Score Function-based Causal Discovery

ICML 2024poster

Score-based methods have demonstrated their effectiveness in discovering causal relationships by scoring different causal structures based on their goodness of fit to the data. Recently, Huang et al. proposed a generalized score function that can handle general data distributions and causal relation…

Cited by 3SourcePDFScholar
2024

Procedural Fairness Through Decoupling Objectionable Data Generating Components

ICLR 2024spotlight

We reveal and address the frequently overlooked yet important issue of _disguised procedural unfairness_, namely, the potentially inadvertent alterations on the behavior of neutral (i.e., not problematic) aspects of data generating process, and/or the lack of procedural assurance of the greatest ben…

2024

S3A: Towards Realistic Zero-Shot Classification via Self Structural Semantic Alignment

AAAI 2024technical

Large-scale pre-trained Vision Language Models (VLMs) have proven effective for zero-shot classification. Despite the success, most traditional VLMs-based methods are restricted by the assumption of partial source supervision or ideal target vocabularies, which rarely satisfy the open-world scenario…

2024

Score-Based Causal Discovery of Latent Variable Causal Models

ICML 2024poster

Identifying latent variables and the causal structure involving them is essential across various scientific fields. While many existing works fall under the category of constraint-based methods (with e.g. conditional independence or rank deficiency tests), they may face empirical challenges such as…

Cited by 4SourcePDFScholar
2024

Structural Estimation of Partially Observed Linear Non-Gaussian Acyclic Model: A Practical Approach with Identifiability

ICLR 2024poster

Conventional causal discovery approaches, which seek to uncover causal relationships among measured variables, are typically fragile to the presence of latent variables. While various methods have been developed to address this confounding issue, they often rely on strong assumptions about the under…

Cited by 5SourcePDFScholar
2024

TNDF-Fusion: Implicit Truncated Neural Distance Field for LiDAR Dense Mapping and Localization in Large Urban Environments

RA-L 2024

Large-scale 3D mapping is an important task for robotics and autonomous driving. However, mobile robots and autonomous vehicles with limited hardware resources may face issues with large memory consumption. It is challenging to achieve a balance between mapping quality and memory consumption. To add

Cited by 5SourceScholar
2024

Towards Understanding Extrapolation: a Causal Lens

NeurIPS 2024poster

Canonical work handling distribution shifts typically necessitates an entire target distribution that lands inside the training distribution. However, practical scenarios often involve only a handful target samples, potentially lying outside the training support, which requires the capability of ext…

Cited by 1SourcePDFScholar
2024

Tree-of-Reasoning Question Decomposition for Complex Question Answering with Large Language Models

AAAI 2024technical

Large language models (LLMs) have recently demonstrated remarkable performance across various Natual Language Processing tasks. In the field of multi-hop reasoning, the Chain-of-thought (CoT) prompt method has emerged as a paradigm, using curated stepwise reasoning demonstrations to enhance LLM's ab…

Cited by 9SourcePDFScholar
2024

Visual-Linguistic Dependency Encoding for Image-Text Retrieval

COLING 2024main

Image-text retrieval is a fundamental task to bridge the semantic gap between natural language and vision. Recent works primarily focus on aligning textual meanings with visual appearance. However, they often overlook the semantic discrepancy caused by syntactic structure in natural language express…

2023

Calibration Matters: Tackling Maximization Bias in Large-scale Advertising Recommendation Systems

ICLR 2023poster

Calibration is defined as the ratio of the average predicted click rate to the true click rate. The optimization of calibration is essential to many online advertising recommendation systems because it directly affects the downstream bids in ads auctions and the amount of money charged to advertiser…

2023

Causal Balancing for Domain Generalization

ICLR 2023poster

While machine learning models rapidly advance the state-of-the-art on various real-world tasks, out-of-domain (OOD) generalization remains a challenging problem given the vulnerability of these models to spurious correlations. We propose a balanced mini-batch sampling strategy to transform a biased…

2023

Causal Discovery with Latent Confounders Based on Higher-Order Cumulants

ICML 2023poster

Causal discovery with latent confounders is an important but challenging task in many scientific areas. Despite the success of some overcomplete independent component analysis (OICA) based methods in certain domains, they are computationally expensive and can easily get stuck into local optima. We n…

Cited by 18SourcePDFScholar
2023

Counterfactual Generation with Identifiability Guarantees

NeurIPS 2023poster

Counterfactual generation lies at the core of various machine learning tasks, including image translation and controllable text generation. This generation process usually requires the identification of the disentangled latent representations, such as content and style, that underlie the observed da…

2023

DORF: A Dynamic Object Removal Framework for Robust Static LiDAR Mapping in Urban Environments

RA-L 2023

3D point cloud maps are widely used in robotic tasks like localization and planning. However, dynamic objects, such as cars and pedestrians, can introduce ghost artifacts during the map generation process, leading to reduced map quality and hindering normal robot navigation. Online dynamic object re

Cited by 14SourceScholar
2023

Disentangling Cognitive Diagnosis with Limited Exercise Labels

NeurIPS 2023poster

Cognitive diagnosis is an important task in intelligence education, which aims at measuring students’ proficiency in specific knowledge concepts. Given a fully labeled exercise-concept matrix, most existing models focused on mining students' response records for cognitive diagnosis. Despite their su…

2023

Evolving Semantic Prototype Improves Generative Zero-Shot Learning

ICML 2023poster

In zero-shot learning (ZSL), generative methods synthesize class-related sample features based on predefined semantic prototypes. They advance the ZSL performance by synthesizing unseen class sample features for better training the classifier. We observe that each class's predefined semantic prototy…

Cited by 22SourcePDFScholar
2023

FactSpotter: Evaluating the Factual Faithfulness of Graph-to-Text Generation

EMNLP 2023long findings

Graph-to-text (G2T) generation takes a graph as input and aims to generate a fluent and faith- ful textual representation of the information in the graph. The task has many applications, such as dialogue generation and question an- swering. In this work, we investigate to what extent the G2T generat…

Cited by 0SourcecodeScholar
2023

Fair Representation Learning for Recommendation: A Mutual Information Perspective

AAAI 2023technical

Recommender systems have been widely used in recent years. By exploiting historical user-item interactions, recommender systems can model personalized potential interests of users and have been widely applied to a wide range of scenarios. Despite their impressive performance, most of them may be sub…

Cited by 20SourcePDFScholar
2023

Feature Expansion for Graph Neural Networks

ICML 2023poster

Graph neural networks aim to learn representations for graph-structured data and show impressive performance in node classification. Recently, many methods have studied the representations of GNNs from the perspective of optimization goals and spectral graph theory. However, the feature space that d…

2023

GAIN: On the Generalization of Instructional Action Understanding

ICLR 2023poster

Despite the great success achieved in instructional action understanding by deep learning and mountainous data, deploying trained models to the unseen environment still remains a great challenge, since it requires strong generalizability of models from in-distribution training data to out-of-distrib…

Cited by 6SourcePDFScholar
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

Identification of Nonlinear Latent Hierarchical Models

NeurIPS 2023poster

Identifying latent variables and causal structures from observational data is essential to many real-world applications involving biological data, medical data, and unstructured data such as images and languages. However, this task can be highly challenging, especially when observed variables are ge…

Cited by 19SourcePDFScholar
2023

Keyword-Specific Acoustic Model Pruning for Open-Vocabulary Keyword Spotting

ICASSP 2023accepted

The open-vocabulary KWS system allows users to customize wake words, but its application is limited by the model size. In this paper, we design a dynamic acoustic model with input-dependent parameters. We find that acoustic frames with similar pronunciation generate similar subnetworks, and differen…

Cited by 0SourceScholar
2023

Learning World Models with Identifiable Factorization

NeurIPS 2023poster

Extracting a stable and compact representation of the environment is crucial for efficient reinforcement learning in high-dimensional, noisy, and non-stationary environments. Different categories of information coexist in such environments -- how to effectively extract and disentangle the informati…

Cited by 16SourcePDFScholar
2023

Measuring the Privacy Leakage via Graph Reconstruction Attacks on Simplicial Neural Networks (Student Abstract)

AAAI 2023technical

In this paper, we measure the privacy leakage via studying whether graph representations can be inverted to recover the graph used to generate them via graph reconstruction attack (GRA). We propose a GRA that recovers a graph's adjacency matrix from the representations via a graph decoder that minim…

Cited by 4SourcePDFScholar
2023

Multi-domain image generation and translation with identifiability guarantees

ICLR 2023top-25%

Multi-domain image generation and unpaired image-to-to-image translation are two important and related computer vision problems. The common technique for the two tasks is the learning of a joint distribution from multiple marginal distributions. However, it is well known that there can be infinitely…

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
2023

PLOT: Prompt Learning with Optimal Transport for Vision-Language Models

ICLR 2023top-25%

With the increasing attention to large vision-language models such as CLIP, there has been a significant amount of effort dedicated to building efficient prompts. Unlike conventional methods of only learning one single prompt, we propose to learn multiple comprehensive prompts to describe diverse ch…

2023

POMDP-Guided Active Force-Based Search for Robotic Insertion

IROS 2023poster

In robotic insertion tasks where the uncertainty exceeds the allowable tolerance, a good search strategy is essential for successful insertion and significantly influences efficiency. The commonly used blind search method is time-consuming and does not exploit the rich contact information. In this p…

Cited by 3SourceScholar
2023

ReFSQL: A Retrieval-Augmentation Framework for Text-to-SQL Generation

EMNLP 2023long findings

Text-to-SQL is the task that aims at translating natural language questions into SQL queries. Existing methods directly align the natural language with SQL Language and train one encoder-decoder-based model to fit all questions. However, they underestimate the inherent structural characteristics of…

Cited by 0SourceScholar
2023

SmartBrush: Text and Shape Guided Object Inpainting With Diffusion Model

CVPR 2023highlight

Generic image inpainting aims to complete a corrupted image by borrowing surrounding information, which barely generates novel content. By contrast, multi-modal inpainting provides more flexible and useful controls on the inpainted content, e.g., a text prompt can be used to describe an object with…

2023

Subspace Identification for Multi-Source Domain Adaptation

NeurIPS 2023spotlight

Multi-source domain adaptation (MSDA) methods aim to transfer knowledge from multiple labeled source domains to an unlabeled target domain. Although current methods achieve target joint distribution identifiability by enforcing minimal changes across domains, they often necessitate stringent conditi…

2023

Tem-Adapter: Adapting Image-Text Pretraining for Video Question Answer

ICCV 2023poster

Video-language pre-trained models have shown remarkable success in guiding video question-answering (VideoQA) tasks. However, due to the length of video sequences, training large-scale video-based models incurs considerably higher costs than training image-based ones. This motivates us to leverage t…

Cited by 17PDFcodeScholar
2023

Temporally Disentangled Representation Learning under Unknown Nonstationarity

NeurIPS 2023poster

In unsupervised causal representation learning for sequential data with time-delayed latent causal influences, strong identifiability results for the disentanglement of causally-related latent variables have been established in stationary settings by leveraging temporal structure. However, in nonsta…

2023

Tier Balancing: Towards Dynamic Fairness over Underlying Causal Factors

ICLR 2023poster

The pursuit of long-term fairness involves the interplay between decision-making and the underlying data generating process. In this paper, through causal modeling with a directed acyclic graph (DAG) on the decision-distribution interplay, we investigate the possibility of achieving long-term fairne…

2023

Uncertainty Guided Label Denoising for Document-level Distant Relation Extraction

ACL 2023long

Document-level relation extraction (DocRE) aims to infer complex semantic relations among entities in a document. Distant supervision (DS) is able to generate massive auto-labeled data, which can improve DocRE performance. Recent works leverage pseudo labels generated by the pre-denoising model to r…

2023

Understanding Masked Autoencoders via Hierarchical Latent Variable Models

CVPR 2023highlight

Masked autoencoder (MAE), a simple and effective self-supervised learning framework based on the reconstruction of masked image regions, has recently achieved prominent success in a variety of vision tasks. Despite the emergence of intriguing empirical observations on MAE, a theoretically principled…

2023

Unpaired Image-to-Image Translation With Shortest Path Regularization

CVPR 2023poster

Unpaired image-to-image translation aims to learn proper mappings that can map images from one domain to another domain while preserving the content of the input image. However, with large enough capacities, the network can learn to map the inputs to any random permutation of images in another domai…

2023

Unsupervised Sampling Promoting for Stochastic Human Trajectory Prediction

CVPR 2023poster

The indeterminate nature of human motion requires trajectory prediction systems to use a probabilistic model to formulate the multi-modality phenomenon and infer a finite set of future trajectories. However, the inference processes of most existing methods rely on Monte Carlo random sampling, which…

2023

Vision-based Six-Dimensional Peg-in-Hole for Practical Connector Insertion

ICRA 2023poster

We study six-dimensional (6D) perceptive peg-in-hole problem for practical connector insertion task in this paper. To enable the manipulator system to handle different types of pegs in complex environment, we develop a perceptive robotic assembly system that utilizes an in-hand RGB-D camera for peg-…

Cited by 11SourceScholar
2023

Which is Better for Learning with Noisy Labels: The Semi-supervised Method or Modeling Label Noise?

ICML 2023poster

In real life, accurately annotating large-scale datasets is sometimes difficult. Datasets used for training deep learning models are likely to contain label noise. To make use of the dataset containing label noise, two typical methods have been proposed. One is to employ the semi-supervised method b…

Cited by 10SourcePDFScholar
2022

Action-Sufficient State Representation Learning for Control with Structural Constraints

ICML 2022spotlight

Perceived signals in real-world scenarios are usually high-dimensional and noisy, and finding and using their representation that contains essential and sufficient information required by downstream decision-making tasks will help improve computational efficiency and generalization ability in the ta…

Cited by 48SourcePDFScholar
2022

AdaRL: What, Where, and How to Adapt in Transfer Reinforcement Learning

ICLR 2022spotlight

One practical challenge in reinforcement learning (RL) is how to make quick adaptations when faced with new environments. In this paper, we propose a principled framework for adaptive RL, called AdaRL, that adapts reliably and efficiently to changes across domains with a few samples from the target…

2022

Adversarial Robustness Through the Lens of Causality

ICLR 2022poster

The adversarial vulnerability of deep neural networks has attracted significant attention in machine learning. As causal reasoning has an instinct for modeling distribution change, it is essential to incorporate causality into analyzing this specific type of distribution change induced by adversarial…

Cited by 105SourcePDFScholar
2022

Alleviating Semantics Distortion in Unsupervised Low-Level Image-to-Image Translation via Structure Consistency Constraint

CVPR 2022poster

Unsupervised image-to-image (I2I) translation aims to learn a domain mapping function that can preserve the semantics of the input images without paired data. However, because the underlying semantics distributions in the source and target domains are often mismatched, current distribution matching-…

Cited by 33PDFcodeScholar
2022

Causal Discovery in Linear Latent Variable Models Subject to Measurement Error

NeurIPS 2022accept

We focus on causal discovery in the presence of measurement error in linear systems where the mixing matrix, i.e., the matrix indicating the independent exogenous noise terms pertaining to the observed variables, is identified up to permutation and scaling of the columns. We demonstrate a somewhat s…

2022

Conditional Contrastive Learning with Kernel

ICLR 2022poster

Conditional contrastive learning frameworks consider the conditional sampling procedure that constructs positive or negative data pairs conditioned on specific variables. Fair contrastive learning constructs negative pairs, for example, from the same gender (conditioning on sensitive information), w…

2022

Counterfactual Fairness with Partially Known Causal Graph

NeurIPS 2022accept

Fair machine learning aims to avoid treating individuals or sub-populations unfavourably based on \textit{sensitive attributes}, such as gender and race. Those methods in fair machine learning that are built on causal inference ascertain discrimination and bias through causal effects. Though causali…

Cited by 33SourcePDFScholar
2022

Factored Adaptation for Non-Stationary Reinforcement Learning

NeurIPS 2022accept

Dealing with non-stationarity in environments (e.g., in the transition dynamics) and objectives (e.g., in the reward functions) is a challenging problem that is crucial in real-world applications of reinforcement learning (RL). While most current approaches model the changes as a single shared embed…

Cited by 43SourcePDFScholar
2022

GLaM: Efficient Scaling of Language Models with Mixture-of-Experts

ICML 2022spotlight

Scaling language models with more data, compute and parameters has driven significant progress in natural language processing. For example, thanks to scaling, GPT-3 was able to achieve strong results on in-context learning tasks. However, training these large dense models requires significant amount…

Cited by 765SourcePDFScholar
2022

Identification of Linear Latent Variable Model with Arbitrary Distribution

AAAI 2022technical

An important problem across multiple disciplines is to infer and understand meaningful latent variables. One strategy commonly used is to model the measured variables in terms of the latent variables under suitable assumptions on the connectivity from the latents to the measured (known as measuremen…

Cited by 22SourcePDFScholar
2022

Identification of Linear Non-Gaussian Latent Hierarchical Structure

ICML 2022spotlight

Traditional causal discovery methods mainly focus on estimating causal relations among measured variables, but in many real-world problems, such as questionnaire-based psychometric studies, measured variables are generated by latent variables that are causally related. Accordingly, this paper invest…

Cited by 68SourcePDFScholar
2022

Incorporating Dynamic Semantics into Pre-Trained Language Model for Aspect-based Sentiment Analysis

ACL 2022findings

Aspect-based sentiment analysis (ABSA) predicts sentiment polarity towards a specific aspect in the given sentence. While pre-trained language models such as BERT have achieved great success, incorporating dynamic semantic changes into ABSA remains challenging. To this end, in this paper, we propose…

Cited by 84SourcePDFScholar
2022

Independence Testing-Based Approach to Causal Discovery under Measurement Error and Linear Non-Gaussian Models

NeurIPS 2022accept

Causal discovery aims to recover causal structures generating the observational data. Despite its success in certain problems, in many real-world scenarios the observed variables are not the target variables of interest, but the imperfect measures of the target variables. Causal discovery under meas…

Cited by 13SourcePDFScholar
2022

Invariant Action Effect Model for Reinforcement Learning

AAAI 2022technical

Good representations can help RL agents perform concise modeling of their surroundings, and thus support effective decision-making in complex environments. Previous methods learn good representations by imposing extra constraints on dynamics. However, in the causal perspective, the causation betw…

Cited by 13SourcePDFScholar
2022

Latent Hierarchical Causal Structure Discovery with Rank Constraints

NeurIPS 2022accept

Most causal discovery procedures assume that there are no latent confounders in the system, which is often violated in real-world problems. In this paper, we consider a challenging scenario for causal structure identification, where some variables are latent and they may form a hierarchical graph st…

Cited by 57SourcePDFScholar
2022

Learning Temporally Causal Latent Processes from General Temporal Data

ICLR 2022poster

Our goal is to recover time-delayed latent causal variables and identify their relations from measured temporal data. Estimating causally-related latent variables from observations is particularly challenging as the latent variables are not uniquely recoverable in the most general case. In this work…

2022

Maximum Spatial Perturbation Consistency for Unpaired Image-to-Image Translation

CVPR 2022poster

Unpaired image-to-image translation (I2I) is an ill-posed problem, as an infinite number of translation functions can map the source domain distribution to the target distribution. Therefore, much effort has been put into designing suitable constraints, e.g., cycle consistency (CycleGAN), geometry c…

Cited by 26PDFcodeScholar
2022

Meta-CQG: A Meta-Learning Framework for Complex Question Generation over Knowledge Bases

COLING 2022main

Complex question generation over knowledge bases (KB) aims to generate natural language questions involving multiple KB relations or functional constraints. Existing methods train one encoder-decoder-based model to fit all questions. However, such a one-size-fits-all strategy may not perform well si…

Cited by 11SourcePDFScholar
2022

MissDAG: Causal Discovery in the Presence of Missing Data with Continuous Additive Noise Models

NeurIPS 2022accept

State-of-the-art causal discovery methods usually assume that the observational data is complete. However, the missing data problem is pervasive in many practical scenarios such as clinical trials, economics, and biology. One straightforward way to address the missing data problem is first to impute…

2022

On the Convergence of Continuous Constrained Optimization for Structure Learning

AISTATS 2022poster

Recently, structure learning of directed acyclic graphs (DAGs) has been formulated as a continuous optimization problem by leveraging an algebraic characterization of acyclicity. The constrained problem is solved using the augmented Lagrangian method (ALM) which is often preferred to the quadratic p…

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
2022

Residual Similarity Based Conditional Independence Test and Its Application in Causal Discovery

AAAI 2022technical

Recently, many regression based conditional independence (CI) test methods have been proposed to solve the problem of causal discovery. These methods provide alternatives to test CI by first removing the information of the controlling set from the two target variables, and then testing the independe…

2022

Show Your Faith: Cross-Modal Confidence-Aware Network for Image-Text Matching

AAAI 2022technical

Image-text matching bridges vision and language, which is a crucial task in the field of multi-modal intelligence. The key challenge lies in how to measure image-text relevance accurately as matching evidence. Most existing works aggregate the local semantic similarities of matched region-word pairs…

2022

Towards Federated Bayesian Network Structure Learning with Continuous Optimization

AISTATS 2022poster

Traditionally, Bayesian network structure learning is often carried out at a central site, in which all data is gathered. However, in practice, data may be distributed across different parties (e.g., companies, devices) who intend to collectively learn a Bayesian network, but are not willing to disc…

2022

Truncated Matrix Power Iteration for Differentiable DAG Learning

NeurIPS 2022accept

Recovering underlying Directed Acyclic Graph (DAG) structures from observational data is highly challenging due to the combinatorial nature of the DAG-constrained optimization problem. Recently, DAG learning has been cast as a continuous optimization problem by characterizing the DAG constraint as a…

2022

Unsupervised Image-to-Image Translation with Density Changing Regularization

NeurIPS 2022accept

Unpaired image-to-image translation aims to translate an input image to another domain such that the output image looks like an image from another domain while important semantic information are preserved. Inferring the optimal mapping with unpaired data is impossible without making any assumptions.…

2021

DAE-GAN: Dynamic Aspect-Aware GAN for Text-to-Image Synthesis

ICCV 2021poster

Text-to-image synthesis refers to generating an image from a given text description, the key goal of which lies in photo realism and semantic consistency. Previous methods usually generate an initial image with sentence embedding and then refine it with fine-grained word embedding. Despite the signi…

Cited by 146PDFcodeScholar
2021

DeepTrader: A Deep Reinforcement Learning Approach for Risk-Return Balanced Portfolio Management with Market Conditions Embedding

AAAI 2021technical

Most existing reinforcement learning (RL)-based portfolio management models do not take into account the market conditions, which limits their performance in risk-return balancing. In this paper, we propose DeepTrader, a deep RL method to optimize the investment policy. In particular, to tackle the…

Cited by 115SourcePDFScholar
2021

Domain Adaptation with Invariant Representation Learning: What Transformations to Learn?

NeurIPS 2021poster

Unsupervised domain adaptation, as a prevalent transfer learning setting, spans many real-world applications. With the increasing representational power and applicability of neural networks, state-of-the-art domain adaptation methods make use of deep architectures to map the input features $X$ to a…

2021

Identification of Partially Observed Linear Causal Models: Graphical Conditions for the Non-Gaussian and Heterogeneous Cases

NeurIPS 2021poster

In causal discovery, linear non-Gaussian acyclic models (LiNGAMs) have been studied extensively. While the causally sufficient case is well understood, in many real problems the observed variables are not causally related. Rather, they are generated by latent variables, such as confounders and media…

Cited by 51SourcePDFScholar
2021

Ideography Leads Us to the Field of Cognition: A Radical-Guided Associative Model for Chinese Text Classification

AAAI 2021technical

Cognitive psychology research shows that humans have the instinct for abstract thinking, where association plays an essential role in language comprehension. Especially for Chinese, its ideographic writing system allows radicals to trigger semantic association without the need of phonetics. In fact,…

2021

Instance-dependent Label-noise Learning under a Structural Causal Model

NeurIPS 2021poster

Label noise generally degenerates the performance of deep learning algorithms because deep neural networks easily overfit label errors. Let $X$ and $Y$ denote the instance and clean label, respectively. When $Y$ is a cause of $X$, according to which many datasets have been constructed, e.g., \text…

Cited by 90SourcePDFScholar
2021

Making the Relation Matters: Relation of Relation Learning Network for Sentence Semantic Matching

AAAI 2021technical

Sentence semantic matching is one of the fundamental tasks in natural language processing, which requires an agent to determine the semantic relation among input sentences. Recently, deep neural networks have achieved impressive performance in this area, especially BERT. Despite the effectiveness of…