← Search

Tian Gao

61 accepted papers

2026

CGU-Bayes: Causal Graph Uncertainty-Guided Bayesian Inference for Domain Generalization

CVPR 2026

Causal graphs play a crucial role in AI research as they reveal the data generation processes underlying real-world machine learning and computer vision tasks. Recent studies have leveraged causal graphs to develop more robust and interpretable models. However, limited or biased data often lead to i

Cited by 0SourceScholar
2026

Capability-Oriented Training Induced Alignment Risk

ICML 2026poster

While most AI alignment research focuses on preventing models from generating explicitly harmful content, a more subtle risk is emerging: capability-oriented training induced exploitation. We investigate whether language models, when trained with reinforcement learning (RL) in environments with impl…

Cited by 0SourceScholar
2026

Disentangled Representation Learning for Parametric Partial Differential Equations

ICLR 2026poster

Neural operators (NOs) excel at learning mappings between function spaces, serving as efficient forward solution approximators for PDE-governed systems. However, as black-box solvers, they offer limited insight into the underlying physical mechanism, due to the lack of interpretable representations…

Cited by 0SourcecodeScholar
2026

Resting Neurons, Active Insights: Robustify Activation Sparsity for Large Language Models

ICML 2026poster

Activation sparsity offers a compelling route to accelerate large language model (LLM) inference by selectively suppressing hidden activations, yet existing approaches exhibit severe accuracy degradation at high sparsity. We show that this failure stems from representational instability: *activation…

Cited by 0SourceScholar
2025

BEV-TSR: Text-Scene Retrieval in BEV Space for Autonomous Driving

AAAI 2025technical

The rapid development of the autonomous driving industry has led to a significant accumulation of autonomous driving data. Consequently, there comes a growing demand for retrieving data to provide specialized optimization. However, directly applying previous image retrieval methods faces several cha…

Cited by 2SourcePDFScholar
2025

Enhancing Multimodal Sentiment Analysis for Missing Modality through Self-Distillation and Unified Modality Cross-Attention

ICASSP 2025accepted

In multimodal sentiment analysis, collecting text data is often more challenging than video or audio due to higher annotation costs and inconsistent automatic speech recognition (ASR) quality. To address this challenge, our study has developed a robust model that effectively integrates multimodal se…

Cited by 0SourceScholar
2025

Incorporating Spatial Cues in Modular Speaker Diarization for Multi-channel Multi-party Meetings

ICASSP 2025accepted

Although fully end-to-end speaker diarization systems have made significant progress in recent years, modular systems often achieve superior results in real-world scenarios due to their greater adaptability and robustness. Historically, modular speaker diarization methods have seldom discussed how t…

Cited by 0SourceScholar
2025

Information-Bottleneck Driven Binary Neural Network for Change Detection

ICCV 2025poster

In this paper, we propose Binarized Change Detection (BiCD), the first binary neural network (BNN) designed specifically for change detection. Conventional network binarization approaches, which directly quantize both weights and activations in change detection models, severely limit the network's a…

2025

Integer Programming Based Methods and Heuristics for Causal Graph Learning

AISTATS 2025poster

Acyclic directed mixed graphs (ADMG) – graphs that contain both directed and bidi- rected edges but no directed cycles – are used to model causal and conditional independence relationships between a set of random vari- ables in the presence of latent or unmeasured variables. Bow-free ADMGs, Arid ADM…

Cited by 0SourceScholar
2025

Justice or Prejudice? Quantifying Biases in LLM-as-a-Judge

ICLR 2025poster

LLM-as-a-Judge has been widely utilized as an evaluation method in various benchmarks and served as supervised rewards in model training. However, despite their excellence in many domains, potential issues are under-explored, undermining their reliability and the scope of their utility. Therefore,…

Cited by 49SourcePDFScholar
2025

Meta-D2AG: Causal Graph Learning with Interventional Dynamic Data

NeurIPS 2025poster

Causal discovery in the form of a directed acyclic graph (DAG) for dynamic time series data has been widely studied in various applications. Much of the existing work has focused on observational, offline, and/or stationary settings. In this work, we propose a dynamic DAG discovery algorithm, Meta-D…

Cited by 0SourceScholar
2025

Q-function Decomposition with Intervention Semantics for Factored Action Spaces

AISTATS 2025poster

Many practical reinforcement learning environments have a discrete factored action space that induces a large combinatorial set of actions, thereby posing significant challenges. Existing approaches leverage the regular structure of the action space and resort to a linear decomposition of Q-functio…

Cited by 0SourceScholar
2025

Scaling Bioacoustic Signal Pre-training with Million Samples Via Mask-Modeling

ICASSP 2025accepted

