← Search

Rui Shu

14 accepted papers

2025

Stop Diverse OOD Attacks: Knowledge Ensemble for Reliable Defense

AAAI 2025technical

Enhancing defense through model ensemble is an emerging trend, where the challenge lies in how to use ensemble knowledge to counter Out-of-Distribution (OOD) attacks. In this paper, we propose the Reliable Defense Ensemble (REE) to address this issue. REE optimizes the ensemble knowledge of models t…

Cited by 0SourcePDFScholar
2021

Anytime Sampling for Autoregressive Models via Ordered Autoencoding

ICLR 2021poster

Autoregressive models are widely used for tasks such as image and audio generation. The sampling process of these models, however, does not allow interruptions and cannot adapt to real-time computational resources. This challenge impedes the deployment of powerful autoregressive models, which involv…

2021

Temporal Predictive Coding For Model-Based Planning In Latent Space

ICML 2021spotlight

High-dimensional observations are a major challenge in the application of model-based reinforcement learning (MBRL) to real-world environments. To handle high-dimensional sensory inputs, existing approaches use representation learning to map high-dimensional observations into a lower-dimensional lat…

Cited by 61SourcePDFScholar
2020

Fair Generative Modeling via Weak Supervision

ICML 2020poster

Real-world datasets are often biased with respect to key demographic factors such as race and gender. Due to the latent nature of the underlying factors, detecting and mitigating bias is especially challenging for unsupervised machine learning. We present a weakly supervised algorithm for overcoming…

2020

Prediction, Consistency, Curvature: Representation Learning for Locally-Linear Control

ICLR 2020poster

Many real-world sequential decision-making problems can be formulated as optimal control with high-dimensional observations and unknown dynamics. A promising approach is to embed the high-dimensional observations into a lower-dimensional latent representation space, estimate the latent dynamics mode…

Cited by 33SourceScholar
2020

Predictive Coding for Locally-Linear Control

ICML 2020poster

High-dimensional observations and unknown dynamics are major challenges when applying optimal control to many real-world decision making tasks. The Learning Controllable Embedding (LCE) framework addresses these challenges by embedding the observations into a lower dimensional latent space, estimati…

2020

Weakly Supervised Disentanglement with Guarantees

ICLR 2020poster

Learning disentangled representations that correspond to factors of variation in real-world data is critical to interpretable and human-controllable machine learning. Recently, concerns about the viability of learning disentangled representations in a purely unsupervised manner has spurred a shift t…

Cited by 169SourcecodeScholar
2019

Training Variational Autoencoders with Buffered Stochastic Variational Inference

AISTATS 2019poster

The recognition network in deep latent variable models such as variational autoencoders (VAEs) relies on amortized inference for efficient posterior approximation that can scale up to large datasets. However, this technique has also been demonstrated to select suboptimal variational parameters, ofte…

Cited by 5SourcePDFScholar
2018

Constructing Unrestricted Adversarial Examples with Generative Models

NeurIPS 2018poster

Adversarial examples are typically constructed by perturbing an existing data point within a small matrix norm, and current defense methods are focused on guarding against this type of attack. In this paper, we propose a new class of adversarial examples that are synthesized entirely from scratch us…

2018

Robust Locally-Linear Controllable Embedding

AISTATS 2018poster

Embed-to-control (E2C) is a model for solving high-dimensional optimal control problems by combining variational auto-encoders with locally-optimal controllers. However, the E2C model suffers from two major drawbacks: 1) its objective function does not correspond to the likelihood of the data seque…

Cited by 0SourcePDFScholar