← Search

Yazhe Li

11 accepted papers

2025

On the Hardness of Conditional Independence Testing In Practice

NeurIPS 2025spotlight

Tests of conditional independence (CI) underpin a number of important problems in machine learning and statistics, from causal discovery to evaluation of predictor fairness and out-of-distribution robustness. Shah and Peters (2020) showed that, contrary to the unconditional case, no universally fini…

Cited by 0SourceScholar
2023

Efficient Conditionally Invariant Representation Learning

ICLR 2023top-5%

We introduce the Conditional Independence Regression CovariancE (CIRCE), a measure of conditional independence for multivariate continuous-valued variables. CIRCE applies as a regularizer in settings where we wish to learn neural features $\varphi(X)$ of data $X$ to estimate a target $Y$, while bein…

2023

Investigating the Role of Model-Based Learning in Exploration and Transfer

ICML 2023poster

State of the art reinforcement learning has enabled training agents on tasks of ever increasing complexity. However, the current paradigm tends to favor training agents from scratch on every new task or on collections of tasks with a view towards generalizing to novel task configurations. The former…

Cited by 8SourcePDFScholar
2022

Procedural generalization by planning with self-supervised world models

ICLR 2022poster

One of the key promises of model-based reinforcement learning is the ability to generalize using an internal model of the world to make predictions in novel environments and tasks. However, the generalization ability of model-based agents is not well understood because existing work has focused on m…

Cited by 39SourcePDFScholar
2021

Self-Supervised Learning with Kernel Dependence Maximization

NeurIPS 2021poster

We approach self-supervised learning of image representations from a statistical dependence perspective, proposing Self-Supervised Learning with the Hilbert-Schmidt Independence Criterion (SSL-HSIC). SSL-HSIC maximizes dependence between representations of transformations of an image and the image i…

2021

Vector Quantized Models for Planning

ICML 2021spotlight

Recent developments in the field of model-based RL have proven successful in a range of environments, especially ones where planning is essential. However, such successes have been limited to deterministic fully-observed environments. We present a new approach that handles stochastic and partially-o…

Cited by 62SourcePDFScholar
2019

Low Bit-rate Speech Coding with VQ-VAE and a WaveNet Decoder

ICASSP 2019accepted

In order to efficiently transmit and store speech signals, speech codecs create a minimally redundant representation of the input signal which is then decoded at the receiver with the best possible perceptual quality. In this work we demonstrate that a neural network architecture based on VQ-VAE wit…

Cited by 0SourceScholar
2018

Parallel WaveNet: Fast High-Fidelity Speech Synthesis

ICML 2018oral

The recently-developed WaveNet architecture is the current state of the art in realistic speech synthesis, consistently rated as more natural sounding for many different languages than any previous system. However, because WaveNet relies on sequential generation of one audio sample at a time, it is…

Cited by 1053SourcePDFScholar