VLDB 2026 Research / reviewers in the wild / expert
Zhihua Zhang 0004
dblp:52/5331-4
· DBLP profile ↗
64ranked-venue papers
16as first author
21since 2021 · last 2025
0000-0003-3165-5213ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 61 · 16 first-author · 21 since 2021Graphics, computer vision, multimedia, augmented reality and games · 8 · 2 first-authorDatabases, data management, data science and information retrieval · 7 · 1 first-author
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2025 | On the Convergence of Projected Policy Gradient for Any Constant Step SizesabstractProjected policy gradient (PPG) is a basic policy optimization method in reinforcement learning. Given access to exact policy evaluations, previous studies have established the sublinear convergence of PPG for sufficiently small step sizes based on the smoothness and the gradient domination properties of the value function. However, as the step size goes to infinity, PPG reduces to the classic policy iteration method, which suggests the convergence of PPG even for large step sizes. In this paper, we fill this gap and show that PPG admits a sublinear convergence for any constant step sizes. Due to the existence of the state-wise visitation measure in the expression of policy gradient, the existing optimization-based analysis framework for a preconditioned version of PPG (i.e., projected Q-ascent) is not applicable, to the best of our knowledge. Instead, we proceed the proof by computing the state-wise improvement lower bound of PPG based on its inherent structure. In addition, the finite iteration convergence of PPG for any constant step size is further established, which is also new. Jiacai Liu, Wenye Li 0002, Dachao Lin, Ke Wei 0001, Zhihua Zhang 0004 |
J. Mach. Learn. Res. | 5 |
| 2024 | Statistical Efficiency of Distributional Temporal Difference LearningabstractDistributional reinforcement learning (DRL) has achieved empirical success in various domains.
One of the core tasks in the field of DRL is distributional policy evaluation, which involves estimating the return distribution $\eta^\pi$ for a given policy $\pi$.
The distributional temporal difference learning has been accordingly proposed, which
is an extension of the temporal difference learning (TD) in the classic RL area.
In the tabular case, Rowland et al. [2018] and Rowland et al. [2023] proved the asymptotic convergence of two instances of distributional TD, namely categorical temporal difference learning (CTD) and quantile temporal difference learning (QTD), respectively.
In this paper, we go a step further and analyze the finite-sample performance of distributional TD.
To facilitate theoretical analysis, we propose a non-parametric distributional TD learning (NTD).
For a $\gamma$-discounted infinite-horizon tabular Markov decision process,
we show that for NTD we need $\widetilde O\left(\frac{1}{\varepsilon^{2p}(1-\gamma)^{2p+1}}\right)$ iterations to achieve an $\varepsilon$-optimal estimator with high probability, when the estimation error is measured by the $p$-Wasserstein distance.
This sample complexity bound is minimax optimal (up to logarithmic factors) in the case of the $1$-Wasserstein distance.
To achieve this, we establish a novel Freedman's inequality in Hilbert spaces, which would be of independent interest.
In addition, we revisit CTD, showing that the same non-asymptotic convergence bounds hold for CTD in the case of the $p$-Wasserstein distance. Liangyu Zhang, Zhihua Zhang 0004 |
NeurIPS | 3 |
| 2024 | A Random Projection Approach to Personalized Federated Learning: Enhancing Communication Efficiency, Robustness, and FairnessabstractPersonalized Federated Learning (FL) faces many challenges such as expensive communication costs, training-time adversarial attacks, and performance unfairness across devices. Recent developments witness a trade-off between a reference model and local models to achieve personalization. Following the avenue, we propose a personalized FL method toward the three goals. When it is time to communicate, our method projects local models into a shared-and-fixed low-dimensional random subspace and uses infimal convolution to control the deviation between the reference model and projected local models. We theoretically show our method converges for both strongly convex and non-convex but smooth objectives with square regularizers and the convergence dependence on the projection dimension is mild. We also illustrate the benefits of robustness and fairness on a class of linear problems. Finally, we conduct a large number of experiments to show the empirical superiority of our method over several state-of-the-art methods on the three aspects. Yuze Han, Xiang Li 0050, Shiyun Lin, Zhihua Zhang 0004 |
J. Mach. Learn. Res. | 4 |
| 2024 | Fedpower: privacy-preserving distributed eigenspace estimation
Xiang Li 0050, Xiangyu Chang, Shusen Wang, Zhihua Zhang 0004 |
Mach. Learn. | 5 |
| 2024 | Semi-Infinitely Constrained Markov Decision Processes and Provably Efficient Reinforcement LearningabstractWe propose a novel generalization of constrained Markov decision processes (CMDPs) that we call the semi-infinitely constrained Markov decision process (SICMDP). Particularly, we consider a continuum of constraints instead of a finite number of constraints as in the case of ordinary CMDPs. We also devise two reinforcement learning algorithms for SICMDPs that we refer to as SI-CMBRL and SI-CPO. SI-CMBRL is a model-based reinforcement learning algorithm. Given an estimate of the transition model, we first transform the reinforcement learning problem into a linear semi-infinitely programming (LSIP) problem and then use the dual exchange method in the LSIP literature to solve it. SI-CPO is a policy optimization algorithm. Borrowing ideas from the cooperative stochastic approximation approach, we make alternative updates to the policy parameters to maximize the reward or minimize the cost. To the best of our knowledge, we are the first to apply tools from semi-infinitely programming (SIP) to solve constrained reinforcement learning problems. We present theoretical analysis for SI-CMBRL and SI-CPO, identifying their iteration complexity and sample complexity. We also conduct extensive numerical experiments to illustrate the SICMDP model and demonstrate that our proposed algorithms are able to solve complex control tasks leveraging modern deep reinforcement learning techniques. Liangyu Zhang, Zhihua Zhang 0004 |
IEEE Trans. Pattern Anal. Mach. Intell. | 4 |
| 2023 | A Statistical Analysis of Polyak-Ruppert Averaged Q-LearningabstractWe study Q-learning with Polyak-Ruppert averaging (a.k.a., averaged Q-learning) in a discounted markov decision process in synchronous and tabular settings. Under a Lipschitz condition, we establish a functional central limit theorem for the averaged iteration $\bar{\mathbf{Q}}_T$ and show that its standardized partial-sum process converges weakly to a rescaled Brownian motion. The FCLT implies a fully online inference method for reinforcement learning. Furthermore, we show that $\bar{\mathbf{Q}}_T$ is the regular asymptotically linear (RAL) estimator for the optimal Q-value function $\mathbf{Q}^*$ that has the most efficient influence function. We present a nonasymptotic analysis for the $\ell_{\infty}$ error, $\mathbb{E}\|\bar{\mathbf{Q}}_T-\mathbf{Q}^*\|_{\infty}$, showing that it matches the instance-dependent lower bound for polynomial step sizes. Similar results are provided for entropy-regularized Q-Learning without the Lipschitz condition. Xiang Li 0050, Jiadong Liang, Zhihua Zhang 0004, Michael I. Jordan |
AISTATS | 4 |
| 2023 | Stochastic Distributed Optimization under Average Second-order Similarity: Algorithms and AnalysisabstractWe study finite-sum distributed optimization problems involving a master node and $n-1$ local nodes under the popular $\delta$-similarity and $\mu$-strong convexity conditions. We propose two new algorithms, SVRS and AccSVRS, motivated by previous works. The non-accelerated SVRS method combines the techniques of gradient sliding and variance reduction and achieves a better communication complexity of $\tilde{\mathcal{O}}(n {+} \sqrt{n}\delta/\mu)$ compared to existing non-accelerated algorithms. Applying the framework proposed in Katyusha X, we also develop a directly accelerated version named AccSVRS with the $\tilde{\mathcal{O}}(n {+} n^{3/4}\sqrt{\delta/\mu})$ communication complexity. In contrast to existing results, our complexity bounds are entirely smoothness-free and exhibit superiority in ill-conditioned cases. Furthermore, we establish a nearly matched lower bound to verify the tightness of our AccSVRS method. Dachao Lin, Yuze Han, Haishan Ye, Zhihua Zhang 0004 |
NeurIPS | 4 |
| 2022 | Federated Reinforcement Learning with Environment HeterogeneityabstractWe study Federated Reinforcement Learning (FedRL) problem in which $n$ agents collaboratively learn a single policy without sharing the trajectories they collected during agent-environment interaction. In this paper, we stress the constraint of environment heterogeneity, which means $n$ environments corresponding to these $n$ agents have different state-transitions. To obtain a value function or a policy function which optimizes the overall performance in all environments, we propose two algorithms, we propose two federated RL algorithms, QAvg and PAvg. We theoretically prove that these algorithms converge to suboptimal solutions, while such suboptimality depends on how heterogeneous these $n$ environments are. Moreover, we propose a heuristic that achieves personalization by embedding the $n$ environments into $n$ vectors. The personalization heuristic not only improves the training but also allows for better generalization to new environments. Shusen Wang, Zhihua Zhang 0004 |
AISTATS | 5 |
| 2022 | Statistical Estimation and Online Inference via Local SGDabstractWe analyze the novel Local SGD in federated Learning, a multi-round estimation procedure that uses intermittent communication to improve communication efficiency. Under a $2{+}\delta$ moment condition on stochastic gradients, we first establish a {\it functional central limit theorem} that shows the averaged iterates of Local SGD converge weakly to a rescaled Brownian motion. We next provide two iterative inference methods: the {\it plug-in} and the {\it random scaling}. Random scaling constructs an asymptotically pivotal statistic for inference by using the information along the whole Local SGD path. Both the methods are communication efficient and applicable to online data. Our results show that Local SGD simultaneously achieves both statistical efficiency and communication efficiency. Xiang Li 0050, Jiadong Liang, Xiangyu Chang, Zhihua Zhang 0004 |
COLT | 4 |
| 2022 | On Non-local Convergence Analysis of Deep Linear NetworksabstractIn this paper, we study the non-local convergence properties of deep linear networks. Specifically, under the quadratic loss, we consider optimizing deep linear networks in which there is at least a layer with only one neuron. We describe the convergent point of trajectories with an arbitrary balanced starting point under gradient flow, including the paths which converge to one of the saddle points. We also show specific convergence rates of trajectories that converge to the global minimizers by stages. We conclude that the rates vary from polynomial to linear. As far as we know, our results are the first to give a non-local analysis of deep linear neural networks with arbitrary balanced initialization, rather than the lazy training regime which has dominated the literature on neural networks or the restricted benign initialization. Dachao Lin, Zhihua Zhang 0004 |
ICML | 3 |
| 2022 | Asymptotic Behaviors of Projected Stochastic Approximation: A Jump Diffusion PerspectiveabstractIn this paper, we consider linearly constrained stochastic approximation problems with federated learning (FL) as a special case. We propose a stochastic approximation algorithm named by LPSA with probabilistic projections to ensure feasibility so that projections are performed with probability $p_n$ at the $n$-th iteration. Considering a specific family of the probability $p_n$ and step size $\eta_n$, we analyze our algorithm from an asymptotic and continuous perspective. Using a novel jump diffusion approximation, we show that the trajectories consisting of properly rescaled last iterates weakly converge to the solution of specific SDEs. By analyzing the SDEs, we identify the asymptotic behaviors of LPSA for different choices of $(p_n, \eta_n)$. We find the algorithm presents an intriguing asymptotic bias-variance trade-off according to the relative magnitude of $p_n$ w.r.t. $\eta_n$. It provides insights on how to choose appropriate $\{(p_n, \eta_n)\}_{n \geq 1}$ to minimize the projection complexity. Jiadong Liang, Yuze Han, Xiang Li 0050, Zhihua Zhang 0004 |
NeurIPS | 4 |
| 2022 | Personalized Federated Learning towards Communication Efficiency, Robustness and FairnessabstractPersonalized Federated Learning faces many challenges such as expensive communication costs, training-time adversarial attacks, and performance unfairness across devices. Recent developments witness a trade-off between a reference model and local models to achieve personalization. We follow the avenue and propose a personalized FL method towards the three goals. When it is time to communicate, our method projects local models into a shared-and-fixed low-dimensional random subspace and uses infimal convolution to control the deviation between the reference model and projected local models. We theoretically show our method converges for smooth objectives with square regularizers and the convergence dependence on the projection dimension is mild. We also illustrate the benefits of robustness and fairness on a class of linear problems. Finally, we conduct a large number of experiments to show the empirical superiority of our method over several state-of-the-art methods on the three aspects. Shiyun Lin, Yuze Han, Xiang Li 0050, Zhihua Zhang 0004 |
NeurIPS | 4 |
| 2022 | Semi-infinitely Constrained Markov Decision ProcessesabstractWe propose a generalization of constrained Markov decision processes (CMDPs) that we call the \emph{semi-infinitely constrained Markov decision process} (SICMDP).Particularly, in a SICMDP model, we impose a continuum of constraints instead of a finite number of constraints as in the case of ordinary CMDPs.We also devise a reinforcement learning algorithm for SICMDPs that we call SI-CRL.We first transform the reinforcement learning problem into a linear semi-infinitely programming (LSIP) problem and then use the dual exchange method in the LSIP literature to solve it.To the best of our knowledge, we are the first to apply tools from semi-infinitely programming (SIP) to solve reinforcement learning problems.We present theoretical analysis for SI-CRL, identifying its sample complexity and iteration complexity.We also conduct extensive numerical examples to illustrate the SICMDP model and validate the SI-CRL algorithm. Liangyu Zhang, Zhihua Zhang 0004 |
NeurIPS | 4 |
| 2022 | On the landscape of one-hidden-layer sparse networks and beyond
Dachao Lin, Ruoyu Sun 0001, Zhihua Zhang 0004 |
Artif. Intell. | 3 |
| 2022 | Explicit Convergence Rates of Greedy and Random Quasi-Newton MethodsabstractOptimization is important in machine learning problems, and quasi-Newton methods have a reputation as the most efficient numerical methods for smooth unconstrained optimization. In this paper, we study the explicit superlinear convergence rates of quasi-Newton methods and address two open problems mentioned by Rodomanov and Nesterov (2021b). First, we extend Rodomanov and Nesterov (2021b)’s results to random quasi-Newton methods, which include common DFP, BFGS, SR1 methods. Such random methods employ a random direction for updating the approximate Hessian matrix in each iteration. Second, we focus on the specific quasi-Newton methods: SR1 and BFGS methods. We provide improved versions of greedy and random methods with provable better explicit (local) superlinear convergence rates. Our analysis is closely related to the approximation of a given Hessian matrix, unconstrained quadratic objective, as well as the general strongly convex, smooth, and strongly self-concordant functions. Dachao Lin, Haishan Ye, Zhihua Zhang 0004 |
J. Mach. Learn. Res. | 3 |
| 2021 | Multi-split Reversible Transformers Can Enhance Neural Machine TranslationabstractLarge-scale transformers have been shown the state-of-the-art on neural machine translation.However, training these increasingly wider and deeper models could be tremendously memory intensive.We reduce the memory burden by employing the idea of reversible networks that a layer's input can be reconstructed from its output.We design three types of multi-split based reversible transformers.We also devise a corresponding backpropagation algorithm, which does not need to store activations for most layers.Furthermore, we present two fine-tuning techniques: splits shuffle and self ensemble, to boost translation accuracy.Specifically, our best models surpass the vanilla transformer by at least 1.4 BLEU points in three datasets.Our largescale reversible models achieve 30.0 BLEU in WMT'14 En-De and 43.5 BLEU in WMT'14 En-Fr, beating several very strong baselines with less than half of the training memory. Yuekai Zhao, Shuchang Zhou 0001, Zhihua Zhang 0004 |
EACL | 3 |
| 2021 | Communication-Efficient Distributed SVD via Local Power IterationsabstractWe study distributed computing of the truncated singular value decomposition (SVD). We develop an algorithm that we call \texttt{LocalPower} for improving communication efficiency. Specifically, we uniformly partition the dataset among $m$ nodes and alternate between multiple (precisely $p$) local power iterations and one global aggregation. In the aggregation, we propose to weight each local eigenvector matrix with orthogonal Procrustes transformation (OPT). As a practical surrogate of OPT, sign-fixing, which uses a diagonal matrix with $\pm 1$ entries as weights, has better computation complexity and stability in experiments. We theoretically show that under certain assumptions \texttt{LocalPower} lowers the required number of communications by a factor of $p$ to reach a constant accuracy. We also show that the strategy of periodically decaying $p$ helps obtain high-precision solutions. We conduct experiments to demonstrate the effectiveness of \texttt{LocalPower}. Xiang Li 0050, Shusen Wang, Zhihua Zhang 0004 |
ICML | 4 |
| 2021 | Faster Directional Convergence of Linear Neural Networks under Spherically Symmetric DataabstractIn this paper, we study gradient methods for training deep linear neural networks with binary cross-entropy loss. In particular, we show global directional convergence guarantees from a polynomial rate to a linear rate for (deep) linear networks with spherically symmetric data distribution, which can be viewed as a specific zero-margin dataset. Our results do not require the assumptions in other works such as small initial loss, presumed convergence of weight direction, or overparameterization. We also characterize our findings in experiments. Dachao Lin, Ruoyu Sun 0001, Zhihua Zhang 0004 |
NeurIPS | 3 |
| 2021 | Greedy and Random Quasi-Newton Methods with Faster Explicit Superlinear ConvergenceabstractIn this paper, we follow Rodomanov and Nesterov’s work to study quasi-Newton methods. We focus on the common SR1 and BFGS quasi-Newton methods to establish better explicit (local) superlinear convergence rates. First, based on the greedy quasi-Newton update which greedily selects the direction to maximize a certain measure of progress, we improve the convergence rate to a condition-number-free superlinear convergence rate. Second, based on the random quasi-Newton update that selects the direction randomly from a spherically symmetric distribution, we show the same superlinear convergence rate established as above. Our analysis is closely related to the approximation of a given Hessian matrix, unconstrained quadratic objective, as well as the general strongly convex, smooth, and strongly self-concordant functions. Dachao Lin, Haishan Ye, Zhihua Zhang 0004 |
NeurIPS | 3 |
| 2021 | Approximate Newton MethodsabstractMany machine learning models involve solving optimization problems. Thus, it is important to address a large-scale optimization problem in big data applications. Recently, subsampled Newton methods have emerged to attract much attention due to their efficiency at each iteration, rectified a weakness in the ordinary Newton method of suffering a high cost in each iteration while commanding a high convergence rate. Other efficient stochastic second order methods have been also proposed. However, the convergence properties of these methods are still not well understood. There are also several important gaps between the current convergence theory and the empirical performance in real applications. In this paper, we aim to fill these gaps. We propose a unifying framework to analyze both local and global convergence properties of second order methods. Accordingly, we present our theoretical results which match the empirical performance in real applications well. Haishan Ye, Luo Luo, Zhihua Zhang 0004 |
J. Mach. Learn. Res. | 3 |
| 2021 | Accelerated Proximal Subsampled Newton MethodabstractComposite function optimization problem often arises in machine learning known as regularized empirical minimization. We introduce the acceleration technique to the Newton-type proximal method and propose a novel algorithm called accelerated proximal subsampled Newton method (APSSN). APSSN only subsamples a small subset of samples to construct an approximate Hessian that achieves computational efficiency. At the same time, APSSN still keeps a fast convergence rate. Furthermore, we obtain the scaled proximal mapping by solving its dual problem using the semismooth Newton method instead of resorting to the first-order methods. Due to our sampling strategy and the fast convergence rate of the semismooth Newton method, we can get the scaled proximal mapping efficiently. Both our theoretical analysis and empirical study show that APSSN is an effective and computationally efficient algorithm for composite function optimization problems. Haishan Ye, Luo Luo, Zhihua Zhang 0004 |
IEEE Trans. Neural Networks Learn. Syst. | 3 |
| 2020 | Do Subsampled Newton Methods Work for High-Dimensional Data?abstractSubsampled Newton methods approximate Hessian matrices through subsampling techniques to alleviate the per-iteration cost. Previous results require Ω (d) samples to approximate Hessians, where d is the dimension of data points, making it less practical for high-dimensional data. The situation is deteriorated when d is comparably as large as the number of data points n, which requires to take the whole dataset into account, making subsampling not useful. This paper theoretically justifies the effectiveness of subsampled Newton methods on strongly convex empirical risk minimization with high dimensional data. Specifically, we provably require only Θ˜(deffγ) samples for approximating the Hessian matrices, where deffγ is the γ-ridge leverage and can be much smaller than d as long as nγ ≫ 1. Our theories work for three types of Newton methods: subsampled Netwon, distributed Newton, and proximal Newton. Xiang Li 0050, Shusen Wang, Zhihua Zhang 0004 |
AAAI | 3 |
| 2020 | Efficient Spectrum-Revealing CUR Matrix DecompositionabstractThe CUR matrix decomposition is an important tool for low-rank matrix approximation. It approximates a data matrix though selecting a small number of columns and rows of the matrix. Those CUR algorithms with gap-dependent approximation bounds can obtain high approximation quality for matrices with good singular value spectrum decay, but they have impractically high time complexities. In this paper, we propose a novel CUR algorithm based on truncated LU factorization with an efficient variant of complete pivoting. Our algorithm has gap-dependent approximation bounds on both spectral and Frobenius norms while maintaining high efficiency. Numerical experiments demonstrate the effectiveness of our algorithm and verify our theoretical guarantees. Cheng Chen 0015, Zhihua Zhang 0004, Weinan Zhang 0001, Yong Yu 0001 |
AISTATS | 3 |
| 2020 | On the Convergence of FedAvg on Non-IID Data
Xiang Li 0050, Kaixuan Huang, Shusen Wang, Zhihua Zhang 0004 |
ICLR | 5 |
| 2020 | Lower Complexity Bounds for Finite-Sum Convex-Concave Minimax Optimization ProblemsabstractThis paper studies the lower bound complexity for minimax optimization problem whose objective function is the average of $n$ individual smooth convex-concave functions. We consider the algorithm which gets access to gradient and proximal oracle for each individual component. For the strongly-convex-strongly-concave case, we prove such an algorithm can not reach an $\varepsilon$-suboptimal point in fewer than $\Omega\left((n+\kappa)\log(1/\varepsilon)\right)$ iterations, where $\kappa$ is the condition number of the objective function. This lower bound matches the upper bound of the existing incremental first-order oracle algorithm stochastic variance-reduced extragradient. We develop a novel construction to show the above result, which partitions the tridiagonal matrix of classical examples into $n$ groups. This construction is friendly to the analysis of incremental gradient and proximal oracle and we also extend the analysis to general convex-concave cases. Guangzeng Xie, Luo Luo, Yijiang Lian, Zhihua Zhang 0004 |
ICML | 4 |
| 2020 | Nesterov's Acceleration for Approximate NewtonabstractOptimization plays a key role in machine learning. Recently, stochastic second-order methods have attracted considerable attention because of their low computational cost in each iteration. However, these methods might suffer from poor performance when the Hessian is hard to be approximate well in a computation-efficient way. To overcome this dilemma, we resort to Nesterov's acceleration to improve the convergence performance of these second-order methods and propose accelerated approximate Newton. We give the theoretical convergence analysis of accelerated approximate Newton and show that Nesterov's acceleration can improve the convergence rate. Accordingly, we propose an accelerated regularized sub-sampled Newton (ARSSN) which performs much better than the conventional regularized sub-sampled Newton empirically and theoretically. Moreover, we show that ARSSN has better performance than classical first-order methods empirically. Haishan Ye, Luo Luo, Zhihua Zhang 0004 |
J. Mach. Learn. Res. | 3 |
| 2019 | Lipschitz Generative Adversarial NetsabstractIn this paper we show that generative adversarial networks (GANs) without restriction on the discriminative function space commonly suffer from the problem that the gradient produced by the discriminator is uninformative to guide the generator. By contrast, Wasserstein GAN (WGAN), where the discriminative function is restricted to 1-Lipschitz, does not suffer from such a gradient uninformativeness problem. We further show in the paper that the model with a compact dual form of Wasserstein distance, where the Lipschitz condition is relaxed, may also theoretically suffer from this issue. This implies the importance of Lipschitz condition and motivates us to study the general formulation of GANs with Lipschitz constraint, which leads to a new family of GANs that we call Lipschitz GANs (LGANs). We show that LGANs guarantee the existence and uniqueness of the optimal discriminative function as well as the existence of a unique Nash equilibrium. We prove that LGANs are generally capable of eliminating the gradient uninformativeness problem. According to our empirical analysis, LGANs are more stable and generate consistently higher quality samples compared with WGAN. Zhiming Zhou 0001, Jiadong Liang, Yuxuan Song 0002, Lantao Yu, Hongwei Wang 0004, Weinan Zhang 0001, Yong Yu 0001, Zhihua Zhang 0004 |
ICML | 8 |
| 2019 | A Regularized Approach to Sparse Optimal Policy in Reinforcement LearningabstractWe propose and study a general framework for regularized Markov decision processes (MDPs) where the goal is to find an optimal policy that maximizes the expected discounted total reward plus a policy regularization term. The extant entropy-regularized MDPs can be cast into our framework. Moreover, under our framework, many regularization terms can bring multi-modality and sparsity, which are potentially useful in reinforcement learning. In particular, we present sufficient and necessary conditions that induce a sparse optimal policy. We also conduct a full mathematical analysis of the proposed regularized MDPs, including the optimality condition, performance error, and sparseness control. We provide a generic method to devise regularization forms and propose off-policy actor critic algorithms in complex environment settings. We empirically analyze the numerical properties of optimal policies and compare the performance of different sparse regularization forms in discrete and continuous environments. Xiang Li 0050, Zhihua Zhang 0004 |
NeurIPS | 3 |
| 2019 | Robust Frequent Directions with Application in Online LearningabstractThe frequent directions (FD) technique is a deterministic approach for online sketching that has many applications in machine learning. The conventional FD is a heuristic procedure that often outputs rank deficient matrices. To overcome the rank deficiency problem, we propose a new sketching strategy called robust frequent directions (RFD) by introducing a regularization term. RFD can be derived from an optimization problem. It updates the sketch matrix and the regularization term adaptively and jointly. RFD reduces the approximation error of FD without increasing the computational cost. We also apply RFD to online learning and propose an effective hyperparameter-free online Newton algorithm. We derive a regret bound for our online Newton algorithm based on RFD, which guarantees the robustness of the algorithm. The experimental studies demonstrate that the proposed method outperforms state-of-the-art second order online learning algorithms. Luo Luo, Cheng Chen 0015, Zhihua Zhang 0004, Wu-Jun Li, Tong Zhang 0001 |
J. Mach. Learn. Res. | 3 |
| 2019 | Fast stochastic second-order method logarithmic in condition number
Haishan Ye, Guangzeng Xie, Luo Luo, Zhihua Zhang 0004 |
Pattern Recognit. | 4 |
| 2018 | Sketched Follow-The-Regularized-Leader for Online Factorization MachineabstractFactorization Machine (FM) is a supervised machine learning model for feature engineering, which is widely used in many real-world applications. In this paper, we consider the case that the data samples arrive sequentially. The existing convex formulation for online FM has the strong theoretical guarantee and stable performance in practice, but the computational cost is typically expensive when the data is high-dimensional. To address this weakness, we devise a novel online learning algorithm called Sketched Follow-The-Regularizer-Leader (SFTRL). SFTRL presents the parameters of FM implicitly by maintaining low-rank matrices and updates the parameters via sketching. More specifically, we propose Generalized Frequent Directions to approximate indefinite symmetric matrices in a streaming way, making that the sum of historical gradients for FM could be estimated with tighter error bound efficiently. With mild assumptions, we prove that the regret bound of SFTRL is close to that of the standard FTRL. Experimental results show that SFTRL has better prediction quality than the state-of-the-art online FM algorithms in much lower time and space complexities. Luo Luo, Wenpeng Zhang 0003, Zhihua Zhang 0004, Wenwu Zhu 0001, Tong Zhang 0001, Jian Pei 0001 |
KDD | 3 |
| 2017 | Communication Lower Bounds for Distributed Convex Optimization: Partition Data on FeaturesabstractRecently, there has been an increasing interest in designing distributed convex optimization algorithms under the setting where the data matrix is partitioned on features. Algorithms under this setting sometimes have many advantages over those under the setting where data is partitioned on samples, especially when the number of features is huge. Therefore, it is important to understand the inherent limitations of these optimization problems. In this paper, with certain restrictions on the communication allowed in the procedures, we develop tight lower bounds on communication rounds for a broad class of non-incremental algorithms under this setting. We also provide a lower bound on communication rounds for a class of (randomized) incremental algorithms. Luo Luo, Zhihua Zhang 0004 |
AAAI | 3 |
| 2017 | Approximate Newton Methods and Their Local ConvergenceabstractMany machine learning models are reformulated as optimization problems. Thus, it is important to solve a large-scale optimization problem in big data applications. Recently, subsampled Newton methods have emerged to attract much attention for optimization due to their efficiency at each iteration, rectified a weakness in the ordinary Newton method of suffering a high cost in each iteration while commanding a high convergence rate. Other efficient stochastic second order methods are also proposed. However, the convergence properties of these methods are still not well understood. There are also several important gaps between the current convergence theory and performance in real applications. In this paper, we aim to fill these gaps. We propose a unifying framework to analyze local convergence properties of second order methods. Based on this framework, our theoretical analysis matches the performance in real applications. Haishan Ye, Luo Luo, Zhihua Zhang 0004 |
ICML | 3 |
| 2017 | Fast Fisher discriminant analysis with randomized algorithms
Haishan Ye, Cheng Chen 0015, Zhihua Zhang 0004 |
Pattern Recognit. | 4 |
| 2016 | Frequent Direction Algorithms for Approximate Matrix Multiplication with Applications in CCA
Qiaomin Ye, Luo Luo, Zhihua Zhang 0004 |
IJCAI | 3 |
| 2016 | Quasi-Newton Hamiltonian Monte Carlo
Tianfan Fu, Luo Luo, Zhihua Zhang 0004 |
UAI | 3 |
| 2016 | SPSD Matrix Approximation vis Column Selection: Theories, Algorithms, and ExtensionsabstractSymmetric positive semidefinite (SPSD) matrix approximation is an important problem with applications in kernel methods. However, existing SPSD matrix approximation methods such as the Nyström method only have weak error bounds. In this paper we conduct in-depth studies of an SPSD matrix approximation model and establish strong relative-error bounds. We call it the prototype model for it has more efficient and effective extensions, and some of its extensions have high scalability. Though the prototype model itself is not suitable for large- scale data, it is still useful to study its properties, on which the analysis of its extensions relies. This paper offers novel theoretical analysis, efficient algorithms, and a highly accurate extension. First, we establish a lower error bound for the prototype model and improve the error bound of an existing column selection algorithm to match the lower bound. In this way, we obtain the first optimal column selection algorithm for the prototype model. We also prove that the prototype model is exact under certain conditions. Second, we develop a simple column selection algorithm with a provable error bound. Third, we propose a so-called spectral shifting model to make the approximation more accurate when the eigenvalues of the matrix decay slowly, and the improvement is theoretically quantified. The spectral shifting method can also be applied to improve other SPSD matrix approximation models. Shusen Wang, Luo Luo, Zhihua Zhang 0004 |
J. Mach. Learn. Res. | 3 |
| 2016 | Towards More Efficient SPSD Matrix Approximation and CUR Matrix DecompositionabstractSymmetric positive semi-definite (SPSD) matrix approximation methods have been extensively used to speed up large-scale eigenvalue computation and kernel learning methods. The standard sketch based method, which we call the prototype model, produces relatively accurate approximations, but is inefficient on large square matrices. The Nyström method is highly efficient, but can only achieve low accuracy. In this paper we propose a novel model that we call the fast SPSD matrix approximation model. The fast model is nearly as efficient as the Nyström method and as accurate as the prototype model. We show that the fast model can potentially solve eigenvalue problems and kernel learning problems in linear time with respect to the matrix size $n$ to achieve $1+\epsilon$ relative-error, whereas both the prototype model and the Nyström method cost at least quadratic time to attain comparable error bound. Empirical comparisons among the prototype model, the Nyström method, and our fast model demonstrate the superiority of the fast model. We also contribute new understandings of the Nyström method. The Nyström method is a special instance of our fast model and is approximation to the prototype model. Our technique can be straightforwardly applied to make the CUR matrix decomposition more efficiently computed without much affecting the accuracy. Shusen Wang, Zhihua Zhang 0004, Tong Zhang 0001 |
J. Mach. Learn. Res. | 2 |
| 2015 | Support Matrix MachinesabstractIn many classification problems such as electroencephalogram (EEG) classification and image classification, the input features are naturally represented as matrices rather than vectors or scalars. In general, the structure information of the original feature matrix is useful and informative for data analysis tasks such as classification. One typical structure information is the correlation between columns or rows in the feature matrix. To leverage this kind of structure information, we propose a new classification method that we call support matrix machine (SMM). Specifically, SMM is defined as a hinge loss plus a so-called spectral elastic net penalty which is a spectral extension of the conventional elastic net over a matrix. The spectral elastic net enjoys a property of grouping effect, i.e., strongly correlated columns or rows tend to be selected altogether or not. Since the optimization problem for SMM is convex, this encourages us to devise an alternating direction method of multipliers algorithm for solving the problem. Experimental results on EEG and face image classification data show that our model is more robust and efficient than the state-of-the-art methods. Luo Luo, Yubo Xie, Zhihua Zhang 0004, Wu-Jun Li |
ICML | 3 |
| 2014 | Multicategory large margin classification methods: Hinge losses vs. coherence functions
Zhihua Zhang 0004, Cheng Chen 0015, Guang Dai, Wu-Jun Li, Dit-Yan Yeung |
Artif. Intell. | 1 |
| 2013 | A Scalable Approach to Column-Based Low-Rank Matrix Approximation
Yifan Pi, Haoruo Peng, Shuchang Zhou 0001, Zhihua Zhang 0004 |
IJCAI | 4 |
| 2012 | Sublinear Algorithms for Penalized Logistic Regression in Massive Datasets
Haoruo Peng, Edward Y. Chang, Shuchang Zhou 0001, Zhihua Zhang 0004 |
ECML/PKDD (1) | 5 |
| 2012 | Coherence functions with applications in large-margin classification methods
Zhihua Zhang 0004, Dehua Liu, Guang Dai, Michael I. Jordan |
J. Mach. Learn. Res. | 1 |
| 2012 | EP-GIG Priors and Applications in Bayesian Sparse Learning
Zhihua Zhang 0004, Shusen Wang, Dehua Liu, Michael I. Jordan |
J. Mach. Learn. Res. | 1 |
| 2011 | Generalized Latent Factor Models for Social Network Analysis
Wu-Jun Li, Dit-Yan Yeung, Zhihua Zhang 0004 |
IJCAI | 3 |
| 2011 | Bayesian Generalized Kernel Mixed Models
Zhihua Zhang 0004, Guang Dai, Michael I. Jordan |
J. Mach. Learn. Res. | 1 |
| 2010 | Regularized Discriminant Analysis, Ridge Regression and Beyond
Zhihua Zhang 0004, Guang Dai, Congfu Xu, Michael I. Jordan |
J. Mach. Learn. Res. | 1 |
| 2010 | A regularization framework for multiclass classification: A deterministic annealing approach
Zhihua Zhang 0004, Gang Wang 0004, Dit-Yan Yeung, Guang Dai, Frederick H. Lochovsky |
Pattern Recognit. | 1 |
| 2009 | Probabilistic Relational PCAabstractOne crucial assumption made by both principal component analysis (PCA) and probabilistic PCA (PPCA) is that the instances are independent and identically distributed (i.i.d.). However, this common i.i.d. assumption is unreasonable for relational data. In this paper, by explicitly modeling covariance between instances as derived from the relational information, we propose a novel probabilistic dimensionality reduction method, called probabilistic relational PCA (PRPCA), for relational data analysis. Although the i.i.d. assumption is no longer adopted in PRPCA, the learning algorithms for PRPCA can still be devised easily like those for PPCA which makes explicit use of the i.i.d. assumption. Experiments on real-world data sets show that PRPCA can effectively utilize the relational information to dramatically outperform PCA and achieve state-of-the-art performance. Wu-Jun Li, Dit-Yan Yeung, Zhihua Zhang 0004 |
NIPS | 3 |
| 2009 | A Flexible and Efficient Algorithm for Regularized Fisher Discriminant Analysis
Zhihua Zhang 0004, Guang Dai, Michael I. Jordan |
ECML/PKDD (2) | 1 |
| 2008 | Posterior Consistency of the Silverman g-prior in Bayesian Model ChoiceabstractKernel supervised learning methods can be unified by utilizing the tools from regularization theory. The duality between regularization and prior leads to interpreting regularization methods in terms of maximum a posteriori estimation and has motivated Bayesian interpretations of kernel methods. In this paper we pursue a Bayesian interpretation of sparsity in the kernel setting by making use of a mixture of a point-mass distribution and prior that we refer to as ``Silverman's g-prior.'' We provide a theoretical analysis of the posterior consistency of a Bayesian model choice procedure based on this prior. We also establish the asymptotic relationship between this procedure and the Bayesian information criterion. Zhihua Zhang 0004, Michael I. Jordan, Dit-Yan Yeung |
NIPS | 1 |
| 2007 | Surrogate maximization/minimization algorithms and extensions
Zhihua Zhang 0004, James T. Kwok, Dit-Yan Yeung |
Mach. Learn. | 1 |
| 2007 | Semiparametric Regression Using Student t ProcessesabstractIn this paper, we propose a latent factor regression model, in which priors are assigned to both the latent regression vector and the error term, by using reproducing kernels. The resulting regression function follows a stochastic process known as a student process. The model is attractive because its implementation is based on a tractable posterior predictive distribution and a simple expectation-maximization (EM) estimation algorithm. In addition, treating the transductive inference as a missing data problem, we devise the EM algorithm to deal with the parameter estimation as well as the response prediction in a single paradigm. The model is also elaborated for multivariate-response regression problems. For this purpose, we present a generalization of multivariate models and some of its properties. Experimental results show our approaches to be effective. Zhihua Zhang 0004, Gang Wu 0005, Edward Y. Chang |
IEEE Trans. Neural Networks | 1 |
| 2006 | Adaptive non-linear clustering in data streamsabstractData stream clustering has emerged as a challenging and interesting problem over the past few years. Due to the evolving nature, and one-pass restriction imposed by the data stream model, traditional clustering algorithms are inapplicable for stream clustering. This problem becomes even more challenging when the data is high-dimensional and the clusters are not linearly separable in the input space. In this paper, we propose a nonlinear stream clustering algorithm that adapts to the stream's evolutionary changes. Using the kernel methods for dealing with the non-linearity of data separation, we propose a novel 2-tier stream clustering architecture. Tier-1 captures the temporal locality in the stream, by partitioning it into segments, using a kernel-based novelty detection approach. Tier-2 exploits this segment structure to continuously project the streaming data nonlinearly onto a low-dimensional space (LDS), before assigning them to a cluster. We demonstrate the effectiveness of our approach through extensive experimental evaluation on various real-world datasets. Zhihua Zhang 0004, Edward Y. Chang |
CIKM | 2 |
| 2006 | Bayesian Multicategory Support Vector Machines
Zhihua Zhang 0004, Michael I. Jordan |
UAI | 1 |
| 2006 | Model-based transductive learning of the kernel matrix
Zhihua Zhang 0004, James T. Kwok, Dit-Yan Yeung |
Mach. Learn. | 1 |
| 2005 | Annealed Discriminant Analysis
Gang Wang 0004, Zhihua Zhang 0004, Frederick H. Lochovsky |
ECML | 2 |
| 2005 | A Bernoulli Relational Model for Nonlinear EmbeddingabstractThe notion of relations is extremely important in mathematics. In this paper, we use relations to describe the embedding problem and propose a novel stochastic relational model for nonlinear embedding. Given some relation among points in a high-dimensional space, we start from preserving the same relation in a low embedded space and model the relation as probabilistic distributions over these two spaces, respectively. We illustrate that the stochastic neighbor embedding and the Gaussian process latent variable model can be derived from our relational model. Moreover we devise a new stochastic embedding model and refer to it as Bernoulli relational embedding (BRE). BRE's ability in nonlinear dimensionality reduction is illustrated on a set of synthetic data and collections of bitmaps of handwritten digits and face images. Gang Wang 0004, Zhihua Zhang 0004, Frederick H. Lochovsky |
ICDM | 3 |
| 2005 | Learning with non-metric proximity matricesabstractMany emerging applications formulate non-metric proximity matrices (non-positive semidefinite), and hence cannot fit into the framework of kernel machines. A popular approach to this problem is to transform the spectrum of the similarity matrix so as to generate a positive semidefinite kernel matrix. In this paper, we explore four representative transformation methods: denoise, flip, diffusion, and shift. Theoretically, we discuss a generalization problem where the test data are not available during transformation, and thus propose an efficient algorithm to address the problem of updating the cross-similarity matrix between test and training data. Extensive experiments have been conducted to evaluate the performance of these methods on several real-world (dis)similarity matrices with semantic meanings. Gang Wu 0005, Edward Y. Chang, Zhihua Zhang 0004 |
ACM Multimedia | 3 |
| 2005 | Kronecker Factorization for Speeding up Kernel MachinesabstractIn kernel machines, such as kernel principal component analysis (KPCA), Gaussian Processes (GPs), and Support Vector Machines (SVMs), the computational complexity of finding a solution is O(n3), where n is the number of training instances. To reduce this expensive computational complexity, we propose using Kronecker factorization, which approximates a positive definite kernel matrix by the Kronecker product of two smaller positive definite matrices. This approximation can speed up the calculation of the kernel-matrix inverse or eigen-decomposition involved in kernel machines. When the two factorized matrices have about the same dimensions, the computational complexity is improved from O(n3) to O(n2). We propose two methods to carry out Kronecker factorization and apply them to speed up KPCA. In Experiments show that our methods can drastically reduce the computation time of kernel machines without any significant degradation in their effectiveness. Gang Wu 0005, Zhihua Zhang 0004, Edward Y. Chang |
SDM | 2 |
| 2004 | Bayesian Inference on Principal Component Analysis Using Reversible Jump Markov Chain Monte Carlo
Zhihua Zhang 0004, Kap Luk Chan, James T. Kwok, Dit-Yan Yeung |
AAAI | 1 |
| 2004 | Surrogate maximization/minimization algorithms for AdaBoost and the logistic regression modelabstractSurrogate maximization (or minimization) (SM) algorithms are a family of algorithm that can be regarded as a generalization of expectation-maximization (EM) algorithms. There are three major approaches to the construction of surrogate function, all relying on the convexity of some function. In this paper, we solve the boosting problem by proposing SM algorithms for the corresponding optimization problem. Specifically, for AdaBoost, we derive an SM algorithm that can be shown to be identical to the algorithm proposed by Collins et al. (2002) based on Bregman distance. More importantly, for LogitBoost (or logistic boosting), we use several methods to construct different surrogate functions which result in different SM algorithms. By combining multiple methods, we are able to derive an SM algorithm that is also the same as an algorithm derived by Collins et al. (2002). Our approach based on SM algorithms is much simpler and convergence results follow naturally. Zhihua Zhang 0004, James T. Kwok, Dit-Yan Yeung |
ICML | 1 |
| 2004 | Bayesian inference for transductive learning of kernel matrix using the Tanner-Wong data augmentation algorithmabstractIn kernel methods, an interesting recent development seeks to learn a good kernel from empirical data automatically. In this paper, by regarding the transductive learning of the kernel matrix as a missing data problem, we propose a Bayesian hierarchical model for the problem and devise the Tanner-Wong data augmentation algorithm for making inference on the model. The Tanner-Wong algorithm is closely related to Gibbs sampling, and it also bears a strong resemblance to the expectation-maximization (EM) algorithm. For an efficient implementation, we propose a simplified Bayesian hierarchical model and the corresponding Tanner-Wong algorithm. We express the relationship between the kernel on the input space and the kernel on the output space as a symmetric-definite generalized eigenproblem. Based on this eigenproblem, an efficient approach to choosing the base kernel matrices is presented. The effectiveness of our Bayesian model with the Tanner-Wong algorithm is demonstrated through some classification experiments showing promising results. Zhihua Zhang 0004, Dit-Yan Yeung, James T. Kwok |
ICML | 1 |
| 2003 | Parametric Distance Metric Learning with Label Information
Zhihua Zhang 0004, James T. Kwok, Dit-Yan Yeung |
IJCAI | 1 |