EDBT 2026 Demo / reviewers in the wild / expert
Shaofeng Zou
dblp:135/4981
· DBLP profile ↗
68ranked-venue papers
17as first author
40since 2021 · last 2026
0000-0002-2821-6941ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 28 · 1 first-author · 23 since 2021Applied, interdisciplinary, general and emerging computing · 16 · 5 first-author · 6 since 2021Graphics, computer vision, multimedia, augmented reality and games · 12 · 3 first-author · 5 since 2021Theory of computation · 12 · 6 first-author · 5 since 2021Systems, architecture and hardware · 4 · 2 first-author · 4 since 2021Security and privacy · 2 · 2 since 2021
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Minimax Optimal Sample Complexity for Iterated CVaR Reinforcement Learning With a Generative ModelabstractStandard Reinforcement Learning (RL) algorithms are typically designed to maximize the expected accumulative reward, which may be inadequate in scenarios where risk sensitivity is critical. In this work, the problem of risk-sensitive RL with Iterated Conditional Value at Risk is studied, where the objective is to optimize outcomes under a specified risk level τ at each step. This work provides the first minimax optimal sample complexity analysis for this problem with a generative model. Specifically, the sample complexity is firstly characterized as a function of the number of statesS, actionsA, and effective horizon (1−γ)−1(resp. horizonHin the finite-horizon setting), and is further shown to be minimax optimal via a novel minimax lower bound analysis when the risk level 0Hin the finite horizon setting). For the case when the risk level is small, the limiting case of τ → 0, termed worst-path RL, is then studied, and the minimax optimal sample complexity is also theoretically characterized. Zilong Deng, Alvaro Velasquez, Shaofeng Zou |
IEEE Trans. Inf. Theory | 3 |
| 2025 | Near-Optimal Sample Complexity for Iterated CVaR Reinforcement Learning with a Generative ModelabstractIn this work, we study the sample complexity problem of risk-sensitive Reinforcement Learning (RL) with a generative model, where we aim to maximize the Conditional Value at Risk (CVaR) with risk tolerance level $\tau$ at each step, named Iterated CVaR. We develop nearly matching upper and lower bounds on the sample complexity for this problem. Specifically, we first prove that a value iteration-based algorithm, ICVaR-VI, achieves an $\epsilon$-optimal policy with at most $\tilde{\mathcal{O}}\left(\frac{SA}{(1-\gamma)^4\tau^2\epsilon^2}\right)$ samples, where $\gamma$ is the discount factor, and $S, A$ are the sizes of the state and action spaces. Furthermore, if $\tau \geq \gamma$, then the sample complexity can be further improved to $\tilde{\mathcal{O}}\left( \frac{SA}{(1-\gamma)^3\epsilon^2} \right)$. We further show a minimax lower bound of ${\tilde{\mathcal{O}}}\left(\frac{(1-\gamma \tau)SA}{(1-\gamma)^4\tau\epsilon^2}\right)$. For a constant risk level $0<\tau\leq 1$, our upper and lower bounds match with each other, demonstrating the tightness and optimality of our analyses. We also investigate a limiting case with a small risk level $\tau$, called Worst-Path RL, where the objective is to maximize the minimum possible cumulative reward. We develop matching upper and lower bounds of $\tilde{\mathcal{O}}\left(\frac{SA}{p_{\min}}\right)$, where $p_{\min}$ denotes the minimum non-zero reaching probability of the transition kernel. Zilong Deng, Simon Khan, Shaofeng Zou |
AISTATS | 3 |
| 2025 | MGDA Converges under Generalized Smoothness, ProvablyabstractMulti-objective optimization (MOO) is receiving more attention in various fields such as multi-task learning. Recent works provide some effective algorithms with theoretical analysis but they are limited by the standard $L$-smooth or bounded-gradient assumptions, which typically do not hold for neural networks, such as Long short-term memory (LSTM) models and Transformers. In this paper, we study a more general and realistic class of generalized $\ell$-smooth loss functions, where $\ell$ is a general non-decreasing function of gradient norm. We revisit and analyze the fundamental multiple gradient descent algorithm (MGDA) and its stochastic version with double sampling for solving the generalized $\ell$-smooth MOO problems, which approximate the conflict-avoidant (CA) direction that maximizes the minimum improvement among objectives. We provide a comprehensive convergence analysis of these algorithms and show that they converge to an $\epsilon$-accurate Pareto stationary point with a guaranteed $\epsilon$-level average CA distance (i.e., the gap between the updating direction and the CA direction) over all iterations, where totally $\mathcal{O}(\epsilon^{-2})$ and $\mathcal{O}(\epsilon^{-4})$ samples are needed for deterministic and stochastic settings, respectively. We prove that they can also guarantee a tighter $\epsilon$-level CA distance in each iteration using more samples. Moreover, we analyze an efficient variant of MGDA named MGDA-FA using only $\mathcal{O}(1)$ time and space, while achieving the same performance guarantee as MGDA. Qi Zhang 0069, Peiyao Xiao, Shaofeng Zou, Kaiyi Ji |
ICLR | 3 |
| 2025 | Revisiting Large-Scale Non-convex Distributionally Robust OptimizationabstractDistributionally robust optimization (DRO) is a powerful technique to train robust machine learning models that perform well under distribution shifts. Compared with empirical risk minimization (ERM), DRO optimizes the expected loss under the worst-case distribution in
an uncertainty set of distributions. This paper revisits the important problem of DRO with non-convex smooth loss functions. For this problem, Jin et al. (2021) showed that its dual problem is generalized $(L_0, L_1)$-smooth condition and gradient noise satisfies the affine variance condition, designed an algorithm of mini-batch normalized gradient descent with momentum, and proved its convergence and complexity. In this paper, we show that the dual problem and the gradient noise satisfy simpler yet more precise partially generalized smoothness condition and partially affine variance condition by studying the optimization variable and dual variable separately, which further yields much simpler algorithm design and convergence analysis. We develop a double stochastic gradient descent with clipping (D-SGD-C) algorithm that converges to an $\epsilon$-stationary point with $\mathcal O(\epsilon^{-4})$ gradient complexity, which matches with results in Jin et al. (2021). Our algorithm does not need to use momentum, and the proof is much simpler, thanks to the more precise characterization of partially generalized smoothness and partially affine variance noise. We further design a variance-reduced method that achieves a lower gradient complexity of $\mathcal O(\epsilon^{-3})$. Our theoretical results and insights are further verified numerically on a number of tasks, and our algorithms outperform the existing DRO method (Jin et al., 2021). Qi Zhang 0069, Yi Zhou 0017, Simon Khan, Ashley Prater-Bennette, Lixin Shen, Shaofeng Zou |
ICLR | 6 |
| 2025 | Rejecting Outliers in 2D-3D Point Correspondences from 2D Forward-Looking Sonar ObservationsabstractRejecting outliers before applying classical robust methods is a common approach to increase the success rate of estimation, particularly when the outlier ratio is extremely high (e.g. 90%). However, this method often relies on sensor- or task-specific characteristics, which may not be easily transferable across different scenarios. In this paper, we focus on the problem of rejecting 2D-3D point correspondence outliers from 2D forward-looking sonar (2D FLS) observations, which is one of the most popular perception device in the underwater field but has a significantly different imaging mechanism compared to widely used perspective cameras and LiDAR. We fully leverage the narrow field of view in the elevation of 2D FLS and develop two compatibility tests for different 3D point configurations: (1) In general cases, we design a pairwise length in-range test to filter out overly long or short edges formed from point sets; (2) In coplanar cases, we design a coplanarity test to check if any four correspondences are compatible under a coplanar setting. Both tests are integrated into outlier rejection pipelines, where they are followed by maximum clique searching to identify the largest consistent measurement set as inliers. Extensive simulations demonstrate that the proposed methods for general and coplanar cases perform effectively under outlier ratios of 80% and 90%, respectively. Jiayi Su, Shaofeng Zou, Jingyu Qian, Fengzhong Qu, Liuqing Yang 0001 |
IROS | 2 |
| 2025 | Understanding Information Disclosure from Secure Computation Output: A Comprehensive Study of Average Salary ComputationabstractSecure multi-party computation has seen substantial performance improvements in recent years and is being increasingly used in commercial products. While a significant amount of work was dedicated to improving its efficiency under standard security models, the threat models do not account for information leakage from the output of secure function evaluation. Quantifying information disclosure about private inputs from observing the function outcome is the subject of this work. Motivated by the City of Boston gender pay gap studies, in this work, we focus on the computation of the average of salaries and quantify information disclosure about private inputs of one or more participants (the target) to an adversary via information-theoretic techniques. We study a number of distributions including log-normal, which is typically used for modeling salaries. We consequently evaluate information disclosure after repeated evaluation of the average function on overlapping inputs, as was done in the Boston gender pay study that ran multiple times, and provide recommendations for using the sum and average functions in secure computation applications. Our goal is to develop mechanisms that lower information disclosure about participants’ inputs to a desired level and provide guidelines for setting up real-world secure evaluation of this function. Alessandro N. Baccarini, Marina Blanton, Shaofeng Zou |
ACM Trans. Priv. Secur. | 3 |
| 2025 | Theoretical Study of Conflict-Avoidant Multi-Objective Reinforcement LearningabstractMulti-objective reinforcement learning (MORL) has shown great promise in many real-world applications. Existing MORL algorithms often aim to learn a policy that optimizes individual objective functions simultaneously with a given prior preference (or weights) on different objectives. However, these methods often suffer from the issue ofgradient conflictsuch that the objectives with larger gradients dominate the update direction, resulting in a performance degeneration on other objectives. In this paper, we develop a novel dynamic weighting multi-objective actor-critic algorithm (MOAC) under two options of sub-procedures named as conflict-avoidant (CA) and faster convergence (FC) in objective weight updates. MOAC-CA aims to find a CA update direction that maximizes the minimum value improvement among objectives, and MOAC-FC targets at a much faster convergence rate. We provide a comprehensive finite-time convergence analysis for both algorithms. We show that MOAC-CA can find a ϵ + ϵapp-accurate Pareto stationary policy using O(ϵ−5) samples, while ensuring a small ϵ+ √ϵapp-level CA distance (defined as the distance to the CA direction), where ϵappis the function approximation error. The analysis also shows that MOAC-FC improves the sample complexity to O(ϵ−3), but with a constant-level CA distance. Our experiments on MT10 demonstrate the improved performance of our algorithms over existing MORL methods with fixed preference. Yudan Wang, Peiyao Xiao, Hao Ban, Kaiyi Ji, Shaofeng Zou |
IEEE Trans. Inf. Theory | 5 |
| 2024 | Large-Scale Non-convex Stochastic Constrained Distributionally Robust OptimizationabstractDistributionally robust optimization (DRO) is a powerful framework for training robust models against data distribution shifts. This paper focuses on constrained DRO, which has an explicit characterization of the robustness level. Existing studies on constrained DRO mostly focus on convex loss function, and exclude the practical and challenging case with non-convex loss function, e.g., neural network. This paper develops a stochastic algorithm and its performance analysis for non-convex constrained DRO. The computational complexity of our stochastic algorithm at each iteration is independent of the overall dataset size, and thus is suitable for large-scale applications. We focus on the general Cressie-Read family divergence defined uncertainty set which includes chi^2-divergences as a special case. We prove that our algorithm finds an epsilon-stationary point with an improved computational complexity than existing methods. Our method also applies to the smoothed conditional value at risk (CVaR) DRO. Qi Zhang 0069, Yi Zhou 0017, Ashley Prater-Bennette, Lixin Shen, Shaofeng Zou |
AAAI | 5 |
| 2024 | Sample Complexity Characterization for Linear Contextual MDPsabstractContextual Markov decision processes (CMDPs) describe a class of reinforcement learning problems in which the transition kernels and reward functions can change over time with different MDPs indexed by a context variable. While CMDPs serve as an important framework to model many real-world applications with time-varying environments, they are largely unexplored from theoretical perspective. In this paper, we study CMDPs under two linear function approximation models: Model I with context-varying representations and common linear weights for all contexts; and Model II with common representations for all contexts and context-varying linear weights. For both models, we propose novel model-based algorithms and show that they enjoy guaranteed $\epsilon$-suboptimality gap with desired polynomial sample complexity. In particular, instantiating our result for the first model to the tabular CMDP improves the existing result by removing the reachability assumption. Our result for the second model is the first-known result for such a type of function approximation models. Comparison between our results for the two models further indicates that having context-varying features leads to much better sample efficiency than having common representations for all contexts under linear CMDPs. Junze Deng, Shaofeng Zou, Yingbin Liang |
AISTATS | 3 |
| 2024 | Understanding Information Disclosure from Secure Computation Output: A Study of Average Salary ComputationabstractSecure multi-party computation has seen substantial performance improvements in recent years and is being increasingly used in commercial products. While a significant amount of work was dedicated to improving its efficiency under standard security models, the threat models do not account for information leakage from the output of secure function evaluation. Quantifying information disclosure about private inputs from observing the function outcome is the subject of this work. Motivated by the City of Boston gender pay gap studies, in this work we focus on the computation of the average of salaries and quantify information disclosure about private inputs of one or more participants (the target) to an adversary via information-theoretic techniques. We study a number of distributions including log-normal, which is typically used for modeling salaries. We consequently evaluate information disclosure after repeated evaluation of the average function on overlapping inputs, as was done in the Boston gender pay study that ran multiple times, and provide recommendations for using the sum and average functions in secure computation applications. Our goal is to develop mechanisms that lower information disclosure about participants' inputs to a desired level and provide guidelines for setting up real-world secure evaluation of this function. Alessandro N. Baccarini, Marina Blanton, Shaofeng Zou |
CODASPY | 3 |
| 2024 | Constrained Reinforcement Learning Under Model MismatchabstractExisting studies on constrained reinforcement learning (RL) may obtain a well-performing policy in the training environment. However, when deployed in a real environment, it may easily violate constraints that were originally satisfied during training because there might be model mismatch between the training and real environments. To address this challenge, we formulate the problem as constrained RL under model uncertainty, where the goal is to learn a policy that optimizes the reward and at the same time satisfies the constraint under model mismatch. We develop a Robust Constrained Policy Optimization (RCPO) algorithm, which is the first algorithm that applies to large/continuous state space and has theoretical guarantees on worst-case reward improvement and constraint violation at each iteration during the training. We show the effectiveness of our algorithm on a set of RL tasks with constraints. Zhongchang Sun, Sihong He, Fei Miao, Shaofeng Zou |
ICML | 4 |
| 2024 | Non-Asymptotic Analysis for Single-Loop (Natural) Actor-Critic with Compatible Function ApproximationabstractActor-critic (AC) is a powerful method for learning an optimal policy in reinforcement learning, where the critic uses algorithms, e.g., temporal difference (TD) learning with function approximation, to evaluate the current policy and the actor updates the policy along an approximate gradient direction using information from the critic. This paper provides the *tightest* non-asymptotic convergence bounds for both the AC and natural AC (NAC) algorithms. Specifically, existing studies show that AC converges to an $\epsilon+\varepsilon_{\text{critic}}$ neighborhood of stationary points with the best known sample complexity of $\mathcal{O}(\epsilon^{-2})$ (up to a log factor), and NAC converges to an $\epsilon+\varepsilon_{\text{critic}}+\sqrt{\varepsilon_{\text{actor}}}$ neighborhood of the global optimum with the best known sample complexity of $\mathcal{O}(\epsilon^{-3})$, where $\varepsilon_{\text{critic}}$ is the approximation error of the critic and $\varepsilon_{\text{actor}}$ is the approximation error induced by the insufficient expressive power of the parameterized policy class. This paper analyzes the convergence of both AC and NAC algorithms with compatible function approximation. Our analysis eliminates the term $\varepsilon_{\text{critic}}$ from the error bounds while still achieving the best known sample complexities. Moreover, we focus on the challenging single-loop setting with a single Markovian sample trajectory. Our major technical novelty lies in analyzing the stochastic bias due to policy-dependent and time-varying compatible function approximation in the critic, and handling the non-ergodicity of the MDP due to the single Markovian sample trajectory. Numerical results are also provided in the appendix. Yudan Wang, Yue Wang 0068, Yi Zhou 0017, Shaofeng Zou |
ICML | 4 |
| 2024 | Moving Object Detection in Shallow Underwater using Multi-Scale Spatial-Temporal LacunarityabstractIn shallow underwater environments, active sonar is often utilized for the detection of small moving targets. However, such systems can suffer from high-level background reverberation, making it challenging to detect and distinguish the target echoes from reverberation. To address this challenge, we propose a fast moving object detection method utilizing multi-scale spatial-temporal lacunarity. Specifically, computing lacunarity solely from temporal or spatial dimensions makes it difficult to distinguish between target and reverberation patterns. Our method overcomes this by characterizing both the static and dynamic patterns of target echoes and background reverberation through spatial-temporal lacunarity. Additionally, we introduce an echo pyramid that enables multi-scale observation while reducing computational complexity. Experimental results demonstrate that our proposed method significantly improves the detection of small moving targets from high-level background reverberation. Furthermore, our method outperforms existing methods in terms of visual quality and quantitative metrics. Shaofeng Zou, Xuyang Wang 0003, Kaihui Zeng, Guolin Li |
ISCAS | 1 |
| 2024 | Robust Multi-Hypothesis Testing with Moment-Constrained Uncertainty SetsabstractThe problem of robust multi-hypothesis testing in the Bayesian setting is studied in this paper. Under the$m\geq 2$hypotheses, the data-generating distributions are assumed to belong to uncertainty sets constructed through some moment functions, i.e., the sets contain distributions whose moments are centered around empirical moments obtained from some training data sequences. The goal is to design a test that performs well under all distributions in the uncertainty sets, i.e., a test that minimizes the worst-case probability of error over the uncertainty sets. Insights on the need for optimization-based approaches to solve the robust testing problem with moment constrained uncertainty sets are provided. The optimal (robust) test based on the optimization approach is derived for the case where the observations belong to a finite-alphabet. When the size of the alphabet is infinite, the optimization problem is infinite-dimensional and intractable, and therefore a tractable finite-dimensional approximation is proposed, whose optimal value converges to the optimal value of the original problem as the size of the dimension of the approximation goes to infinity. A robust test is constructed from the solution to the approximate problem, and guarantees on its worst-case error probability over the uncertainty sets are provided. Numerical results are provided to demonstrate the performance of the proposed robust test. Akshayaa Magesh, Zhongchang Sun, Venugopal V. Veeravalli, Shaofeng Zou |
ISIT | 4 |
| 2024 | A Unified Principle of Pessimism for Offline Reinforcement Learning under Model MismatchabstractIn this paper, we address the challenges of offline reinforcement learning (RL) under model mismatch, where the agent aims to optimize its performance through an offline dataset that may not accurately represent the deployment environment. We identify two primary challenges under the setting: inaccurate model estimation due to limited data and performance degradation caused by the model mismatch between the dataset-collecting environment and the target deployment one. To tackle these issues, we propose a unified principle of pessimism using distributionally robust Markov decision processes. We carefully construct a robust MDP with a single uncertainty set to tackle both data sparsity and model mismatch, and demonstrate that the optimal robust policy enjoys a near-optimal sub-optimality gap under the target environment across three widely used uncertainty models: total variation, $\chi^2$ divergence, and KL divergence. Our results improve upon or match the state-of-the-art performance under the total variation and KL divergence models, and provide the first result for the $\chi^2$ divergence model. Yue Wang 0068, Zhongchang Sun, Shaofeng Zou |
NeurIPS | 3 |
| 2024 | Policy Optimization for Robust Average Reward MDPsabstractThis paper studies first-order policy optimization for robust average cost Markov decision processes (MDPs). Specifically, we focus on ergodic Markov chains. For robust average cost MDPs, the goal is to optimize the worst-case average cost over an uncertainty set of transition kernels. We first develop a sub-gradient of the robust average cost. Based on the sub-gradient, a robust policy mirror descent approach is further proposed. To characterize its iteration complexity, we develop a lower bound on the difference of robust average cost between two policies and further show that the robust average cost satisfies the PL-condition. We then show that with increasing step size, our robust policy mirror descent achieves a linear convergence rate in the optimality gap, and with constant step size, our algorithm converges to an $\epsilon$-optimal policy with an iteration complexity of $\mathcal{O}(1/\epsilon)$. The convergence rate of our algorithm matches with the best convergence rate of policy-based algorithms for robust MDPs. Moreover, our algorithm is the first algorithm that converges to the global optimum with general uncertainty sets for robust average cost MDPs. We provide simulation results to demonstrate the performance of our algorithm. Zhongchang Sun, Sihong He, Fei Miao, Shaofeng Zou |
NeurIPS | 4 |
| 2024 | Model-Free Robust Reinforcement Learning with Sample Complexity AnalysisabstractDistributionally Robust Reinforcement Learning (DR-RL) aims to derive a policy optimizing the worst-case performance within a predefined uncertainty set. Despite extensive research, previous DR-RL algorithms have predominantly favored model-based approaches, with limited availability of model-free methods offering convergence guarantees or sample complexities. This paper proposes a model-free DR-RL algorithm leveraging the Multi-level Monte Carlo (MLMC) technique to close such a gap. Our innovative approach integrates a threshold mechanism that ensures finite sample requirements for algorithmic implementation, a significant departure from previous model-free algorithms. We adapt our algorithm to accommodate uncertainty sets defined by total variation, Chi-square divergence, and KL divergence, and provide finite sample analyses under all three cases. Remarkably, our algorithms represent the first model-free DR-RL approach featuring finite sample complexity for total variation and Chi-square divergence uncertainty sets, while also offering an improved sample complexity and broader applicability compared to existing model-free DR-RL algorithms for the KL divergence model. The complexities of our method establish the tightest results for all three uncertainty models in model-free DR-RL, underscoring the effectiveness and efficiency of our algorithm, and highlighting its potential for practical applications. Yudan Wang, Shaofeng Zou, Yue Wang 0068 |
UAI | 2 |
| 2024 | Robust Average-Reward Reinforcement LearningabstractRobust Markov decision processes (MDPs) aim to find a policy that optimizes the worst-case performance over an uncertainty set of MDPs. Existing studies mostly have focused on the robust MDPs under the discounted reward criterion, leaving the ones under the average-reward criterion largely unexplored. In this paper, we develop the first comprehensive and systematic study of robust average-reward MDPs, where the goal is to optimize the long-term average performance under the worst case. Our contributions are four-folds: (1) we prove the uniform convergence of the robust discounted value function to the robust average-reward function as the discount factor γ goes to 1; (2) we derive the robust average-reward Bellman equation, characterize the structure of its solution set, and prove the equivalence between solving the robust Bellman equation and finding the optimal robust policy; (3) we design robust dynamic programming algorithms, and theoretically characterize their convergence to the optimal policy; and (4) we design two model-free algorithms unitizing the multi-level Monte-Carlo approach, and prove their asymptotic convergence Yue Wang 0068, Alvaro Velasquez, George Atia, Ashley Prater-Bennette, Shaofeng Zou |
J. Artif. Intell. Res. | 5 |
| 2024 | Finite-time error bounds for Greedy-GQ
Yue Wang 0068, Yi Zhou 0017, Shaofeng Zou |
Mach. Learn. | 3 |
| 2024 | Quickest Change Detection in Autoregressive ModelsabstractThe problem of quickest change detection (QCD) in autoregressive (AR) models is investigated. A system is being monitored with sequentially observed samples. At some unknown time, a disturbance signal occurs and changes the distribution of the observations. The disturbance signal follows an AR model, which is dependent over time. Before the change, observations only consist of measurement noise, and are independent and identically distributed (i.i.d.). After the change, observations consist of the disturbance signal and the measurement noise, are dependent over time, which essentially follow a continuous-state hidden Markov model (HMM). The goal is to design a stopping time to detect the disturbance signal as quickly as possible subject to false alarm constraints. Existing approaches for general non-i. i.d. settings and discrete-state HMMs cannot be applied due to their high computational complexity and memory consumption, and they usually assume some asymptotic stability condition. In this paper, the asymptotic stability condition is firstly theoretically proved for the AR model by a novel design of forward variable and auxiliary Markov chain. A computationally efficient Ergodic CuSum algorithm that can be updatedrecursivelyis then constructed and is further shown to be asymptotically optimal. The data-driven setting where the disturbance signal parameters are unknown is further investigated, and an online and computationally efficient gradient ascent CuSum algorithm is designed. The algorithm is constructed by iteratively updating the estimate of the unknown parameters based on the maximum likelihood principle and the gradient ascent approach. The lower bound on its average running length to false alarm is also derived for practical false alarm control. Simulation results are provided to demonstrate the performance of the proposed algorithms. Zhongchang Sun, Shaofeng Zou |
IEEE Trans. Inf. Theory | 2 |
| 2024 | Provably Efficient Offline Reinforcement Learning With Trajectory-Wise RewardabstractThe remarkable success of reinforcement learning (RL) heavily relies on observing the reward of every visited state-action pair. In many real world applications, however, an agent can observe only a score that represents the quality of the whole trajectory, which is referred to as the trajectory-wise reward. In such a situation, it is difficult for standard RL methods to well utilize trajectory-wise reward, and large bias and variance errors can be incurred in policy evaluation. In this work, we propose a novel offline RL algorithm, called Pessimistic vAlue iteRaTion with rEward Decomposition (PARTED), which decomposes the trajectory return into per-step proxy rewards via least-squares-based reward redistribution, and then performs pessimistic value iteration based on the learned proxy reward. To ensure the value functions constructed by PARTED are always pessimistic with respect to the optimal ones, we design a new penalty term to offset the uncertainty of the proxy reward. We first show that our PARTED achieves an$\tilde {\mathcal {O}}(dH^{3}/\sqrt {N})$suboptimality for linear MDPs, where d is the dimension of the feature, H is the episode length, and N is the size of the offline dataset. We further extend our algorithm and results to general large-scale episodic MDPs with neural network function approximation. To the best of our knowledge, PARTED is the first offline RL algorithm that is provably efficient in general MDP with trajectory-wise reward. Tengyu Xu, Yue Wang 0068, Shaofeng Zou, Yingbin Liang |
IEEE Trans. Inf. Theory | 3 |
| 2023 | Robust Average-Reward Markov Decision ProcessesabstractIn robust Markov decision processes (MDPs), the uncertainty in the transition kernel is addressed by finding a policy that optimizes the worst-case performance over an uncertainty set of MDPs. While much of the literature has focused on discounted MDPs, robust average-reward MDPs remain largely unexplored. In this paper, we focus on robust average-reward MDPs, where the goal is to find a policy that optimizes the worst-case average reward over an uncertainty set. We first take an approach that approximates average-reward MDPs using discounted MDPs. We prove that the robust discounted value function converges to the robust average-reward as the discount factor goes to 1, and moreover when it is large, any optimal policy of the robust discounted MDP is also an optimal policy of the robust average-reward. We further design a robust dynamic programming approach, and theoretically characterize its convergence to the optimum. Then, we investigate robust average-reward MDPs directly without using discounted MDPs as an intermediate step. We derive the robust Bellman equation for robust average-reward MDPs, prove that the optimal policy can be derived from its solution, and further design a robust relative value iteration algorithm that provably finds its solution, or equivalently, the optimal robust policy. Yue Wang 0068, Alvaro Velasquez, George Atia, Ashley Prater-Bennette, Shaofeng Zou |
AAAI | 5 |
| 2023 | Robust Hypothesis Testing With Moment Constrained Uncertainty SetsabstractThe problem of robust binary hypothesis testing is studied. Under both hypotheses, the data-generating distributions are assumed to belong to uncertainty sets constructed through moments; in particular, the sets contain distributions whose moments are centered around the empirical moments obtained from training observations. The goal is to design a test that performs well under all distributions in the uncertainty sets, i.e., minimize the worst-case error probability over the uncertainty sets. In the finite-alphabet case, the optimal test is obtained. In the infinite-alphabet case, a tractable approximation to the worst-case error is derived that converges to the optimal value A test is further constructed to generalize to the entire alphabet. An exponentially consistent test for testing batch samples is also proposed. Numerical results are provided to demonstrate the performance of the proposed robust tests. Akshayaa Magesh, Zhongchang Sun, Venugopal V. Veeravalli, Shaofeng Zou |
ICASSP | 4 |
| 2023 | Data-Driven Quickest Change Detection in Markov ModelsabstractThe problem of quickest change detection in Markov models is studied. A sequence of samples are generated from a Markov model, and at some unknown time, the transition kernel of the Markov model changes. The goal is to detect the change as soon as possible subject to false alarm constraints. The data-driven setting is investigated, where neither the pre-nor the post-change Markov transition kernel is known. A kernel based data-driven algorithm is developed, which applies to general state space and is recursive and computationally efficient. Performance bounds on the average running length and worst-case average detection delay are derived. Numerical results are provided to validate the performance of the proposed algorithm. Qi Zhang 0069, Zhongchang Sun, Luis C. Herrera, Shaofeng Zou |
ICASSP | 4 |
| 2023 | Model-Free Robust Average-Reward Reinforcement LearningabstractRobust Markov decision processes (MDPs) address the challenge of model uncertainty by optimizing the worst-case performance over an uncertainty set of MDPs. In this paper, we focus on the robust average-reward MDPs under the model-free setting. We first theoretically characterize the structure of solutions to the robust average-reward Bellman equation, which is essential for our later convergence analysis. We then design two model-free algorithms, robust relative value iteration (RVI) TD and robust RVI Q-learning, and theoretically prove their convergence to the optimal solution. We provide several widely used uncertainty sets as examples, including those defined by the contamination model, total variation, Chi-squared divergence, Kullback-Leibler (KL) divergence, and Wasserstein distance. Yue Wang 0068, Alvaro Velasquez, George Atia, Ashley Prater-Bennette, Shaofeng Zou |
ICML | 5 |
| 2023 | A Robust and Constrained Multi-Agent Reinforcement Learning Electric Vehicle Rebalancing Method in AMoD SystemsabstractElectric vehicles (EVs) play critical roles in autonomous mobility-on-demand (AMoD) systems, but their unique charging patterns increase the model uncertainties in AMoD systems (e.g. state transition probability). Since there usually exists a mismatch between the training and test/true environments, incorporating model uncertainty into system design is of critical importance in real-world applications. However, model uncertainties have not been considered explicitly in EV AMoD system rebalancing by existing literature yet, and the coexistence of model uncertainties and constraints that the decision should satisfy makes the problem even more challenging. In this work, we design a robust and constrained multi-agent reinforcement learning (MARL) framework with state transition kernel uncertainty for EV AMoD systems. We then propose a robust and constrained MARL algorithm (ROCOMA) with robust natural policy gradients (RNPG) that trains a robust EV rebalancing policy to balance the supply-demand ratio and the charging utilization rate across the city under model uncertainty. Experiments show that the ROCOMA can learn an effective and robust rebalancing policy. It outperforms non-robust MARL methods in the presence of model uncertainties. It increases the system fairness by 19.6% and decreases the rebalancing costs by 75.8%. Sihong He, Yue Wang 0068, Shuo Han 0002, Shaofeng Zou, Fei Miao |
IROS | 4 |
| 2023 | Data-Driven Quickest Change Detection in Hidden Markov ModelsabstractThe problem of quickest change detection in hidden Markov models (HMMs) is investigated. A sequence of samples are generated from a HMM, and at some unknown time, the transition kernel and/or the emission probability of the HMM changes. The goal is to detect the change as soon as possible subject to false alarm constraints. The data-driven setting is investigated, where none of the pre-, post-change Markov transition kernels or the emission probabilities are known. In this paper, a kernel based data-driven algorithm is developed. Performance bounds on its average running length (ARL) to false alarm and worst-case average detection delay (WADD) are theoretically characterized, where the WADD is at most of the order of the logarithm of the ARL. Numerical results are provided to validate the performance of the proposed algorithm. Qi Zhang 0069, Zhongchang Sun, Luis C. Herrera, Shaofeng Zou |
ISIT | 4 |
| 2023 | Decentralized Robust V-learning for Solving Markov Games with Model UncertaintyabstractThe Markov game is a popular reinforcement learning framework for modeling competitive players in a dynamic environment. However, most of the existing works on Markov games focus on computing a certain equilibrium following uncertain interactions among the players but ignore the uncertainty of the environment model, which is ubiquitous in practical scenarios. In this work, we develop a theoretical solution to Markov games with environment model uncertainty. Specifically, we propose a new and tractable notion of robust correlated equilibria for Markov games with environment model uncertainty. In particular, we prove that the robust correlated equilibrium has a simple modification structure, and its characterization of equilibria critically depends on the environment model uncertainty. Moreover, we propose the first fully-decentralized stochastic algorithm for computing such the robust correlated equilibrium. Our analysis proves that the algorithm achieves the polynomial episode complexity $\widetilde{O}( SA^2 H^5 \epsilon^{-2})$ for computing an approximate robust correlated equilibrium with $\epsilon$ accuracy. Shaocong Ma, Ziyi Chen 0002, Shaofeng Zou, Yi Zhou 0017 |
J. Mach. Learn. Res. | 3 |
| 2023 | Kernel Robust Hypothesis TestingabstractThe problem of robust hypothesis testing is studied, where under the null and the alternative hypotheses, the data-generating distributions are assumed to be in some uncertainty sets, and the goal is to design a test that performs well under the worst-case distributions over the uncertainty sets. In this paper, uncertainty sets are constructed in a data-driven manner using kernel method, i.e., they are centered around empirical distributions of training samples from the null and alternative hypotheses, respectively; and are constrained via the distance between kernel mean embeddings of distributions in the reproducing kernel Hilbert space, i.e., maximum mean discrepancy (MMD). The Bayesian setting and the Neyman-Pearson setting are investigated. For the Bayesian setting where the goal is to minimize the worst-case error probability, an optimal test is firstly obtained when the alphabet is finite. When the alphabet is infinite, a tractable approximation is proposed to quantify the worst-case average error probability, and a kernel smoothing method is further applied to design test that generalizes to unseen samples. A direct robust kernel test is also proposed and proved to be exponentially consistent. For the Neyman-Pearson setting, where the goal is to minimize the worst-case probability of miss detection subject to a constraint on the worst-case probability of false alarm, an efficient robust kernel test is proposed and is shown to be asymptotically optimal. Numerical results are provided to demonstrate the performance of the proposed robust tests. Zhongchang Sun, Shaofeng Zou |
IEEE Trans. Inf. Theory | 2 |
| 2022 | Sample and Communication-Efficient Decentralized Actor-Critic Algorithms with Finite-Time AnalysisabstractActor-critic (AC) algorithms have been widely used in decentralized multi-agent systems to learn the optimal joint control policy. However, existing decentralized AC algorithms either need to share agents’ sensitive information or lack communication-efficiency. In this work, we develop decentralized AC and natural AC (NAC) algorithms that avoid sharing agents’ local information and are sample and communication-efficient. In both algorithms, agents share only noisy rewards and use mini-batch local policy gradient updates to ensure high sample and communication efficiency. Particularly for decentralized NAC, we develop a decentralized Markovian SGD algorithm with an adaptive mini-batch size to efficiently compute the natural policy gradient. Under Markovian sampling and linear function approximation, we prove that the proposed decentralized AC and NAC algorithms achieve the state-of-the-art sample complexities $\mathcal{O}(\epsilon^{-2}\ln\epsilon^{-1})$ and $\mathcal{O}(\epsilon^{-3}\ln\epsilon^{-1})$, respectively, and achieve an improved communication complexity $\mathcal{O}(\epsilon^{-1}\ln\epsilon^{-1})$. Numerical experiments demonstrate that the proposed algorithms achieve lower sample and communication complexities than the existing decentralized AC algorithms. Ziyi Chen 0002, Yi Zhou 0017, Rong-Rong Chen, Shaofeng Zou |
ICML | 4 |
| 2022 | Policy Gradient Method For Robust Reinforcement LearningabstractThis paper develops the first policy gradient method with global optimality guarantee and complexity analysis for robust reinforcement learning under model mismatch. Robust reinforcement learning is to learn a policy robust to model mismatch between simulator and real environment. We first develop the robust policy (sub-)gradient, which is applicable for any differentiable parametric policy class. We show that the proposed robust policy gradient method converges to the global optimum asymptotically under direct policy parameterization. We further develop a smoothed robust policy gradient method, and show that to achieve an $\epsilon$-global optimum, the complexity is $\mathcal O(\epsilon^{-3})$. We then extend our methodology to the general model-free setting, and design the robust actor-critic method with differentiable parametric policy class and value function. We further characterize its asymptotic convergence and sample complexity under the tabular setting. Finally, we provide simulation results to demonstrate the robustness of our methods. Yue Wang 0068, Shaofeng Zou |
ICML | 2 |
| 2022 | Robust Hypothesis Testing with Kernel Uncertainty SetsabstractIn this paper, the robust hypothesis testing problem is investigated, where under the null and the alternative hypotheses, the distributions are assumed to be in some uncertainty sets. The uncertainty sets are constructed in a data-driven manner, i.e., they are centered around empirical distributions. The distance between kernel mean embeddings of distributions in the reproducing kernel Hilbert space is used as the distance metric of uncertainty sets. The Bayesian setting is studied, where the goal is to minimize the worst-case error probability. An optimal test is firstly obtained for the case with a finite alphabet. For the case with an infinite alphabet, a tractable approximation is proposed to quantify the worst-case error probability, and a kernel smoothing method is further applied to design test that generalizes to unseen samples. A heuristic robust kernel test is also proposed and proved to be exponentially consistent. Numerical results are provided to demonstrate the performance of the proposed tests. Zhongchang Sun, Shaofeng Zou |
ISIT | 2 |
| 2021 | Learning Graph Neural Networks with Approximate Gradient DescentabstractThe first provably efficient algorithm for learning graph neural networks (GNNs) with one hidden layer for node information convolution is provided in this paper. Two types of GNNs are investigated, depending on whether labels are attached to nodes or graphs. A comprehensive framework for designing and analyzing convergence of GNN training algorithms is developed. The algorithm proposed is applicable to a wide range of activation functions including ReLU, Leaky ReLU, Sigmod, Softplus and Swish. It is shown that the proposed algorithm guarantees a linear convergence rate to the underlying true parameters of GNNs. For both types of GNNs, sample complexity in terms of the number of nodes or the number of graphs is characterized. The impact of feature dimension and GNN structure on the convergence rate is also theoretically characterized. Numerical experiments are further provided to validate our theoretical analysis. Qunwei Li, Shaofeng Zou, Leon Wenliang Zhong |
AAAI | 2 |
| 2021 | Greedy-GQ with Variance Reduction: Finite-time Analysis and Improved Complexity
Shaocong Ma, Ziyi Chen 0002, Yi Zhou 0017, Shaofeng Zou |
ICLR | 4 |
| 2021 | An Object Enhancement Method for Forward-Looking Sonar Images Based on Multi-Frame FusionabstractForward-looking sonar (FLS) often suffers from complex underwater environments. It is hard to detect small objects from the FLS imagery characterized by low signal-to- noise ratio and low resolution. To highlight the object from severe noise background, we propose an object enhancement method for objects in the regions of interest (ROIs) based on multi-frame image fusion. This method includes two crucial steps: 1) A Fourier-based multi-stage registration algorithm is proposed to solve the problem of a drastic change of object position between frames due to long target distance and rapid change of azimuth angle. 2) A multi-frame fusion algorithm based on self-supervised deep learning is adopted to enhance the ROIs. Experimental results demonstrate that our proposed enhancement method can significantly highlight the objects in the ROIs and has excellent noise suppression in terms of quantitative metrics and visual quality. Shaofeng Zou, Xuyang Wang 0003, Guolin Li, Zhihua Wang 0001 |
ISCAS | 1 |
| 2021 | A Computationally Efficient Algorithm for Quickest Change Detection in Anonymous Heterogeneous Sensor NetworksabstractThe problem of quickest change detection in anonymous heterogeneous sensor networks is studied. The sensors are clustered into$K$groups, and different groups follow different data generating distributions. At some unknown time, an event occurs in the network and changes the data generating distribution of the sensors. The goal is to detect the change as quickly as possible, subject to false alarm constraints. The anonymous setting is studied, where at each time step, the fusion center receives unordered samples without knowing which sensor each sample comes from, and thus does not know its exact distribution. In [1], an optimal algorithm was provided, which however is not computational efficient for large networks. In this paper, a computationally efficient test is proposed and a novel theoretical characterization of its false alarm rate is further developed. Zhongchang Sun, Qunwei Li, Ruizhi Zhang 0001, Shaofeng Zou |
ISIT | 4 |
| 2021 | Quickest Dynamic Anomaly Detection in Anonymous Heterogeneous Sensor NetworksabstractThe problem of quickest dynamic anomaly detection in anonymous heterogeneous sensor networks is studied. The$n$heterogeneous sensors can be divided into$K$types with different data generating distributions. At some unknown time, an anomaly emerges in the network and changes the data generating distribution of the sensors. The goal is to detect the anomaly as quickly as possible, subject to false alarm constraints. The anonymous setting is studied, where the fusion center does not know which sensor that each sample comes from, and thus does not know its exact distribution. Firstly, the static setting is investigated where the sensor affected by the anomaly does not change with time. A generalized mixture CuSum algorithm is constructed and is further shown to be asymptotically optimal. The problem is then extended to a dynamic setting where the sensor affected by the anomaly changes with time. An asymptotically optimal weighted mixture CuSum algorithm is proposed. Numerical results are also provided to validate the theoretical results. Zhongchang Sun, Shaofeng Zou |
ISIT | 2 |
| 2021 | A Data-Driven Approach to Robust Hypothesis Testing Using Kernel MMD Uncertainty SetsabstractThe problem of robust hypothesis testing is studied, where under the null and alternative hypotheses, data generating distributions are assumed to belong to some uncertainty sets. In this paper, uncertainty sets are constructed in a data-driven manner, i.e., they are centered around empirical distributions of training samples from the null and alternative hypotheses, respectively; and are constrained via the distance between kernel mean embeddings of distributions in the reproducing kernel Hilbert space. The Neyman-Pearson setting is investigated, where the goal is to minimize the worst-case probability of miss detection subject to the constraint on the worst-case probability of false alarm. An efficient robust kernel test is proposed and is further shown to be asymptotically optimal. Numerical results are further provided to demonstrate the performance of the proposed robust test. Zhongchang Sun, Shaofeng Zou |
ISIT | 2 |
| 2021 | Online Robust Reinforcement Learning with Model UncertaintyabstractRobust reinforcement learning (RL) is to find a policy that optimizes the worst-case performance over an uncertainty set of MDPs. In this paper, we focus on model-free robust RL, where the uncertainty set is defined to be centering at a misspecified MDP that generates samples, and is assumed to be unknown. We develop a sample-based approach to estimate the unknown uncertainty set, and design robust Q-learning algorithm (tabular case) and robust TDC algorithm (function approximation setting), which can be implemented in an online and incremental fashion. For the robust Q-learning algorithm, we prove that it converges to the optimal robust Q function, and for the robust TDC algorithm, we prove that it converges asymptotically to some stationary points. Unlike the results in [Roy et al., 2017], our algorithms do not need any additional conditions on the discount factor to guarantee the convergence. We further characterize the finite-time error bounds of the two algorithms, and show that both the robust Q-learning and robust TDC algorithms converge as fast as their vanilla counterparts (within a constant factor). Our numerical experiments further demonstrate the robustness of our algorithms. Our approach can be readily extended to robustify many other algorithms, e.g., TD, SARSA, and other GTD algorithms. Yue Wang 0068, Shaofeng Zou |
NeurIPS | 2 |
| 2021 | Non-Asymptotic Analysis for Two Time-scale TDC with General Smooth Function ApproximationabstractTemporal-difference learning with gradient correction (TDC) is a two time-scale algorithm for policy evaluation in reinforcement learning. This algorithm was initially proposed with linear function approximation, and was later extended to the one with general smooth function approximation. The asymptotic convergence for the on-policy setting with general smooth function approximation was established in [Bhatnagar et al., 2009], however, the non-asymptotic convergence analysis remains unsolved due to challenges in the non-linear and two-time-scale update structure, non-convex objective function and the projection onto a time-varying tangent plane. In this paper, we develop novel techniques to address the above challenges and explicitly characterize the non-asymptotic error bound for the general off-policy setting with i.i.d. or Markovian samples, and show that it converges as fast as $\mathcal O(1/\sqrt T)$ (up to a factor of $\mathcal O(\log T)$). Our approach can be applied to a wide range of value-based reinforcement learning algorithms with general smooth function approximation. Yue Wang 0068, Shaofeng Zou, Yi Zhou 0017 |
NeurIPS | 2 |
| 2020 | Information-Theoretic Understanding of Population Risk Improvement with Model CompressionabstractWe show that model compression can improve the population risk of a pre-trained model, by studying the tradeoff between the decrease in the generalization error and the increase in the empirical risk with model compression. We first prove that model compression reduces an information-theoretic bound on the generalization error; this allows for an interpretation of model compression as a regularization technique to avoid overfitting. We then characterize the increase in empirical risk with model compression using rate distortion theory. These results imply that the population risk could be improved by model compression if the decrease in generalization error exceeds the increase in empirical risk. We show through a linear regression example that such a decrease in population risk due to model compression is indeed possible. Our theoretical results further suggest that the Hessian-weighted K-means clustering compression approach can be improved by regularizing the distance between the clustering centers. We provide experiments with neural networks to support our theoretical assertions. Yuheng Bu, Weihao Gao, Shaofeng Zou, Venugopal V. Veeravalli |
AAAI | 3 |
| 2020 | Quickest Change Detection In Anonymous Heterogeneous Sensor NetworksabstractThe problem of quickest change detection (QCD) in anonymous heterogeneous sensor networks is studied. There are n heterogeneous sensors and a fusion center. The sensors are clustered into K groups, and different groups follow different data generating distributions. At some unknown time, an event occurs in the network and changes the data generating distribution of the sensors. The goal is to detect the change as quickly as possible, subject to false alarm constraints. The anonymous setting is studied in this paper, where at each time step, the fusion center receives n unordered samples. The fusion center does not know which sensor each sample comes from, and thus does not know its exact distribution. In this paper, a simple optimality proof is derived for the Mixture Likelihood Ratio Test (MLRT), which was constructed and proved to be optimal for the non-sequential anonymous setting in [1]. For the QCD problem, a mixture CuSum algorithm is constructed in this paper, and is further shown to be optimal under Lorden's criterion [2]. Zhongchang Sun, Shaofeng Zou, Qunwei Li |
ICASSP | 2 |
| 2020 | A Game-Theoretic Approach to Sequential Detection in Adversarial EnvironmentsabstractThe problem of sequential binary hypothesis testing in an adversarial environment is investigated. Specifically, if there is no adversary, the samples are generated independently by a distribution p; and if the adversary is present, the samples are generated independently by another distribution q. The adversary picks a distribution q ∈ Q with cost c(q). The goal of the defender is to decide whether there is an adversary using samples as few as possible; and the goal of the adversary is to fool the defender. The problem is formulated as a non-zero-sum game between the adversary and the defender. A pair of strategies (attack strategy from the adversary and the sequential hypothesis testing scheme from the detector) is proposed and proved to be a Nash equilibrium pair for the non-zero-sum game asymptotically. Numerical experiments are provided to validate our results. Ruizhi Zhang 0001, Shaofeng Zou |
ISIT | 2 |
| 2020 | Variance-Reduced Off-Policy TDC Learning: Non-Asymptotic Convergence AnalysisabstractVariance reduction techniques have been successfully applied to temporal-difference (TD) learning and help to improve the sample complexity in policy evaluation. However, the existing work applied variance reduction to either the less popular one time-scale TD algorithm or the two time-scale GTD algorithm but with a finite number of i.i.d.\ samples, and both algorithms apply to only the on-policy setting. In this work, we develop a variance reduction scheme for the two time-scale TDC algorithm in the off-policy setting and analyze its non-asymptotic convergence rate over both i.i.d.\ and Markovian samples. In the i.i.d setting, our algorithm achieves an improved sample complexity $\calO(\epsilon^{-\frac{3}{5}} \log{\epsilon}^{-1})$ over the state-of-the-art result $\calO(\epsilon^{-1} \log {\epsilon}^{-1})$. In the Markovian setting, our algorithm achieves the state-of-the-art sample complexity $\calO(\epsilon^{-1} \log {\epsilon}^{-1})$ that is near-optimal. Experiments demonstrate that the proposed variance-reduced TDC achieves a smaller asymptotic convergence error than both the conventional TDC and the variance-reduced TD. Shaocong Ma, Yi Zhou 0017, Shaofeng Zou |
NeurIPS | 3 |
| 2020 | Finite-sample Analysis of Greedy-GQ with Linear Function Approximation under Markovian NoiseabstractGreedy-GQ is an off-policy two timescale algorithm for optimal control in reinforcement learning. This paper develops the first finite-sample analysis for the Greedy-GQ algorithm with linear function approximation under Markovian noise. Our finite-sample analysis provides theoretical justification for choosing stepsizes for this two timescale algorithm for faster convergence in practice, and suggests a trade-off between the convergence rate and the quality of the obtained policy. Our paper extends the finite-sample analyses of two timescale reinforcement learning algorithms from policy evaluation to optimal control, which is of more practical interest. Specifically, in contrast to existing finite-sample analyses for two timescale methods, e.g., GTD, GTD2 and TDC, where their objective functions are convex, the objective function of the Greedy-GQ algorithm is non-convex. Moreover, the Greedy-GQ algorithm is also not a linear two-timescale stochastic approximation algorithm. Our techniques in this paper provide a general framework for finite-sample analysis of non-convex value-based reinforcement learning algorithms for optimal control. Yue Wang 0068, Shaofeng Zou |
UAI | 2 |
| 2020 | Quickest Detection of Dynamic Events in NetworksabstractThe problem of quickest detection of dynamic events in networks is studied. At some unknown time, an event occurs, and a number of nodes in the network are affected by the event, in that they undergo a change in the statistics of their observations. It is assumed that the event is dynamic, in that it can propagate along the edges in the network, and affect more and more nodes with time. The event propagation dynamics is assumed to be unknown. The goal is to design a sequential algorithm that can detect a “significant” event, i.e., when the event has affected no fewer than η nodes, as quickly as possible, while controlling the false alarm rate. Fully connected networks are studied first, and the results are then extended to arbitrarily connected networks. The designed algorithms are shown to be adaptive to the unknown propagation dynamics, and their first-order asymptotic optimality is demonstrated as the false alarm rate goes to zero. The algorithms can be implemented with linear computational complexity in the network size at each time step, which is critical for online implementation. Numerical simulations are provided to validate the theoretical results. Shaofeng Zou, Venugopal V. Veeravalli, Jian Li 0008, Don Towsley |
IEEE Trans. Inf. Theory | 1 |
| 2019 | Distributed Quickest Detection of Significant Events in NetworksabstractThe problem of quickest detection of significant events in networks is studied. A distributed setting is investigated, where there is no fusion center, and each node only communicates with its neighbors. After an event occurs in the network, a number of nodes are affected, which changes the statistics of their observations. The nodes may possibly perceive the event at different times. The goal is to design a distributed sequential detection rule that can detect when the event is "significant", i.e., the event has affected no less than η nodes, as quickly as possible, subject to false alarm constraints. A distributed algorithm is proposed, which is based on a novel combination of the alternating direction method of multipliers (ADMM) and average consensus approaches. Numerical results are provided to demonstrate the performance of the proposed algorithm. Shaofeng Zou, Venugopal V. Veeravalli, Jian Li 0008, Don Towsley, Ananthram Swami |
ICASSP | 1 |
| 2019 | Tightening Mutual Information Based Bounds on Generalization ErrorabstractA mutual information based upper bound on the generalization error of a supervised learning algorithm is derived in this paper. The bound is constructed in terms of the mutual information between each individual training sample and the output of the learning algorithm, which requires weaker conditions on the loss function, but provides a tighter characterization of the generalization error than existing studies. Examples are further provided to demonstrate that the bound derived in this paper is tighter, and has a broader range of applicability. Application to noisy and iterative algorithms, e.g., stochastic gradient Langevin dynamics (SGLD), is also studied, where the constructed bound provides a tighter characterization of the generalization error than existing results. Yuheng Bu, Shaofeng Zou, Venugopal V. Veeravalli |
ISIT | 2 |
| 2019 | Quickest Detection of a Moving Target in a Sensor NetworkabstractTo be considered for the 2019 IEEE Jack Keil Wolf ISIT Student Paper Award. The problem of quickest detection of a moving target in sensor networks is studied. At some unknown time, a target emerges in the sensor network, and one of the sensors in the network is affected, whose data generating distribution undergoes a change. It is assumed that as the target moves around in the sensor network, the sensor that is affected by the target changes with time. Specifically, if a sensor becomes unaffected, then its data generating distribution changes back to the pre-change mode. A discrete time Markov chain is used to model the location of the affected sensor, and thus the data generating distribution of the sensor network after the target emerges is a hidden Markov model. The goal is to detect the existence of the target as quickly as possible subject to false alarm constraints. A windowed test based on a generalized likelihood ratio approach is constructed, and its asymptotic optimality is further established. Numerical results are provided to demonstrate its performance. Georgios Rovatsos, Shaofeng Zou, Venugopal V. Veeravalli |
ISIT | 2 |
| 2019 | Two Time-scale Off-Policy TD Learning: Non-asymptotic Analysis over Markovian SamplesabstractGradient-based temporal difference (GTD) algorithms are widely used in off-policy learning scenarios. Among them, the two time-scale TD with gradient correction (TDC) algorithm has been shown to have superior performance. In contrast to previous studies that characterized the non-asymptotic convergence rate of TDC only under identical and independently distributed (i.i.d.) data samples, we provide the first non-asymptotic convergence analysis for two time-scale TDC under a non-i.i.d.\ Markovian sample path and linear function approximation. We show that the two time-scale TDC can converge as fast as O(log t/t^(2/3)) under diminishing stepsize, and can converge exponentially fast under constant stepsize, but at the cost of a non-vanishing error. We further propose a TDC algorithm with blockwisely diminishing stepsize, and show that it asymptotically converges with an arbitrarily small error at a blockwisely linear convergence rate. Our experiments demonstrate that such an algorithm converges as fast as TDC under constant stepsize, and still enjoys comparable accuracy as TDC under diminishing stepsize. Tengyu Xu, Shaofeng Zou, Yingbin Liang |
NeurIPS | 2 |
| 2019 | Finite-Sample Analysis for SARSA with Linear Function ApproximationabstractSARSA is an on-policy algorithm to learn a Markov decision process policy in reinforcement learning. We investigate the SARSA algorithm with linear function approximation under the non-i.i.d.\ setting, where a single sample trajectory is available. With a Lipschitz continuous policy improvement operator that is smooth enough, SARSA has been shown to converge asymptotically. However, its non-asymptotic analysis is challenging and remains unsolved due to the non-i.i.d. samples, and the fact that the behavior policy changes dynamically with time. In this paper, we develop a novel technique to explicitly characterize the stochastic bias of a type of stochastic approximation procedures with time-varying Markov transition kernels. Our approach enables non-asymptotic convergence analyses of this type of stochastic approximation algorithms, which may be of independent interest. Using our bias characterization technique and a gradient descent type of analysis, we further provide the finite-sample analysis on the mean square error of the SARSA algorithm. In the end, we present a fitted SARSA algorithm, which includes the original SARSA algorithm and its variant as special cases. This fitted SARSA algorithm provides a framework for \textit{iterative} on-policy fitted policy iteration, which is more memory and computationally efficient. For this fitted SARSA algorithm, we also present its finite-sample analysis. Shaofeng Zou, Tengyu Xu, Yingbin Liang |
NeurIPS | 1 |
| 2019 | Quickest Change Detection Under Transient Dynamics: Theory and Asymptotic AnalysisabstractThe problem of quickest change detection under transient dynamics is studied, where the change from the initial distribution to the final persistent distribution does not happen instantaneously, but after a series of transient phases. The observations within the different phases are generated by different distributions. The objective is to detect the change as quickly as possible, while controlling the average run length (ARL) to false alarm, when the durations of the transient phases are completely unknown. Two algorithms are considered: the dynamic Cumulative Sum (CuSum) algorithm, proposed in earlier work, and a newly constructed weighted dynamic CuSum algorithm. Both algorithms admit recursions that facilitate their practical implementation, and they are adaptive to the unknown transient durations. Specifically, their asymptotic optimality is established with respect to both Lorden's and Pollak's criteria as the ARL to false alarm and the durations of the transient phases go to infinity at any relative rate. Numerical results are provided to demonstrate the adaptivity of the proposed algorithms and to validate the theoretical results. Shaofeng Zou, Georgios Fellouris, Venugopal V. Veeravalli |
IEEE Trans. Inf. Theory | 1 |
| 2018 | Quickest Detection of Dynamic Events in Sensor NetworksabstractWe consider the problem of quickest detection of dynamic events in sensor networks. After an event occurs, a number of sensors are affected and undergo a change in the statistics of their observations. We assume that the event is dynamic and can propagate with time, i.e., different sensors perceive the event at different times. The goal is to design a sequential algorithm that can detect when the event has affected no less than η sensors as quickly as possible, subject to false alarm constraints. We design a computationally efficient algorithm that is adaptive to unknown propagation dynamics, and demonstrate its asymptotic optimality as the false alarm rate goes to zero. We also provide numerical simulations to validate our theoretical results. Shaofeng Zou, Venugopal V. Veeravalli |
ICASSP | 1 |
| 2018 | Estimation of KL Divergence: Optimal Minimax RateabstractThe problem of estimating the Kullback-Leibler divergence D(P∥Q) between two unknown distributions P and Q is studied, under the assumption that the alphabet size k of the distributions can scale to infinity. The estimation is based on m independent samples drawn from P and n independent samples drawn from Q. It is first shown that there does not exist any consistent estimator that guarantees asymptotically small worst case quadratic risk over the set of all pairs of distributions. A restricted set that contains pairs of distributions, with density ratio bounded by a function f (k) is further considered. An augmented plug-in estimator is proposed, and its worst case quadratic risk is shown to be within a constant factor of ((k/m) + (kf (k)/n))2+ (log2f (k)/m) + ( f (k)/n), if m and n exceed a constant factor of k and kf (k), respectively. Moreover, the minimax quadratic risk is characterized to be within a constant factor of ((k/(m log k)) + (kf (k)/(n log k)))2+ (log2f (k)/m) + ( f (k)/n), if m and n exceed a constant factor of k/ log(k) and kf (k)/ log k, respectively. The lower bound on the minimax quadratic risk is characterized by employing a generalized Le Cam's method. A minimax optimal estimator is then constructed by employing both the polynomial approximation and the plug-in approaches. Yuheng Bu, Shaofeng Zou, Yingbin Liang, Venugopal V. Veeravalli |
IEEE Trans. Inf. Theory | 2 |
| 2018 | Degraded Broadcast Channel With Secrecy Outside a Bounded RangeabstractThe K-receiver degraded broadcast channel with secrecy outside a bounded range is studied, in which a transmitter sends K messages to K receivers, and the channel quality gradually degrades from receiver K to receiver 1. Each receiver k is required to decode message W1, ..., Wk, for 1 ≤ k ≤ K, and to be kept ignorant of Wk+2, .. ., WK, fork = 1, ..., K -2. Thus, each message Wkis kept secure from receivers with at least two-level worse channel quality, i.e., receivers 1, ..., k-2. The secrecy capacity region is fully characterized. The achievable scheme designates one superposition layer to each message with binning employed for each layer. Joint embedded coding and binning are employed to protect all upper-layer messages from lower-layer receivers. Furthermore, the scheme allows adjacent layers to share rates so that part of the rate of each message can be shared with its immediate upper-layer message to enlarge the rate region. More importantly, an induction approach is developed to perform Fourier-Motzkin elimination of 2Kvariables from the order of K2bounds to obtain a close-form achievable rate region. An outer bound is developed that matches the achievable rate region, whose proof involves recursive construction of the rate bounds and exploits the intuition gained from the achievable scheme. Shaofeng Zou, Yingbin Liang, Lifeng Lai, H. Vincent Poor, Shlomo Shamai |
IEEE Trans. Inf. Theory | 1 |
| 2017 | Quickest change detection under transient dynamicsabstractThe problem of transient quickest change detection (QCD) is studied, in which the change from the initial to the final phase does not happen instantaneously, but after a series of cascading transient phases of finite durations, each one corresponding to a different probability distribution. The goal is to design a stopping rule to detect the change as quickly as possible, subject to false alarm constraints. In previous work, the D-CuSum algorithm was proposed for such a QCD problem. The D-CuSum does not incorporate any prior statistical information about the durations of the transient periods. In this work, we develop an algorithm, the D-S-R algorithm, which incorporates geometric priors on the durations of the transient periods. We compare the D-CuSum and D-S-R algorithms in numerical examples to develop some insights about the role of the prior on the transient durations on the performance. Georgios Rovatsos, Shaofeng Zou, Venugopal V. Veeravalli |
ICASSP | 2 |
| 2017 | Linear-complexity exponentially-consistent tests for universal outlying sequence detectionabstractWe study a universal outlying sequence detection problem, in which there are M sequences of samples out of which a small subset of outliers need to be detected. A sequence is considered as an outlier if the observations therein are generated by a distribution different from those generating the observations in the majority of the sequences. In the universal setting, the goal is to identify all the outliers without any knowledge about the underlying generating distributions. In prior work, this problem was studied as a universal hypothesis testing problem, and a generalized likelihood (GL) test was constructed and its asymptotic performance characterized. In this paper, we propose a different class of tests for this problem based on distribution clustering. Such tests are shown to be exponentially consistent and their time complexity is linear in the total number of sequences, in contrast with the GL test, which has time complexity that is exponential in the number of outliers. Furthermore, our tests based on clustering are applicable to more general scenarios. For example, when both the typical and outlier distributions form clusters, the clustering based test is exponentially consistent, but the GL test is not even applicable. Yuheng Bu, Shaofeng Zou, Venugopal V. Veeravalli |
ISIT | 2 |
| 2017 | Asymptotic optimality of D-CuSum for quickest change detection under transient dynamicsabstractThe problem of quickest change detection (QCD) under transient dynamics is studied, in which the change from the initial distribution to the final persistent distribution does not happen instantaneously, but after a series of cascading transient phases. It is assumed that the durations of the transient phases are deterministic but unknown. The goal is to detect the change as quickly as possible subject to a constraint on the average run length to false alarm. The dynamic CuSum (D-CuSum) algorithm is investigated, which is based on reformulating the QCD problem into a dynamic composite hypothesis testing problem, and has a recursion that facilitates implementation. We show that this algorithm is adaptive to the unknown change point, as well as the unknown transient duration. And under mild conditions of the pre-change and post-change distributions, its asymptotic optimality is demonstrated for all possible asymptotic regimes as the transient duration and the average run length to false alarm go to infinity. Shaofeng Zou, Georgios Fellouris, Venugopal V. Veeravalli |
ISIT | 1 |
| 2016 | Universal outlying sequence detection for continuous observationsabstractThe following detection problem is studied, in which there are M sequences of samples out of which one outlier sequence needs to be detected. Each typical sequence contains n independent and identically distributed (i.i.d.) continuous observations from a known distribution π, and the outlier sequence contains n i.i.d. observations from an outlier distribution μ, which is distinct from n, but otherwise unknown. A universal test based on Kullback-Leibler (KL) divergence is built to approximate the maximum likelihood test, with known π and unknown μ. A KL divergence estimator based on data-dependent partitions is employed, and is shown to converge to its true value exponentially fast when the density ratio satisfies 0 <; Kl ≤ dμ/dπ ≤ K2, where K1 and K2 are positive constants. The performance of such a KL divergence estimator further implies that the outlier detection test is exponentially consistent. The detection performance of the KL divergence based test is compared with that of a recently introduced test for this problem based on the machine learning approach of maximum mean discrepancy (MMD). Regimes in which the KL divergence based test is better than the MMD based test are identified. Yuheng Bu, Shaofeng Zou, Yingbin Liang, Venugopal V. Veeravalli |
ICASSP | 2 |
| 2016 | Nonparametric detection of an anomalous disk over a two-dimensional lattice networkabstractNonparametric detection of existence of an anomalous disk over a lattice network is investigated. If an anomalous disk exists, then all nodes belonging to the disk observe samples generated by a distribution q, whereas all other nodes observe samples generated by a distribution p that is distinct from q. If there does not exist an anomalous disk, then all nodes receive samples generated by p. The distributions p and q are arbitrary and unknown. The goal is to design statistically consistent test as the network size becomes asymptotically large. A kernel-based test is proposed based on maximum mean discrepancy (MMD) which measures the distance between mean embeddings of distributions into a reproducing kernel Hilbert space (RKHS). A sufficient condition on the minimum size of candidate anomalous disks is characterized in order to guarantee the consistency of the proposed test. A necessary condition that any universally consistent test must satisfy is further derived. Comparison of sufficient and necessary conditions yields that the proposed test is order-level optimal. Shaofeng Zou, Yingbin Liang, H. Vincent Poor |
ICASSP | 1 |
| 2016 | Estimation of KL divergence between large-alphabet distributionsabstractThe problem of estimating the KL divergence between two unknown distributions is studied. The alphabet size k of the distributions can scale to infinity. The estimation is based on m and n independent samples respectively drawn from the two distributions. It is first shown that there does not exist any consistent estimator to guarantee asymptotic small worst-case quadratic risk over the set of all pairs of distributions. A restricted set that contains pairs of distributions with bounded ratio f(k) is further considered. An augmented plug-in estimator is proposed, and is shown to be consistent if and only if m = ω(k ⋁ log2(f(k)) and n = ω(k f(k)). Furthermore, if f(k) ≥ log2k and log2(f(k)) = o(k), it is shown that any consistent estimator must satisfy the necessary conditions: m = ω( k/log k ⋁ log2(f(k)) and n = ω( k f(k)/log k). Yuheng Bu, Shaofeng Zou, Yingbin Liang, Venugopal V. Veeravalli |
ISIT | 2 |
| 2016 | K-user degraded broadcast channel with secrecy outside a bounded rangeabstractA K-receiver degraded broadcast channel with secrecy outside a bounded range is studied, in which a transmitter sends K messages respectively to K receivers, and the channel quality gradually degrades from receiver K to receiver 1. Each receiver k is required to decode messages W1, …, Wk, for 1 ≤ k ≤ K. Furthermore, each message Wkshould be kept secure from receivers with two-level worse channel quality, i.e., receivers 1, …, k − 2. The secrecy capacity region is fully characterized. The achievable scheme designates one superposition layer to each message with random binning employed for each layer for protecting all upper-layer messages from lower-layer receivers. Furthermore, the scheme allows adjacent layers to share rates so that part of the rate of each message can potentially be shared with its immediate upper-layer message to enlarge the rate region. More importantly, an induction approach is developed to perform Fourier-Motzkin elimination over 2K variables among Θ(K2) bounds to obtain a close-form achievable rate region. A converse proof is developed that matches the achievable rate region, which involves recursive construction of the rate bounds. Shaofeng Zou, Yingbin Liang, Lifeng Lai, H. Vincent Poor, Shlomo Shamai |
ITW | 1 |
| 2015 | Rate splitting and sharing for degraded broadcast channel with secrecy outside a bounded rangeabstractA four-receiver degraded broadcast channel with secrecy outside a bounded range is studied, over which a transmitter sends four messages to four receivers. In the model considered, the channel quality gradually degrades from receiver 4 to receiver 1, and receiver k is required to decode the first k messages for k = 1, …, 4. Furthermore, message 3 is required to be secured from receiver 1, and message 4 is required to be secured from receivers 1 and 2. The secrecy capacity region is established. The achievable scheme includes not only superposition, binning and embedded coding used in previous studies, but also rate splitting and sharing particularly designed for this model, which is shown to be critical to further enlarge the achievable region and enable the development of the converse proof. Shaofeng Zou, Yingbin Liang, Lifeng Lai, Shlomo Shamai |
ISIT | 1 |
| 2015 | Degraded broadcast channel: Secrecy outside of a bounded rangeabstractA three-receiver degraded broadcast channel with secrecy outside of a bounded range is studied, in which the channel quality gradually degrades from receiver 3 to receiver 1. The transmitter has three messages intended for the receivers with receiver 3 decoding all messages, receiver 2 decoding the first two messages, and receiver 1 decoding only the first message. Furthermore, the third message should be kept secure from receiver 1. The discrete memoryless channel is studied and the secrecy capacity region is characterized. The achievable scheme is based on superposition coding and random binning, in which one superposition layer and random binning together provide secrecy. The converse proof is derived based on the insight obtained from the achievable scheme so that manipulations of terms yield tight rate bounds. Shaofeng Zou, Yingbin Liang, Lifeng Lai, Shlomo Shamai |
ITW | 1 |
| 2015 | Broadcast Networks With Layered Decoding and Layered Secrecy: Theory and ApplicationsabstractRecent information-theoretic results on a class of broadcast channels with layered decoding and/or layered secrecy are reviewed. In this class of models, a transmitter sends multiple messages to a set of legitimate receivers in the presence of a set of eavesdroppers, whose channels can be ordered based on the quality of received signals. Receivers with better channel quality are required to decode more messages, and eavesdroppers with worse channel quality are required to be kept ignorant of more messages. The design of achievable schemes and the characterization of the corresponding secrecy capacity regions are presented. Comparison of the designs for different models is discussed. Applications of these information-theoretic models to the study of secure communication over fading wiretap channels and secret sharing are also presented to illustrate potential applications of these models. Shaofeng Zou, Yingbin Liang, Lifeng Lai, H. Vincent Poor, Shlomo Shamai |
Proc. IEEE | 1 |
| 2015 | An Information Theoretic Approach to Secret SharingabstractA novel information theoretic approach is proposed to solve the secret sharing problem, in which a dealer distributes one or multiple secrets among a set of participants in such a manner that for each secret only qualified sets of users can recover this secret by pooling their shares together while nonqualified sets of users obtain no information about the secret even if they pool their shares together. While existing secret sharing systems (implicitly) assume that communications between the dealer and participants are noiseless, this paper takes a more practical assumption that the dealer delivers shares to the participants via a noisy broadcast channel. Thus, in contrast to the existing solutions that are mainly based on number theoretic tools, an information theoretic approach is proposed, which exploits the channel randomness during delivery of shares as additional resources to achieve secret sharing requirements. In this way, secret sharing problems can be reformulated as equivalent secure communication problems via wiretap channel models, and can hence be solved by employing the powerful information theoretic security techniques. This approach is first developed for the classic secret sharing problem, in which only one secret is to be shared. This classic problem is shown to be equivalent to a communication problem over a compound wiretap channel. Thus, the lower and upper bounds on the secrecy capacity of the compound channel provide the corresponding bounds on the secret sharing rate, and the secrecy scheme designed for the compound channel provides the secret sharing schemes. The power of the approach is further demonstrated by a more general layered multisecret sharing problem, which is shown to be equivalent to the degraded broadcast multiple-input multiple-output (MIMO) channel with layered decoding and secrecy constraints. The secrecy capacity region for the degraded MIMO broadcast channel is characterized, which provides the secret sharing capacity region. Furthermore, the secure encoding scheme that achieves the secrecy capacity region provides an information theoretic scheme for sharing the secrets. Shaofeng Zou, Yingbin Liang, Lifeng Lai, Shlomo Shamai |
IEEE Trans. Inf. Theory | 1 |
| 2014 | Layered secure broadcasting over MIMO channels and application in secret sharingabstractIn this paper, the degraded Gaussian Multiple-Input-Multiple-Output (MIMO) broadcast channel with layered decoding and secrecy constraints is investigated. In this model, there are in total K messages and K receivers that are ordered by the channel quality. Each receiver is required to decode one more message than the receiver with one level worse channel quality. Furthermore, this message should be kept secure from the receivers with worse channel qualities. The secrecy capacity region for this model is fully characterized. The converse proof relies on a novel construction of a series of covariance matrices. An application of this model to the problem of sharing multiple secrets, which is difficult to solve using number theoretic tools, is investigated. The secret sharing capacity region is characterized by reformulating the secret sharing problem as the secure communication problem over the K-receiver degraded Gaussian MIMO broadcast channel. Shaofeng Zou, Yingbin Liang, Lifeng Lai, Shlomo Shamai |
ISIT | 1 |
| 2013 | Multiple access channel with state uncertainty at transmittersabstractTwo-user fading multiple access channel (MAC) is investigated, which is corrupted by random fading coefficients and additive Gaussian noise. It is assumed that the channel is block fading, and each transmitter knows only its own channel state to the receiver, but does not know the other transmitter's channel state. The receiver has full knowledge of channel state information (CSI). The performance measure, the expected capacity region over channel statistics, is studied for two scenarios. For the first scenario, in which user 1 has multiple states, and user 2 has one state, most part of the boundary of the expected capacity region is characterized. Interestingly, these rate points are also on the boundary of the capacity region (i.e., the best achievable rate pairs) when the CSI is fully known at both transmitters. Furthermore the expected capacity region is fully characterized for some asymptotic regimes. For the second scenario, in which both users 1 and 2 have two states, a number of achievable regions are studied, and are demonstrated to be close to an outer bound numerically. Shaofeng Zou, Yingbin Liang, Shlomo Shamai |
ISIT | 1 |