cs.LGJul 12, 2026

Bandit PCA with Minimax Optimal Regret

Authors: Moïse BlanchardDmitrii OstrovskiiAadirupa Saha

Organizations: Georgia Tech · University of Illinois, Chicago

Abstract

We study the bandit-feedback version of online principal component analysis (Bandit PCA): in each round t=1,,Tt = 1,\dots,T, the adversary selects a d×dd \times d symmetric gain matrix GtG_t with spectrum in [0,1][0,1] and rank at most rr; the learner simultaneously selects a unit vector wtSd1w_t \in S^{d-1} and receives the reward wtGtwtw_t^\top G_t w_t. The learner receives no other feedback, and aims to minimize the regret against the best unit vector in hindsight. This problem was introduced by Kotlowski and Neu (2019), who gave an algorithm with regret O(drTlogT)O(d\sqrt{rT \log T}) and showed the lower bound of Ω(rT/logT)Ω(r\sqrt{T/\log T}). We improve upon both of these bounds and essentially bridge the gap between them, establishing the minimax regret of order rdTr\sqrt{dT} up to polylogarithmic factors in dd and TT. The upper bound is attained by a novel algorithm, which combines online mirror descent on the spectrahedron of (real) density matrices with a multiscale exploration scheme in which the eigenspaces with different spectral magnitudes are updated at different rates. For the lower bound, we construct an adaptive adversary that refines a hidden large-reward subspace based on the learner's actions, in such a way that low regret is impossible without estimating the subspace; as a result, lower-bounding the regret reduces to studying the arising subspace estimation problem. Finally, we discuss connections of Bandit PCA with adaptive-measurement quantum tomography.

Explore similar work

May 27, 2026quant-ph

Quantum principal component analysis without eigenvector recovery

Principal component analysis (PCA) is traditionally implemented through a covariance or kernel matrix, leading-eigenvector extraction, and hard rank-kk projection. These steps can be computationally costly in high-dimensional and quantum-data settings, sensitive to small eigengaps, and unnecessary when downstream tasks only require principal-subspace scores. Such score-based objectives are important in applications such as anomaly detection, spectral-energy profiling, and other postselection tasks. To address these needs, we introduce a measurement-based soft PCA framework replacing the hard top-kk projector with an entropy-regularized Fermi--Dirac filter. This filter is the unique optimizer of an entropy-regularized variational formulation of PCA and converges to the classical PCA projector in the zero-temperature limit. This filter has a direct interpretation as a quantum measurement, which naturally suggests a quantum approach. For centered covariance operators represented by quantum feature states, a single fixed circuit, together with threshold calibration, accesses all optimal filters for different rank budgets or retained-variance levels without rank-dependent circuit updates or eigenvector recovery. For new inputs, the same calibrated quantum circuit yields soft principal subspace scores, spectral energy profiles, and postselected filtered states. The required centering of both training and test data is performed coherently inside the quantum protocol, which is particularly important for quantum data where no classical feature vectors or centered Gram matrix are directly available. By reframing PCA as a calibrated measurement task, this framework bypasses the need for iterative eigenvector extraction and achieves a dimension-independent sample complexity O(η2)O(η^{-2}) for normalized fractional-rank or retained variance scoring at additive accuracy ηη.
Yewei Yuan, Michele Minervini, Mark M. Wilde +1
Jun 2, 2026stat.ML

A Robust Optimization Approach to Sparse Principal Component Analysis

While principal component analysis (PCA) is a fundamental tool for dimensionality reduction, its dense representations make it ill-suited for high-dimensional data. Existing methods address this by promoting sparsity through explicit 1\ell_1-penalties, but these are not obvious to tune due to the unsupervised nature of the task. In contrast, we propose Adversarial PCA (AdvPCA), which leverages robust optimization to achieve sparsity by optimizing the reconstruction objective against bounded, worst-case latent space perturbations. We show that this formulation admits a closed-form reduction, leading to a practical iterative algorithm that alternates between adversarial linear regression-style updates for the sparse encoder and orthogonal updates for the decoder. By theoretically characterizing the solution, we derive a data-adaptive parameterization that allows the algorithm to perform effectively out of the box. We validate these claims through numerical experiments on synthetic and real-world genomics data.
David Vävinggren, Francis Bach, André M. H. Teixeira +2
May 18, 2026cs.LG

Catching a Moving Subspace: Low-Rank Bandits Beyond Stationarity

Many bandit deployments (recommendation, clinical dosing, ad targeting) share two facts prior work handles only in isolation: rewards live on a low-dimensional latent subspace, and that subspace drifts. Stationary low-rank bandits exploit rank but break under subspace change; non-stationary linear bandits adapt to drift but pay ambient rate O~(dT)\widetilde{O}(d\sqrt{T}). We study piecewise-stationary low-rank linear contextual bandits with scalar feedback: θt=Bkwtθ_t = B_k^\star w_t with rank-rr factor BkRd×rB_k^\star\in\mathbb{R}^{d\times r} constant within each of KK unknown segments and able to shift at boundaries. Our results are tight along three axes. (i) Identification boundary. With single-play scalar rewards, the moving subspace is recoverable through quadratic functionals of rewards iff three probe-side conditions hold: known noise variance, bounded state-noise coupling, and full-dimensional probe support. Each is necessary in the unrestricted-second-moment problem, and jointly they are sufficient, characterizing the boundary of the solvable region. (ii) Algorithm and dynamic regret. SPSC interleaves isotropic probes with windowed projected ridge-UCB exploitation inside the learned rr-dimensional subspace; a CUSUM-style variant discovers segment boundaries online. The costed dynamic regret is O~(rT)+O~(T2/3)+O(WVin)\widetilde{O}(r\sqrt{T})+\widetilde{O}(T^{2/3})+O(W\,V_{\mathrm{in}}), replacing the ambient dTd\sqrt{T} rate with the intrinsic rank. (iii) Empirics. On eleven benchmarks spanning synthetic, UCI/MovieLens, semi-synthetic clinical, and ZOZOTOWN production-log data, SPSC outperforms non-stationary and low-rank baselines whenever drT1/6d-r\gtrsim T^{1/6}, matching the analytical crossover. To our knowledge, this is the first work to characterize the identification boundary and attain the intrinsic-rank dynamic-regret rate in this setting.
Hamed Khosravi, Xiaoming Huo