← Search

Cheng Zhang

102 accepted papers

2026

A3: an Analytical Low-Rank Approximation Framework for Attention

ICML 2026poster

Large language models have demonstrated remarkable performance; however, their massive parameter counts make deployment highly expensive. Low-rank approximation offers a promising compression solution, yet existing approaches have two main limitations: (1) They focus on minimizing the output error o…

Cited by 0SourceScholar
2026

ART: Articulated Reconstruction Transformer

CVPR 2026

We introduce ART, Articulated Reconstruction Transformer--a category-agnostic, feed-forward model that reconstructs complete 3D articulated objects from only sparse, multi-state RGB images. Previous methods for articulated object reconstruction either rely on slow optimization with fragile cross-sta

Cited by 0SourceScholar
2026

Compute as Teacher: Turning Inference Compute Into Reference-Free Supervision

ICML 2026poster

Where do learning signals come from when there is no ground truth in post-training? We show that inference compute itself can serve as supervision. By generating parallel rollouts and converting them into reference estimates, models can learn without human labels—critically, even in non-verifiable d…

Cited by 0SourceScholar
2026

Continual Unlearning for Text-to-Image Diffusion Models: A Regularization Perspective

ICLR 2026poster

Machine unlearning—the ability to remove designated concepts from a pre-trained model—has advanced rapidly, particularly for text-to-image diffusion models. However, existing methods typically assume that unlearning requests arrive all at once, whereas in practice they often arrive sequentially. We…

Cited by 0SourceScholar
2026

Cook and Clean Together: Teaching Embodied Agents for Parallel Task Execution

AAAI 2026technical

Task scheduling has become increasingly critical for embodied AI, where agents need to follow natural language instructions and execute actions efficiently in 3D physical worlds. Existing datasets for task planning in 3D environments often simplify the problem, lacking operations research knowledge

Cited by 0SourcePDFScholar
2026

KernelCraft: Benchmarking for Agentic Close-to-Metal Kernel Generation on Emerging Hardware

ICML 2026poster

New AI accelerators with novel instruction set architectures (ISAs) often require developers to manually craft low-level kernels - a time-consuming, laborious, and error-prone process that cannot scale across diverse hardware targets. This prevents emerging hardware platforms from reaching the marke…

Cited by 0SourceScholar
2026

MAPS: Memory-Aware Predictive Scheduling Framework for Large Language Models Serving

ICML 2026poster

The surge of large language model (LLM) applications on personal devices imposes massive, bursty workloads on cloud serving infrastructure. While prefill-decode disaggregation improves throughput and scalability, memory-bound decode instances often suffer from persistent load imbalance, as output le…

Cited by 0SourceScholar
2026

MSCFL: Model Structure-Aware Clustered Federated Learning for System Heterogeneity and Data Drift

AAAI 2026technical

Federated Learning (FL) faces significant challenges arising from both data and system heterogeneity. While Clustered Federated Learning (CFL) mitigates data heterogeneity by grouping clients with similar data distributions, it remains vulnerable to system heterogeneity, which can slow convergence d

Cited by 0SourcePDFScholar
2026

MiniBEE: A New Form Factor for Compact Bimanual Dexterity

ICRA 2026poster

Bimanual robot manipulators can achieve impressive dexterity, but typically rely on two full six- or seven-degree-of-freedom arms so that paired grippers can coordinate effectively. This traditional framework increases system complexity and footprint while only exploiting a fraction of the overall w…

2026

PanFlow: Decoupled Motion Control for Panoramic Video Generation

AAAI 2026technical

Panoramic video generation has attracted growing attention due to its applications in virtual reality and immersive media. However, existing methods lack explicit motion control and struggle to generate scenes with large and complex motions. We propose PanFlow a novel approach that exploits the sphe

Cited by 0SourcePDFScholar
2026

Probing How Scalable Table Data Enhances General Long-Context Reasoning

ICML 2026poster

As real-world tasks grow increasingly complex, long-context reasoning has become a core capability for Large Language Models (LLMs). However, few studies explore which data types are effective for long-context reasoning and why. We find that structured table data with periodic structures shows stron…

Cited by 0SourceScholar
2026

SEMC: Structure-Enhanced Mixture-of-Experts Contrastive Learning for Ultrasound Standard Plane Recognition

