← Search

Yuantao Gu

35 accepted papers

2026

Stage-wise Distortion–Perception Traversal in Zero-shot Inverse Problems with Diffusion Models

ICML 2026poster

The distortion–perception (D–P) tradeoff is a fundamental phenomenon of Bayesian inverse problems, which characterizes the inherent tension between distortion performance and perceptual quality. Enabling flexible traversal of the D-P tradeoff at inference time is crucial for practical applications. …

Cited by 0SourceScholar
2025

Angle Domain Guidance: Latent Diffusion Requires Rotation Rather Than Extrapolation

ICML 2025poster

Classifier-free guidance (CFG) has emerged as a pivotal advancement in text-to-image latent diffusion models, establishing itself as a cornerstone technique for achieving high-quality image synthesis. However, under high guidance weights, where text-image alignment is significantly enhanced, CFG als…

2025

Improving Diffusion-based Inverse Algorithms under Few-Step Constraint via Linear Extrapolation

NeurIPS 2025poster

Diffusion-based inverse algorithms have shown remarkable performance across various inverse problems, yet their reliance on numerous denoising steps incurs high computational costs. While recent developments of fast diffusion ODE solvers offer effective acceleration for diffusion sampling without o…

Cited by 0SourcecodeScholar
2024

EPA: Neural Collapse Inspired Robust Out-of-distribution Detector

ICASSP 2024accepted

Out-of-distribution (OOD) detection plays a crucial role in ensuring the security of neural networks. Existing works have leveraged the fact that In-distribution (ID) samples form a subspace in the feature space, achieving state-of-the-art (SOTA) performance. However, the comprehensive characteristi…

Cited by 0SourceScholar
2024

Unleashing the Denoising Capability of Diffusion Prior for Solving Inverse Problems

NeurIPS 2024poster

The recent emergence of diffusion models has significantly advanced the precision of learnable priors, presenting innovative avenues for addressing inverse problems. Previous works have endeavored to integrate diffusion priors into the maximum a posteriori estimation (MAP) framework and design optim…

2024

Unravel Anomalies: an End-to-End Seasonal-Trend Decomposition Approach for Time Series Anomaly Detection

ICASSP 2024accepted

Traditional Time-series Anomaly Detection (TAD) methods often struggle with the composite nature of complex time-series data and a diverse array of anomalies. We introduce TADNet, an end-to-end TAD model that leverages Seasonal-Trend Decomposition to link various types of anomalies to specific decom…

Cited by 0SourceScholar
2023

Incremental Aggregated Riemannian Gradient Method for Distributed PCA

AISTATS 2023poster

We consider the problem of distributed principal component analysis (PCA) where the data samples are dispersed across different agents. Despite the rich literature on this problem under various specific settings, there is still a lack of efficient algorithms that are amenable to decentralized and as…

2021

Breaking the Sample Complexity Barrier to Regret-Optimal Model-Free Reinforcement Learning

NeurIPS 2021spotlight

Achieving sample efficiency in online episodic reinforcement learning (RL) requires optimally balancing exploration and exploitation. When it comes to a finite-horizon episodic Markov decision process with $S$ states, $A$ actions and horizon length $H$, substantial progress has been achieved toward…

Cited by 64SourcePDFScholar
2021

Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisiting

NeurIPS 2021poster

Low-complexity models such as linear function representation play a pivotal role in enabling sample-efficient reinforcement learning (RL). The current paper pertains to a scenario with value-based linear representation, which postulates linear realizability of the optimal Q-function (also called the…

Cited by 36SourcePDFScholar
2021

Tightening the Dependence on Horizon in the Sample Complexity of Q-Learning

ICML 2021spotlight

Q-learning, which seeks to learn the optimal Q-function of a Markov decision process (MDP) in a model-free fashion, lies at the heart of reinforcement learning. Focusing on the synchronous setting (such that independent samples for all state-action pairs are queried via a generative model in each it…

Cited by 23SourcePDFScholar
2020

Breaking the Sample Size Barrier in Model-Based Reinforcement Learning with a Generative Model

NeurIPS 2020poster

We investigate the sample efficiency of reinforcement learning in a $\gamma$-discounted infinite-horizon Markov decision process (MDP) with state space S and action space A, assuming access to a generative model. Despite a number of prior work tackling this problem, a complete picture of the trade-…

Cited by 152SourcePDFScholar
2020

Principal Angle Detector for Subspace Signal with Structured Unknown Interference

ICASSP 2020accepted

Detecting subspace signals is an important problem in radar and sonar signal processing, hyperspectral image processing, wireless communication, and other fields. Among these problems, a typical scenario is that one needs to detect a signal lying in a given target subspace, contaminated by interfere…

Cited by 0SourceScholar
2020

Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance Reduction

NeurIPS 2020poster

Asynchronous Q-learning aims to learn the optimal action-value function (or Q-function) of a Markov decision process (MDP), based on a single trajectory of Markovian samples induced by a behavior policy. Focusing on a $\gamma$-discounted MDP with state space S and action space A, we demonstrate tha…

Cited by 107SourcePDFScholar
2019

Enhanced Streaming Based Subspace Clustering Applied to Acoustic Scene Data Clustering

ICASSP 2019accepted

Labelled data are often required to train an acoustic scene classification system. However, it is time-consuming and expensive to label the data manually. An unsupervised clustering algorithm can be used to facilitate the labelling process by dividing the acoustic data into different categories. Nev…

Cited by 0SourceScholar
2018

Change-Point Detection of Gaussian Graph Signals with Partial Information

ICASSP 2018accepted

In a change-point detection problem, a sequence of signals switches from one distribution to another at an unknown time step, and the goal is to quickly and reliably detect this change. By providing new insight into signal processing and data analysis, graph signal processing promises various applic…

Cited by 0SourceScholar
2018

Convergence Analysis on a Fast Iterative Phase Retrieval Algorithm Without Independence Assumption

ICASSP 2018accepted

Phase retrieval has been an attractive problem, and many algorithms have been proposed. Randomized Kaczmarz method is a fast iterative method with good performance in both convergence rate and computational cost with theoretical analysis. However, they all assume that the iteratively updated variabl…

Cited by 0SourceScholar
2016

Beyond union of subspaces: Subspace pursuit on Grassmann manifold for data representation

ICASSP 2016accepted

Discovering the underlying structure of a high-dimensional signal or big data has always been a challenging topic, and has become harder to tackle especially when the observations are exposed to arbitrary sparse perturbations. In this paper, built on the model of a union of subspaces (UoS) with spar…

Cited by 0SourceScholar
2015

Averaging random projection: A fast online solution for large-scale constrained stochastic optimization

ICASSP 2015accepted

Stochastic optimization finds wide application in signal processing, online learning, and network problems, especially problems processing large-scale data. We propose an Incremental Constraint Averaging Projection Method (ICAPM) that is tailored to optimization problems involving a large number of…

Cited by 0SourceScholar
2015

Dynamic zero-point attracting projection for time-varying sparse signal recovery

ICASSP 2015accepted

Sparse signal recovery in the static case has been well studied under the framework of Compressive Sensing (CS), while in recent years more attention has also been paid to the dynamic case. In this paper, enlightened by the idea of modified-CS with partially known support, and based on a non-convex…

Cited by 0SourceScholar