arXiv:2606.16045cs.LGcs.DS2026-06

用低秩结构选数据,比传统聚类更高效准确。

Active Learning with Low-Rank Structure for Data Selection

论文配图:Active Learning with Low-Rank Structure for Data Selection
图 1 · 摘自论文原文
  • 基于低秩近似与残差采样,从嵌入数据中选代表性样本。
  • 理论上只需约 $\tilde{O}(k + \frac{1}{\varepsilon^2})$ 个点,误差可控。
  • 适合高维数据、具全局结构的场景,如图像或文本嵌入。

在数据选择问题中,目标是从数据集中挑选一个小型且具有代表性的子集,以高效训练机器学习模型。已有研究表明,在数据嵌入满足几何假设的前提下,基于 $k$-center 聚类的启发式方法可有效实现数据选择。近期工作进一步提出基于 $k$-means 和敏感性采样的策略。然而,这些方法依赖于数据内在几何结构,而现代许多数据集则表现出更适合低秩近似或主成分分析的全局代数结构。本文提出一种新框架,结合低秩近似与残差采样,从行子集选择和损失保持共核构造的角度建模。给定满足弱正则性条件的数据嵌入(可理解为代数或角度意义上的Lipschitz连续性),我们证明可选取一个加权子集,其大小为 $\tilde{O}(k + \frac{1}{\varepsilon^2})$,使得其平均损失相对于全集平均损失的相对误差在 $(1+\varepsilon)$ 以内,外加一个 $\varepsilon Φ_k$ 的附加项,其中 $Φ_k$ 为嵌入矩阵最优秩-$k$ 近似代价。我们在多个真实数据集上进行了实证评估,结果表明该方法优于基于均匀采样或聚类敏感性采样的现有策略。

原文摘要 · Abstract (English)

In the data selection problem, the objective is to choose a small, representative subset of data that can be used to efficiently train a machine learning model. Sener and Savarese [ICLR 2018] showed that, given an embedding representation of the data and suitable geometric assumptions, heuristics based on $k$-center clustering can be used to perform data selection. This perspective was further explored by Axiotis et. al. [ICML 2024], who proposed a data selection approach based on $k$-means clustering and sensitivity sampling. However, these methods rely on the assumption that the dataset exhibits intrinsic geometric structure that can be effectively captured by clustering, whereas many modern datasets instead possess global algebraic structure that is better exploited by low-rank approximation or principal component analysis. In this paper, we introduce a new data selection framework based on low-rank approximation and residual-based sampling, formulated through the lens of row subset selection and loss-preserving coreset construction. Given an embedding representation of the data satisfying mild regularity conditions, which can be interpreted as algebraic or angular notions of Lipschitz continuity, we show that it is possible to select a weighted subset of $\tilde{O}\left(k + \frac{1}{\varepsilon^2}\right)$ data points whose average loss approximates the average loss over the full dataset within a $(1+\varepsilon)$ relative error, up to an additive $\varepsilon Φ_k$ term, where $Φ_k$ denotes the optimal rank-$k$ approximation cost of the embedding matrix. We complement these theoretical guarantees with empirical evaluations, demonstrating that on a range of real-world datasets, our data selection approach achieves improved performance over prior strategies based on uniform sampling or clustering-based sensitivity sampling.

数据选择低秩近似主动学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。