AAAI 2026technical

Ultrasound standard plane recognition is essential for clinical tasks such as disease screening, organ evaluation, and biometric measurement. However, existing methods fail to effectively exploit shallow structural information and struggle to capture fine-grained semantic differences through contras

Cited by 0SourcePDFScholar
2026

Stratos: An End-to-End Distillation Pipeline for Customized LLMs Under Distributed Cloud Environments

AAAI 2026technical

The growing industrial demand for customized and cost-efficient large language models (LLMs) is fueled by the rise of vertical, domain-specific tasks and the need to optimize performance under constraints such as latency and budget. Knowledge distillation, as an efficient model compression and trans

Cited by 0SourcePDFScholar
2026

Uncertainty Matters in Dynamic Gaussian Splatting for Monocular 4D Reconstruction

ICLR 2026poster

Reconstructing dynamic 3D scenes from monocular input is fundamentally under-constrained, with ambiguities arising from occlusion and extreme novel views. While dynamic Gaussian Splatting offers an efficient representation, vanilla models optimize all Gaussian primitives uniformly, ignoring whether…

Cited by 0SourceScholar
2026

Unified Camera Positional Encoding for Controlled Video Generation

CVPR 2026

Transformers have emerged as a universal backbone across 3D perception, video generation, and world models for autonomous driving and embodied AI, where understanding camera geometry is essential for grounding visual observations in three-dimensional space. However, existing camera encoding methods

Cited by 0SourcecodeScholar
2026

Variational Autoencoding Discrete Diffusion with Enhanced Dimensional Correlations Modeling

ICLR 2026poster

Discrete diffusion models have recently shown great promise for modeling complex discrete data, with masked diffusion models (MDMs) offering a compelling trade-off between quality and generation speed. MDMs denoise by progressively unmasking multiple dimensions from an all-masked input, but their pe…

Cited by 0SourcecodeScholar
2026

What Characterizes Effective Reasoning? Revisiting Length, Review, and Structure of CoT

ICML 2026poster

Large reasoning models (LRMs) spend substantial test-time compute on long chain-of-thought (CoT) traces, but what *characterizes* an effective CoT remains unclear. While prior work reports gains from lengthening CoTs and increasing review via appended *wait* tokens, recent studies suggest that short…

Cited by 0SourceScholar
2025

Calibrating Verbal Uncertainty as a Linear Feature to Reduce Hallucinations

EMNLP 2025

LLMs often adopt an assertive language style also when making false claims. Such ”overconfident hallucinations” mislead users and erode trust. Achieving the ability to express in language the actual degree of uncertainty around a claim is therefore of great importance. We find that ”verbal uncertain

Cited by 0SourcePDFScholar
2025

Digital Twin Catalog: A Large-Scale Photorealistic 3D Object Digital Twin Dataset

CVPR 2025highlight

We introduce Digital Twin Catalog (DTC), a new large-scale photorealistic 3D object digital twin dataset. A digital twin of a 3D object is a highly detailed, virtually indistinguishable representation of a physical object, accurately capturing its shape, appearance, physical properties, and other at…

2025

HVI: A New Color Space for Low-light Image Enhancement

CVPR 2025poster

Low-Light Image Enhancement (LLIE) is a crucial computer vision task that aims to restore detailed visual information from corrupted low-light images. Many existing LLIE methods are based on standard RGB (sRGB) space, which often produce color bias and brightness artifacts due to inherent high color…

2025

HalluLens: LLM Hallucination Benchmark

ACL 2025long

Large language models (LLMs) often generate responses that deviate from user input or training data, a phenomenon known as “hallucination.” These hallucinations undermine user trust and hinder the adoption of generative AI systems. Addressing hallucinations is important for the advancement of LLMs.…

2025

Hardware and Software Platform Inference

ICML 2025poster

It is now a common business practice to buy access to large language model (LLM) inference rather than self-host, because of significant upfront hardware infrastructure and energy costs. However, as a buyer, there is no mechanism to verify the authenticity of the advertised service including the ser…

Cited by 0SourcePDFScholar
2025

Integrating Learning-Based Manipulation and Physics-Based Locomotion for Whole-Body Badminton Robot Control

ICRA 2025

