arXiv:2606.22002cs.CVcs.LG2026-06中稿 · MICCAI 2026

用图覆盖方法一次性选医疗图像样本,省去训练和标注成本。

One-Shot Data Selection for Medical Image Classification via Graph Coverage

论文配图:One-Shot Data Selection for Medical Image Classification via Graph Coverage
图 1 · 摘自论文原文
  • 基于预训练模型嵌入构建近邻图,用双项覆盖核捕捉局部结构。
  • 在五个数据集上九次达到最高准确率,尤其在类别不平衡时优势明显。
  • 无需重新训练模型,适合标注预算有限的医学图像分类场景。

在标注预算受限时,对完整数据集训练医学图像分类器是低效的:并非所有样本都同等重要,而专家标注成本高昂。主动学习虽能降低标注成本,但依赖反复查询并需多轮模型训练。一次性几何方法(如设施选址)避免重训练,但仅基于成对距离,忽略数据流形的局部结构。我们提出一种基于图的单次选择方法,完全在冻结的基础模型嵌入上操作。给定预训练编码器的嵌入,我们在所有训练样本上构建k近邻图,并从热扩散核推导出两项覆盖核,同时捕捉直接和两跳邻域关系。在该核上进行贪心设施选址,选出类平衡的子集以最大化数据流形覆盖。该双项核在选择行为上与全谱热核一致,却将计算降至稀疏矩阵操作,仅含一个超参数。我们在五个涵盖组织病理学、放射学和显微镜学的MedMNIST数据集上评估,对比了基于训练动态和几何的基线方法。我们的方法在十组数据集-比例条件下有九次达到最高平衡准确率,尤其在类别不平衡数据集上提升显著,因全局图结构捕获了跨类关联,而逐类方法无法察觉。整个选择过程无需任何模型训练。代码已公开于 https://github.com/zahiriddin-rustamov/graph-coverage-selection。

原文摘要 · Abstract (English)

Training medical image classifiers on entire datasets is wasteful when annotation budgets are limited: not all samples contribute equally, yet acquiring expert labels is expensive. Active learning reduces annotation cost through iterative querying, but assumes repeated access to an oracle and requires multiple rounds of model training. One-shot geometry-based methods such as facility location avoid retraining but operate on pairwise distances that ignore the local structure of the data manifold. We propose a graph-based one-shot selection method that operates entirely on frozen foundation model embeddings. Given embeddings from a pretrained encoder, we construct a k-nearest neighbor graph over all training samples and derive a two-term coverage kernel from the heat diffusion kernel, capturing both direct and two-hop neighborhood relationships. Greedy facility location on this kernel selects class-balanced subsets that maximize coverage of the data manifold. The two-term kernel matches the full spectral heat kernel in selection behavior while reducing computation to sparse matrix operations with a single hyperparameter. We evaluate on five MedMNIST datasets spanning histopathology, radiology, and microscopy, comparing against both training-dynamics and geometry-based baselines. Our method achieves the highest balanced accuracy on nine of ten dataset-ratio conditions, with the largest gains on class-imbalanced datasets where global graph construction captures cross-class structure that per-class methods miss, all without any model training during selection. Code is available at https://github.com/zahiriddin-rustamov/graph-coverage-selection.

医学图像数据选择图神经网络零样本

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