Deep learning-based bioacoustic audio analysis holds immense potential across various applications. However, existing studies in bioacoustics often focus on a limited number of species, potentially hindering the transferability of models across different species. Furthermore, the manual annotation o…

Cited by 0SourceScholar
2024

CHRONOS: A Schema-Based Event Understanding and Prediction System

AAAI 2024technical

Chronological and Hierarchical Reasoning Over Naturally Occurring Schemas (CHRONOS) is a system that combines language model-based natural language processing with symbolic knowledge representations to analyze and make predictions about newsworthy events. CHRONOS consists of an event-centric informa…

Cited by 1SourcePDFScholar
2024

Effective Causal Discovery under Identifiable Heteroscedastic Noise Model

AAAI 2024technical

Capturing the underlying structural causal relations represented by Directed Acyclic Graphs (DAGs) has been a fundamental task in various AI disciplines. Causal DAG learning via the continuous optimization framework has recently achieved promising performance in terms of accuracy and efficiency. How…

2024

Implicit Enhancement of Target Speaker in Speaker-Adaptive ASR through Efficient Joint Optimization

ICASSP 2024accepted

In multi-speaker scenarios, automatic speech recognition (ASR) models rely on pre-processed audio after speaker separation. However, when the target speaker is not accurately separated, ASR models face limitations in reaching their peak performance. To address this issue, we propose a speaker-adapti…

Cited by 0SourceScholar
2024

Integrating Markov Blanket Discovery into Causal Representation Learning for Domain Generalization

ECCV 2024poster

"Identifying low-dimensional, semantic latent causal representations for high-dimensional data has become a dynamic field in computer vision and machine learning. Causal domain generalization methods aim to identify latent causal variables that generate input data and build invariant causal mechanis…

Cited by 1SourcePDFScholar
2024

Nonlocal Attention Operator: Materializing Hidden Knowledge Towards Interpretable Physics Discovery

NeurIPS 2024spotlight

Despite recent popularity of attention-based neural architectures in core AI fields like natural language processing (NLP) and computer vision (CV), their potential in modeling complex physical systems remains under-explored. Learning problems in physical systems are often characterized as discoveri…

Cited by 9SourcePDFScholar
2024

PRIME: Scaffolding Manipulation Tasks With Behavior Primitives for Data-Efficient Imitation Learning

RA-L 2024