Learning-based methods, such as imitation learning (IL) and reinforcement learning (RL), can produce excel control policies over challenging agile robot tasks, such as sports robot. However, no existing work has harmonized learning-based policy with model-based methods to reduce training complexity

Cited by 6SourceScholar
2025

PanSplat: 4K Panorama Synthesis with Feed-Forward Gaussian Splatting

CVPR 2025poster

With the advent of portable 360deg cameras, panorama has gained significant attention in applications like virtual reality (VR), virtual tours, robotics, and autonomous driving. As a result, wide-baseline panorama view synthesis has emerged as a vital task, where high resolution, fast inference, and…

2025

Panorama Generation From NFoV Image Done Right

CVPR 2025highlight

Generating 360-degree panoramas from narrow field of view (NFoV) image is a promising computer vision task for Virtual Reality (VR) applications. Existing methods mostly assess the generated panoramas with InceptionNet or CLIP based metrics, which tend to perceive the image quality and is not suitab…

2025

PhyloVAE: Unsupervised Learning of Phylogenetic Trees via Variational Autoencoders

ICLR 2025poster

Learning informative representations of phylogenetic tree structures is essential for analyzing evolutionary relationships. Classical distance-based methods have been widely used to project phylogenetic trees into Euclidean space, but they are often sensitive to the choice of distance metric and may…

2025

Provable Sample-Efficient Transfer Learning Conditional Diffusion Models via Representation Learning

NeurIPS 2025poster

While conditional diffusion models have achieved remarkable success in various applications, they require abundant data to train from scratch, which is often infeasible in practice. To address this issue, transfer learning has emerged as an essential paradigm in small data regimes. Despite its empir…

Cited by 0SourceScholar
2025

QERA: an Analytical Framework for Quantization Error Reconstruction

ICLR 2025poster

The growing number of parameters and computational demands of large language models (LLMs) present significant challenges for their efficient deployment. Recently, there is an increasing interest in quantizing weights to extremely low precision while offsetting the resulting error with low-rank, hig…

Cited by 0SourcePDFScholar
2025

Refining Salience-Aware Sparse Fine-Tuning Strategies for Language Models

ACL 2025long

Parameter-Efficient Fine-Tuning (PEFT) has gained prominence through low-rank adaptation methods like LoRA. In this paper, we focus on sparsity-based PEFT (SPEFT), which introduces trainable sparse adaptations to the weight matrices in the model, offering greater flexibility in selecting fine-tuned…

2025

Rethinking Removal Attack and Fingerprinting Defense for Model Intellectual Property Protection: A Frequency Perspective

IJCAI 2025

Training deep neural networks is resource-intensive, making it crucial to protect their intellectual property from infringement. However, current model ownership resolution (MOR) methods predominantly address general removal attacks that involve weight modifications, with limited research considerin

2025

TaxaDiffusion: Progressively Trained Diffusion Model for Fine-Grained Species Generation

ICCV 2025poster

We propose TaxaDiffusion, a taxonomy-informed training framework for diffusion models to generate fine-grained animal images with high morphological and identity accuracy. Unlike standard approaches that treat each species as an independent category, TaxaDiffusion incorporates domain knowledge that…

2024

A Fixed-Point Approach for Causal Generative Modeling

ICML 2024poster

We propose a novel formalism for describing Structural Causal Models (SCMs) as fixed-point problems on causally ordered variables, eliminating the need for Directed Acyclic Graphs (DAGs), and establish the weakest known conditions for their unique recovery given the topological ordering (TO). Based…

2024

An Empirical Study and Analysis of Text-to-Image Generation Using Large Language Model-Powered Textual Representation

ECCV 2024poster

"One critical prerequisite for faithful text-to-image generation is the accurate understanding of text inputs. Existing methods leverage the text encoder of the CLIP model to represent input prompts. However, the pre-trained CLIP model can merely encode English with a maximum token length of 77. Mor…

2024

Bio-Inspired Pupal-Mode Actuator with Ultra-Crossing Capability for Soft Robots

ICRA 2024poster

Robot-assisted Natural Orifice Translu-minal Endoscopic Surgery (NOTES) represents a paradigm shift in surgical practice, significantly mini-mizing patient morbidity. However, the variability of inner diameter and the inter-luminal crossing within the luminal tracts lead to challenge for effective r…

