λ-Regularized A-Optimal Design and its Approximation by λ-Regularized Proportional Volume Sampling
In this work, we study the λ-regularized A-optimal design problem and introduce the λ-regularized proportional volume sampling algorithm, generalized from [Nikolov, Singh, and Tantipongpipat, 2019], for this problem with the approximation guarantee that extends upon the previous work. In this problem, we are given vectors v_1,…,v_n∈ℝ^d in d dimensions, a budget k≤ n, and the regularizer parameter λ≥0, and the goal is to find a subset S⊆ [n] of size k that minimizes the trace of (∑_i∈ Sv_iv_i^⊤ + λ I_d)^-1 where I_d is the d× d identity matrix. The problem is motivated from optimal design in ridge regression, where one tries to minimize the expected squared error of the ridge regression predictor from the true coefficient in the underlying linear model. We introduce λ-regularized proportional volume sampling and give its polynomial-time implementation to solve this problem. We show its (1+ϵ/√(1+λ'))-approximation for k=Ω(d/ϵ+log 1/ϵ/ϵ^2) where λ' is proportional to λ, extending the previous bound in [Nikolov, Singh, and Tantipongpipat, 2019] to the case λ>0 and obtaining asymptotic optimality as λ→∞.
READ FULL TEXT