Element-wise Core-Sets
- Element-wise core-sets are a deterministic subset selection method for least squares estimation that targets high-magnitude entries in each column, ensuring unbiased and efficient results.
- The approach minimizes variance by focusing on the most informative data, providing concrete theoretical guarantees through spectral approximation and variance control.
- Enhanced with a median-of-means extension, the method robustly handles outliers while offering significant runtime savings compared to traditional row-based sampling techniques.
Element-wise core-sets (core-elements) constitute a deterministic subset selection methodology for large-scale least squares estimation. Unlike conventional coreset methods that subsample rows of the design matrix, the core-elements approach targets individual entries with highest magnitude in each column. This element-wise paradigm is especially effective for sparse or numerically sparse predictor matrices by allocating the sampling budget to the most informative data, yielding computationally efficient and statistically robust estimators with concrete theoretical guarantees (Li et al., 2022).
1. Definition and Construction of Core-Elements
Consider the linear model
with as the design matrix, the regression parameter, and i.i.d. noise. Element-wise core-sets select entries per column, constructing an indicator matrix with ones in each column: $s_{ij} = \begin{cases} 1, & i\in\{\text{indices of %%%%6%%%% largest } |x_{ij}|\}, \ 0, & \text{otherwise}. \end{cases}$ The sparsified matrix is (Hadamard product), with exactly retained entries. This method exploits sparsity by ensuring the sampling budget targets high-information content rather than allocating uniformly or by row.
2. Estimation Method and Unbiasedness
Given 0, the core-elements estimator is
1
Unbiasedness holds when 2 is invertible: 3 This property marks a distinction from stochastic row sampling, as core-elements deterministically preserves—columnwise—the leading signal, subject to selection by magnitude.
3. Variance Bounds and Statistical Guarantees
The conditional variance is
4
Control of variance is achieved by selecting 5 largest-magnitude entries, which approximately minimizes key matrix norms. For 6 and spectral radius 7, the upper bound is
8
Thus, variance is suppressed by minimizing 9 and ensuring 0 is small, both favored by magnitude-based selection.
A finite-sample, coreset-style approximation is guaranteed. If 1 approximates 2 in operator norm: 3 for 4, then
5
This shows near-optimality of the estimator under spectral approximation (Li et al., 2022).
4. Algorithmic Implementation and Computational Complexity
The deterministic construction involves:
- Initializing 6.
- For each column 7, identify indices of 8 largest 9 and set corresponding 0.
- Form 1.
- Solve for 2 as above.
Computational costs:
- Column selection: 3 using selection algorithms (e.g., partial sorting).
- Computation of 4: 5.
- Overall (for 6): 7, substantially faster than the 8 cost of OLS or leverage-score methods.
5. Robust Extension via Median-of-Means
To address outlier contamination, median-of-means (MOM) is integrated:
- Partition 9 into 0 random, equal-sized blocks.
- On block 1, apply core-elements with 2 budget to obtain 3.
- Aggregate by coordinate-wise median: 4.
Under bounded per-block condition and few block corruptions, 5. The breakdown point is 6. This robustification ensures both statistical consistency and high breakdown under data contamination (Li et al., 2022).
6. Comparative Evaluation and Empirical Performance
Empirical studies benchmarked CORE and MOM-CORE against uniform subsampling, doubly-sketched, leverage methods (BLEV/SLEV), IBOSS, OSS, D-optimal (DOPT), and full OLS on synthetic (Normal, Log-Normal, 7) and real datasets, under varying sparsity (8). Metrics included normalized estimation error (MSE) and prediction error (PMSE). Key findings:
- CORE obtains the lowest MSE/PMSE across all distributions and sparsity levels, with an advantage accentuated for sparse 9.
- MOM-CORE shows maximal robustness to outlier schemes.
- Runtime for CORE is marginally above UNIF/DOUBLY but dramatically below other deterministic methods, scaling as 0.
A real-world illustration, using single-cell RNA-seq (1, 2, 3 zeros), demonstrates near-full-OLS accuracy with substantial runtime savings.
| Method | Statistical Efficacy (MSE/PMSE) | CPU Time Scaling |
|---|---|---|
| CORE | Lowest across settings | 4 |
| MOM-CORE | Highest robustness | Slightly above CORE |
| BLEV/SLEV | Higher error | 5 |
| UNIF/DOUBLY | Fast, higher error | Fastest |
7. Characteristics, Applications, and Future Directions
Element-wise core-sets are motivated by the inefficiency of row-based subsampling under high sparsity: many rows are mostly zero, leading to wasted sample budget. By contrast, element-wise selection extracts information where it is densest.
Advantages:
- Unbiasedness and near-optimal variance bounds.
- Deterministic selection, eliminating sampling variability.
- Runtime linear in 6 plus 7.
- Seamless integration with MOM for robust, efficient estimation.
Applications naturally include large-scale regressions, especially when the data is sparse or contaminated. Anticipated research directions include nonparametric or penalized spline extensions, privacy-preserving adaptations for federated settings, handling other loss functions (e.g., GLMs, quantile regression), and adaptation to streaming scenarios (Li et al., 2022).