Cited by 0SourceScholar
2024

EasyTS: The Express Lane to Long Time Series Forecasting

AAAI 2024technical

Responding to the escalating interest in long-term forecasting within the industry, we introduce EasyTS, a comprehensive toolkit engineered to streamline data collection, analysis, and model creation procedures. EasyTS acts as a unified solution, driving progress in long-term time series forecasting…

2024

Functional Gradient Flows for Constrained Sampling

NeurIPS 2024poster

Recently, through a unified gradient flow perspective of Markov chain Monte Carlo (MCMC) and variational inference (VI), particle-based variational inference methods (ParVIs) have been proposed that tend to combine the best of both worlds. While typical ParVIs such as Stein Variational Gradient Desc…

2024

Generalizable Human Gaussians for Sparse View Synthesis

ECCV 2024poster

"Recent progress in neural rendering has brought forth pioneering methods, such as NeRF and Gaussian Splatting, which revolutionize view rendering across various domains like AR/VR, gaming, and content creation. While these methods excel at interpolating within the training data, the challenge of ge…

2024

Kernel Semi-Implicit Variational Inference

ICML 2024poster

Semi-implicit variational inference (SIVI) extends traditional variational families with semi-implicit distributions defined in a hierarchical manner. Due to the intractable densities of semi-implicit distributions, classical SIVI often resorts to surrogates of evidence lower bound (ELBO) that would…

2024

LQER: Low-Rank Quantization Error Reconstruction for LLMs

ICML 2024poster

Post-training quantization of Large Language Models (LLMs) is challenging. In this work, we introduce **L**ow-rank **Q**uantization **E**rror **R**eduction (LQER), which combines quantization and low-rank approximation to recover the model capability. LQER leverages an activation-induced scale matri…

2024

Make Your ViT-based Multi-view 3D Detectors Faster via Token Compression

ECCV 2024poster

"Slow inference speed is one of the most crucial concerns for deploying multi-view 3D detectors to tasks with high real-time requirements like autonomous driving. Although many sparse query-based methods have already attempted to improve the efficiency of 3D detectors, they neglect to consider the b…

2024

ProAgent: Building Proactive Cooperative Agents with Large Language Models

AAAI 2024technical

Building agents with adaptive behavior in cooperative tasks stands as a paramount goal in the realm of multi-agent systems. Current approaches to developing cooperative agents rely primarily on learning-based methods, whose policy generalization depends heavily on the diversity of teammates they int…

2024

Taming Stable Diffusion for Text to 360 Panorama Image Generation

CVPR 2024highlight

Generative models e.g. Stable Diffusion have enabled the creation of photorealistic images from text prompts. Yet the generation of 360-degree panorama images from text remains a challenge particularly due to the dearth of paired text-panorama data and the domain gap between panorama and perspective…

2024

TextureDreamer: Image-Guided Texture Synthesis Through Geometry-Aware Diffusion

CVPR 2024poster

We present TextureDreamer a novel image-guided texture synthesis method to transfer relightable textures from a small number of input images (3 to 5) to target 3D shapes across arbitrary categories. Texture creation is a pivotal challenge in vision and graphics. Industrial companies hire experienced…

2024

Towards Causal Foundation Model: on Duality between Optimal Balancing and Attention

ICML 2024poster

Foundation models have brought changes to the landscape of machine learning, demonstrating sparks of human-level intelligence across a diverse array of tasks. However, a gap persists in complex tasks such as causal inference, primarily due to challenges associated with intricate reasoning steps and…

Cited by 4SourcePDFScholar
2024

Towards Effective Usage of Human-Centric Priors in Diffusion Models for Text-based Human Image Generation

CVPR 2024poster

Vanilla text-to-image diffusion models struggle with generating accurate human images commonly resulting in imperfect anatomies such as unnatural postures or disproportionate limbs. Existing methods address this issue mostly by fine-tuning the model with extra images or adding additional controls --…

Cited by 9SourcePDFScholar
2024

Visual Data Diagnosis and Debiasing with Concept Graphs

NeurIPS 2024poster

The widespread success of deep learning models today is owed to the curation of extensive datasets significant in size and complexity. However, such models frequently pick up inherent biases in the data during the training process, leading to unreliable predictions. Diagnosing and debiasing datasets…

