arXiv:2608.26309cs.LG2026-08

用代数多重网格加速标签传播,大幅降低大规模图像数据的计算开销。

Algebraic Multigrid Acceleration for Efficient Label Spreading

论文配图:Algebraic Multigrid Acceleration for Efficient Label Spreading
图 1 · 摘自论文原文
  • 基于代数多重网格构建快速邻域图并替代传统随机游走迭代
  • 单次多级循环即可完成任意规模图上的标签传播,效率提升显著
  • 对超参数不敏感,适合小样本下的高精度标签生成

现代机器学习模型依赖大量标注数据,但大规模数据集的人工标注成本高且耗时。标签传播是一种半监督学习方法,通过少量标注样本向大量无标签数据传播信息来缓解这一问题。然而,其在大规模、高维数据上的应用受限于计算成本和内存约束。为此,我们提出代数多重网格加速的高效标签传播框架(AMELS),通过快速构建邻域图并引入代数多重网格求解器,取代传统随机游走迭代。由于代数多重网格的多级结构,AMELS可在一次多级循环内完成任意规模图上的标签传播。实验表明,相比现有实现,AMELS显著减少运行时间,且在运行时间和分类精度上对超参数更鲁棒。该框架使大规模图像数据集上的高效标签传播成为可能,即使仅有少量标注样本也能生成准确标签。

原文摘要 · Abstract (English)

Modern machine learning models rely on large amounts of labeled data. However, manual annotation of large-scale datasets is expensive and time-consuming. Label spreading is a semi-supervised learning technique that addresses this challenge by propagating information from a few labeled examples to a larger pool of unlabeled data. Despite its effectiveness, its application to large-scale, high-dimensional datasets is limited by computational costs and memory constraints. To address these limitations, we propose Algebraic Multigrid Acceleration for Efficient Label Spreading (AMELS), an efficient label spreading framework that improves scalability by fast construction of neighborhood graphs and the incorporation of algebraic multigrid solvers. The latter is an iterative solver that replaces the ordinary random walk iteration typically performed in label spreading. Due to the multilevel nature of algebraic multigrid solvers, AMELS spreads given label information across a graph of any size in a single multigrid cycle. We demonstrate that AMELS achieves significant runtime reductions compared to existing implementations while also being more robust to hyperparameter choices in terms of both runtime and classification accuracy. Our framework therefore enables efficient label spreading on large-scale image datasets and produces accurate labels even when only a few labeled samples are available.

标签传播多尺度计算半监督学习

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