Imitation learning has shown great potential for enabling robots to acquire complex manipulation behaviors. However, these algorithms suffer from high sample complexity in long-horizon tasks, where compounding errors accumulate over the task horizons. We present PRIME (<underline xmlns:mml="http://w

Cited by 15SourceScholar
2024

Theory-guided Message Passing Neural Network for Probabilistic Inference

AISTATS 2024poster

Probabilistic inference can be tackled by minimizing a variational free energy through message passing. To improve performance, neural networks are adopted for message computation. Neural message learning is heuristic and requires strong guidance to perform well. In this work, we propose a {\em theo…

2023

A Multi-Scale Feature Aggregation Based Lightweight Network for Audio-Visual Speech Enhancement

ICASSP 2023accepted

Audio-visual speech enhancement (AVSE) was shown to be superior over conventional audio-only counterpart for improving the speech quality. However, most existing AVSE models are heavyweight in the sense of parameter count, which is inappropriate for the deployment and practical applications. In this…

Cited by 0SourceScholar
2023

An Actor-Centric Causality Graph for Asynchronous Temporal Inference in Group Activity

CVPR 2023poster

The causality relation modeling remains a challenging task for group activity recognition. The causality relations describe the influence of some actors (cause actors) on other actors (effect actors). Most existing graph models focus on learning the actor relation with synchronous temporal features,…

Cited by 16SourcePDFScholar
2023

Approximate Inference in Logical Credal Networks

IJCAI 2023poster

The Logical Credal Network or LCN is a recent probabilistic logic designed for effective aggregation and reasoning over multiple sources of imprecise knowledge. An LCN specifies a set of probability distributions over all interpretations of a set of logical formulas for which marginal and conditiona…

Cited by 3SourcePDFScholar
2023

Biomechanics-Guided Facial Action Unit Detection Through Force Modeling

CVPR 2023poster

Existing AU detection algorithms are mainly based on appearance information extracted from 2D images, and well-established facial biomechanics that governs 3D facial skin deformation is rarely considered. In this paper, we propose a biomechanics-guided AU detection approach, where facial muscle acti…

Cited by 25SourcePDFScholar
2023

Concurrent Multi-Label Prediction in Event Streams

AAAI 2023technical

Streams of irregularly occurring events are commonly modeled as a marked temporal point process. Many real-world datasets such as e-commerce transactions and electronic health records often involve events where multiple event types co-occur, e.g. multiple items purchased or multiple diseases diagnos…

2023

Pairwise Causality Guided Transformers for Event Sequences

NeurIPS 2023poster

Although pairwise causal relations have been extensively studied in observational longitudinal analyses across many disciplines, incorporating knowledge of causal pairs into deep learning models for temporal event sequences remains largely unexplored. In this paper, we propose a novel approach for e…

Cited by 3SourcePDFScholar
2023

Probabilistic Attention-to-Influence Neural Models for Event Sequences

ICML 2023poster

Discovering knowledge about which types of events influence others, using datasets of event sequences without time stamps, has several practical applications. While neural sequence models are able to capture complex and potentially long-range historical dependencies, they often lack the interpretabi…

Cited by 3SourcePDFScholar
2023

Quantum Transfer Learning Using the Large-Scale Unsupervised Pre-Trained Model Wavlm-Large for Synthetic Speech Detection

ICASSP 2023accepted

The development of quantum machine learning demonstrates its quantum advantages over traditional deep learning, which promises to discover new patterns on supervised classification datasets. This work proposes a classical-to-quantum transfer learning system based on the large-scale unsupervised pre-…

Cited by 0SourceScholar
2023

Score-Based Learning of Graphical Event Models with Background Knowledge Augmentation

AAAI 2023technical

Graphical event models (GEMs) are representations of temporal point process dynamics between different event types. Many real-world applications however involve limited event stream data, making it challenging to learn GEMs from data alone. In this paper, we introduce approaches that can work togeth…

2022

IDYNO: Learning Nonparametric DAGs from Interventional Dynamic Data

ICML 2022spotlight

Causal discovery in the form of a directed acyclic graph (DAG) for time series data has been widely studied in various domains. The resulting DAG typically represents a dynamic Bayesian network (DBN), capturing both the instantaneous and time-delayed relationships among variables of interest. We pro…

2022

Learning and Retrieval from Prior Data for Skill-based Imitation Learning

CoRL 2022poster

Imitation learning offers a promising path for robots to learn general-purpose tasks, but traditionally has enjoyed limited scalability due to high data supervision requirements and brittle generalization. Inspired by recent work on skill-based imitation learning, we investigate whether leveraging p…

Cited by 53SourcecodeScholar
2022

Linearizing contextual bandits with latent state dynamics

UAI 2022poster

In many real-world applications of multi-armed bandit problems, both rewards and contexts are often influenced by confounding latent variables which evolve stochastically over time. While the observed contexts and rewards are nonlinearly related, we show that prior knowledge of latent causal structu…

2022

Logical Credal Networks

NeurIPS 2022accept

We introduce Logical Credal Networks (or LCNs for short) -- an expressive probabilistic logic that generalizes prior formalisms that combine logic and probability. Given imprecise information represented by probability bounds and conditional probability bounds on logic formulas, an LCN specifies a s…

Cited by 6SourcePDFScholar
2022

Phasic Self-Imitative Reduction for Sparse-Reward Goal-Conditioned Reinforcement Learning

ICML 2022spotlight

It has been a recent trend to leverage the power of supervised learning (SL) towards more effective reinforcement learning (RL) methods. We propose a novel phasic solution by alternating online RL and offline SL for tackling sparse-reward goal-conditioned problems. In the online phase, we perform RL…

Cited by 22SourcePDFScholar
2022

Variational message passing neural network for Maximum-A-Posteriori (MAP) inference

UAI 2022poster

Maximum-A-Posteriori (MAP) inference is a fundamental task in probabilistic inference and belief propagation (BP) is a widely used algorithm for MAP inference. Though BP has been applied successfully to many different fields, it offers no performance guarantee and often performs poorly on loopy grap…

2021

Causal Inference for Event Pairs in Multivariate Point Processes

NeurIPS 2021poster

Causal inference and discovery from observational data has been extensively studied across multiple fields. However, most prior work has focused on independent and identically distributed (i.i.d.) data. In this paper, we propose a formalization for causal inference between pairs of event variables i…

Cited by 15SourcePDFScholar
2021

DAGs with No Curl: An Efficient DAG Structure Learning Approach

ICML 2021spotlight

Recently directed acyclic graph (DAG) structure learning is formulated as a constrained continuous optimization problem with continuous acyclicity constraints and was solved iteratively through subproblem optimization. To further improve efficiency, we propose a novel learning framework to model and…

2021

Integer Programming for Causal Structure Learning in the Presence of Latent Variables

ICML 2021oral

The problem of finding an ancestral acyclic directed mixed graph (ADMG) that represents the causal relationships between a set of variables is an important area of research on causal inference. Most existing score-based structure learning methods focus on learning directed acyclic graph (DAG) models…

2021

Ordinal Historical Dependence in Graphical Event Models with Tree Representations

AAAI 2021technical

Graphical event models are representations that capture process independence between different types of events in multivariate temporal point processes. The literature consists of various parametric models and approaches to learn them from multivariate event stream data. Since these models are inter…

2021

Timeline Summarization based on Event Graph Compression via Time-Aware Optimal Transport

EMNLP 2021main

Timeline Summarization identifies major events from a news collection and describes them following temporal order, with key dates tagged. Previous methods generally generate summaries separately for each date after they determine the key dates of events. These methods overlook the events’ intra-stru…

2021

Type-augmented Relation Prediction in Knowledge Graphs

AAAI 2021technical

Knowledge graphs (KGs) are of great importance to many real world applications, but they generally suffer from incomplete information in the form of missing relations between entities. Knowledge graph completion (also known as relation prediction) is the task of inferring missing facts given existin…

Cited by 53SourcePDFScholar
2020

Cause-Effect Association between Event Pairs in Event Datasets

IJCAI 2020poster

Causal discovery from observational data has been intensely studied across fields of study. In this paper, we consider datasets involving irregular occurrences of various types of events over the timeline. We propose a suite of scores and related algorithms for estimating the cause-effect associatio…

Cited by 0SourcePDFScholar
2020

Characterization of Overlap in Observational Studies

AISTATS 2020poster

Overlap between treatment groups is required for non-parametric estimation of causal effects. If a subgroup of subjects always receives the same intervention, we cannot estimate the effect of intervention changes on that subgroup without further assumptions. When overlap does not hold globally, ch…

2020

DAGs with No Fears: A Closer Look at Continuous Optimization for Learning Bayesian Networks

NeurIPS 2020spotlight

This paper re-examines a continuous optimization framework dubbed NOTEARS for learning Bayesian networks. We first generalize existing algebraic characterizations of acyclicity to a class of matrix polynomials. Next, focusing on a one-parameter-per-edge setting, it is shown that the Karush-Kuhn-Tuck…

2020

Order-Dependent Event Models for Agent Interactions

IJCAI 2020poster

In multivariate event data, the instantaneous rate of an event's occurrence may be sensitive to the temporal sequence in which other influencing events have occurred in the history. For example, an agent’s actions are typically driven by preceding actions taken by the agent as well as those of other…

Cited by 0SourcePDFScholar
2020

Progressive Multi-Target Network Based Speech Enhancement with Snr-Preselection for Robust Speaker Diarization

ICASSP 2020accepted

In this paper, we design a novel front-end processing system for speaker diarization under realistic conditions with challenging background noises. To cope with diversified environments, we first extend our perviously proposed progressive learning based speech enhancement model by adding multi-task…

Cited by 0SourceScholar
2019

A Two-stage Single-channel Speaker-dependent Speech Separation Approach for Chime-5 Challenge

ICASSP 2019accepted

In this paper, we design a two-stage single-channel speaker-dependent speech separation approach for the CHiME-5 Challenge, targeting the problem of far-field and multi-talker conversational speech recognition in dinner party scenarios involving background noises, reverberations and overlapping spee…

Cited by 0SourceScholar
2018

A Novel LSTM-Based Speech Preprocessor for Speaker Diarization in Realistic Mismatch Conditions

ICASSP 2018accepted

In this study, we investigate on the effects of deep learning based speech enhancement as a preprocessor to speaker diarization in quite challenging realistic environments involving the background noises, reverberations and overlapping speech. To improve the generalization capability, the advanced l…

Cited by 0SourceScholar
2018

Densely Connected Progressive Learning for LSTM-Based Speech Enhancement

ICASSP 2018accepted

Recently, we proposed a novel progressive learning (PL) framework for deep neural network (DNN) based speech enhancement to improve the performance in low signal-to-noise ratio (SNR) environments. In this study, several new contributions are made to this framework. First, the advanced long short-ter…

Cited by 0SourceScholar
2015

Joint training of front-end and back-end deep neural networks for robust speech recognition

ICASSP 2015accepted

Based on the recently proposed speech pre-processing front-end with deep neural networks (DNNs), we first investigate different feature mapping directly from noisy speech via DNN for robust speech recognition. Next, we propose to jointly train a single DNN for both feature mapping and acoustic model…

Cited by 0SourceScholar
2015

Structured Feature Selection

ICCV 2015poster

Feature dimensionality reduction has been widely used in various computer vision tasks. We explore feature selection as the dimensionality reduction technique and propose to use a structured approach, based on the Markov Blanket (MB), to select features. We first introduce a new MB discovery algorit…

Cited by 12PDFScholar