2023

BayesDAG: Gradient-Based Posterior Inference for Causal Discovery

NeurIPS 2023poster

Bayesian causal discovery aims to infer the posterior distribution over causal models from observed data, quantifying epistemic uncertainty and benefiting downstream tasks. However, computational challenges arise due to joint inference over combinatorial space of Directed Acyclic Graphs (DAGs) and n…

2023

CO-BED: Information-Theoretic Contextual Optimization via Bayesian Experimental Design

ICML 2023poster

We formalize the problem of contextual optimization through the lens of Bayesian experimental design and propose CO-BED---a general, model-agnostic framework for designing contextual experiments using information-theoretic principles. After formulating a suitable information-based objective, we empl…

2023

Causal Reasoning in the Presence of Latent Confounders via Neural ADMG Learning

ICLR 2023poster

Latent confounding has been a long-standing obstacle for causal reasoning from observational data. One popular approach is to model the data using acyclic directed mixed graphs (ADMGs), which describe ancestral relations between variables using directed and bidirected edges. However, existing method…

2023

Hierarchical Semi-Implicit Variational Inference with Application to Diffusion Model Acceleration

NeurIPS 2023poster

Semi-implicit variational inference (SIVI) has been introduced to expand the analytical variational families by defining expressive semi-implicit distributions in a hierarchical manner. However, the single-layer architecture commonly used in current SIVI methods can be insufficient when the target p…

2023

ITI-GEN: Inclusive Text-to-Image Generation

ICCV 2023oral

Text-to-image generative models often reflect the biases of the training data, leading to unequal representations of underrepresented groups. This study investigates inclusive text-to-image generative models that generate images based on human-written prompts and ensure the resulting images are unif…

Cited by 66PDFcodeScholar
2023

Neural-PBIR Reconstruction of Shape, Material, and Illumination

ICCV 2023poster

Reconstructing the shape and spatially varying surface appearances of a physical-world object as well as its surrounding illumination based on 2D images (e.g., photographs) of the object has been a long-standing problem in computer vision and graphics. In this paper, we introduce an accurate and hig…

Cited by 30PDFcodeScholar
2023

Particle-based Variational Inference with Generalized Wasserstein Gradient Flow

NeurIPS 2023poster

Particle-based variational inference methods (ParVIs) such as Stein variational gradient descent (SVGD) update the particles based on the kernelized Wasserstein gradient flow for the Kullback-Leibler (KL) divergence. However, the design of kernels is often non-trivial and can be restrictive for the…

Cited by 8SourcePDFScholar
2023

Revisiting Block-based Quantisation: What is Important for Sub-8-bit LLM Inference?

EMNLP 2023long main

The inference of Large language models (LLMs) requires immense computation and memory resources. To curtail these costs, quantisation has emerged as a promising solution, but existing LLM quantisation mainly focuses on 8-bit. In this work, we explore the statistical and learning properties of the LL…

Cited by 0SourcecodeScholar
2023

Rhino: Deep Causal Temporal Relationship Learning with History-dependent Noise

ICLR 2023top-25%

Discovering causal relationships between different variables from time series data has been a long-standing challenge for many domains. For example, in stock markets, the announcement of acquisitions from leading companies may have immediate effects on stock prices and increase the uncertainty of th…

Cited by 39SourcePDFScholar
2023

SAM-RL: Sensing-Aware Model-Based Reinforcement Learning via Differentiable Physics-Based Simulation and Rendering

RSS 2023poster

Model-based reinforcement learning (MBRL) is recognized with the potential to be significantly more sample efficient than model-free RL. How an accurate model can be developed automatically and efficiently from raw sensory inputs (such as images), especially for complex environments and tasks, is a…

Cited by 28SourcePDFScholar
2023

TokenHPE: Learning Orientation Tokens for Efficient Head Pose Estimation via Transformers

CVPR 2023poster

Head pose estimation (HPE) has been widely used in the fields of human machine interaction, self-driving, and attention estimation. However, existing methods cannot deal with extreme head pose randomness and serious occlusions. To address these challenges, we identify three cues from head images, na…

2022

Crossroads, Buildings and Neighborhoods: A Dataset for Fine-grained Location Recognition

NAACL 2022long

General domain Named Entity Recognition (NER) datasets like CoNLL-2003 mostly annotate coarse-grained location entities such as a country or a city. But many applications require identifying fine-grained locations from texts and mapping them precisely to geographic sites, e.g., a crossroad, an apart…

2022

Exploring and Evaluating Image Restoration Potential in Dynamic Scenes

CVPR 2022poster

In dynamic scenes, images often suffer from dynamic blur due to superposition of motions or low signal-noise ratio resulted from quick shutter speed when avoiding motions. Recovering sharp and clean result from the captured images heavily depends on the ability of restoration methods and the quality…

Cited by 13PDFcodeScholar
2022

Fingerprinting Deep Neural Networks Globally via Universal Adversarial Perturbations

CVPR 2022oral

In this paper, we propose a novel and practical mechanism which enables the service provider to verify whether a suspect model is stolen from the victim model via model extraction attacks. Our key insight is that the profile of a DNN model's decision boundary can be uniquely characterized by its Uni…

Cited by 87PDFScholar
2022

Learning with Free Object Segments for Long-Tailed Instance Segmentation

ECCV 2022poster

"One fundamental challenge in building an instance segmentation model for a large number of classes in complex scenes is the lack of training examples, especially for rare objects. In this paper, we explore the possibility to increase the training examples without laborious data collection and annot…

2022

MCS: An In-battle Commentary System for MOBA Games

COLING 2022main

This paper introduces a generative system for in-battle real-time commentary in mobile MOBA games. Event commentary is important for battles in MOBA games, which is applicable to a wide range of scenarios like live streaming, e-sports commentary and combat information analysis. The system takes real…

Cited by 1SourcePDFScholar
2022

Simultaneous Missing Value Imputation and Structure Learning with Groups

NeurIPS 2022accept

Learning structures between groups of variables from data with missing values is an important task in the real world, yet difficult to solve. One typical scenario is discovering the structure among topics in the education domain to identify learning pathways. Here, the observations are student perfo…

Cited by 22SourcePDFScholar
2021

DeepPanoContext: Panoramic 3D Scene Understanding With Holistic Scene Context Graph and Relation-Based Optimization

ICCV 2021poster

Panorama images have a much larger field-of-view thus naturally encode enriched scene context information compared to standard perspective images, which however is not well exploited in the previous scene understanding methods. In this paper, we propose a novel method for panoramic 3D scene understa…

Cited by 40PDFcodeScholar
2021

Discovering the Unknown Knowns: Turning Implicit Knowledge in the Dataset into Explicit Training Examples for Visual Question Answering

EMNLP 2021main

Visual question answering (VQA) is challenging not only because the model has to handle multi-modal information, but also because it is just so hard to collect sufficient training examples — there are too many questions one can ask about an image. As a result, a VQA model trained solely on human-ann…

2021

Holistic 3D Scene Understanding From a Single Image With Implicit Representation

CVPR 2021poster

We present a new pipeline for holistic 3D scene understanding from a single image, which could predict object shape, object pose and scene layout. As it is a highly ill-posed problem, existing methods usually suffer from inaccurate estimation of both shapes and layout especially for the cluttered sc…

Cited by 129PDFcodeScholar
2021

Meta-Learning Divergences for Variational Inference

AISTATS 2021poster

Variational inference (VI) plays an essential role in approximate Bayesian inference due to its computational efficiency and broad applicability. Crucial to the performance of VI is the selection of the associated divergence measure, as VI approximates the intractable distribution by minimizing this…

Cited by 7SourcePDFScholar
2021

MosaicOS: A Simple and Effective Use of Object-Centric Images for Long-Tailed Object Detection

ICCV 2021poster

Many objects do not appear frequently enough in complex scenes (e.g., certain handbags in living rooms) for training an accurate object detector, but are often found frequently by themselves (e.g., in product images). Yet, these object-centric images are not effectively leveraged for improving objec…

Cited by 50PDFcodeScholar
2021

On Model Calibration for Long-Tailed Object Detection and Instance Segmentation

NeurIPS 2021poster

Vanilla models for object detection and instance segmentation suffer from the heavy bias toward detecting frequent objects in the long-tailed setting. Existing methods address this issue mostly during training, e.g., by re-sampling or re-weighting. In this paper, we investigate a largely overlooked…

2021

Sparse Uncertainty Representation in Deep Learning with Inducing Weights

NeurIPS 2021poster

Bayesian Neural Networks and deep ensembles represent two modern paradigms of uncertainty quantification in deep learning. Yet these approaches struggle to scale mainly due to memory inefficiency, requiring parameter storage several times that of their deterministic counterparts. To address this, we…

Cited by 22SourcePDFScholar
2020

AMRL: Aggregated Memory For Reinforcement Learning

ICLR 2020poster

In many partially observable scenarios, Reinforcement Learning (RL) agents must rely on long-term memory in order to learn an optimal policy. We demonstrate that using techniques from NLP and supervised learning fails at RL tasks due to stochasticity from the environment and from exploration. Utiliz…

Cited by 26SourceScholar
2020

Blindly Assess Image Quality in the Wild Guided by a Self-Adaptive Hyper Network

CVPR 2020poster

Blind image quality assessment (BIQA) for authentically distorted images has always been a challenging problem, since images captured in the wild include varies contents and diverse types of distortions. The vast majority of prior BIQA methods focus on how to predict synthetic image quality, but fai…

Cited by 788PDFcodeScholar
2020

How do fair decisions fare in long-term qualification?

NeurIPS 2020poster

Although many fairness criteria have been proposed for decision making, their long-term impact on the well-being of a population remains unclear. In this work, we study the dynamics of population qualification and algorithmic decisions under a partially observed Markov decision problem setting. By c…

2020

VAEM: a Deep Generative Model for Heterogeneous Mixed Type Data

NeurIPS 2020poster

Deep generative models often perform poorly in real-world applications due to the heterogeneity of natural data sets. Heterogeneity arises from data containing different types of features (categorical, ordinal, continuous, etc.) and features of the same type having different marginal distributions.…

2019

Causal Discovery in the Presence of Missing Data

AISTATS 2019poster

Missing data are ubiquitous in many domains such as healthcare. When these data entries are not missing completely at random, the (conditional) independence relations in the observed data may be different from those in the complete data generated by the underlying causal process. Consequently, simpl…

2019

EDDI: Efficient Dynamic Discovery of High-Value Information with Partial VAE

ICML 2019oral

Many real-life decision making situations allow further relevant information to be acquired at a specific cost, for example, in assessing the health status of a patient we may decide to take additional measurements such as diagnostic tests or imaging scans before making a final assessment. Acquiring…

2019

Generalization in Reinforcement Learning with Selective Noise Injection and Information Bottleneck

NeurIPS 2019poster

The ability for policies to generalize to new environments is key to the broad application of RL agents. A promising approach to prevent an agent’s policy from overfitting to a limited set of training environments is to apply regularization techniques originally developed for supervised learning. Ho…

2019

Icebreaker: Element-wise Efficient Information Acquisition with a Bayesian Deep Latent Gaussian Model

NeurIPS 2019poster

In this paper, we address the ice-start problem, i.e., the challenge of deploying machine learning models when only a little or no training data is initially available, and acquiring each feature element of data is associated with costs. This setting is representative of the real-world machine learn…

2019

L2 Learners' Emotion Production in Video Dubbing Practices

ICASSP 2019accepted

Video dubbing is a new type of language learning practice. Because of the fun it brings into learning, video dubbing mobile applications have become quite popular. During video dubbing, learners not only mimic characters' pronunciations but also other voicing characteristics, e.g., emotions. In this…

Cited by 4SourceScholar
2019

Neuropathic Pain Diagnosis Simulator for Causal Discovery Algorithm Evaluation

NeurIPS 2019poster

Discovery of causal relations from observational data is essential for many disciplines of science and real-world applications. However, unlike other machine learning algorithms, whose development has been greatly fostered by a large amount of available benchmark datasets, causal discovery algorithm…

2018

Performance of Interleaved Training for Single-User Hybrid Massive Antenna Downlink

ICASSP 2018accepted

In this paper, we study the beam-based training design for the single-user (SU) hybrid massive antenna system based on outage probability performance. First, an interleaved training design is proposed where the feedback is concatenated with the training procedure to monitor the training status and t…

Cited by 0SourceScholar