arXiv:2508.14588cs.CV2025-08ICCV被引 2

提出可控制的病理图像增强方法,提升小样本下的诊断模型性能。

Controllable Latent Space Augmentation for Digital Pathology

  • 在隐空间生成可控的病理图像增强,支持语义明确的变换。
  • 在多器官低数据场景下显著提升MIL模型准确率,最高增益12.3%。
  • 适合病理图像分析、小样本学习的研究者和医疗AI开发者。

数字病理学中的全切片图像(WSI)分析面临分辨率高达吉像素及密集监督信号稀缺的挑战。尽管多实例学习(MIL)适用于切片级任务,但训练鲁棒模型需要大规模且多样化的数据集。传统基于补丁级别的图像增强因需处理海量补丁而成本高昂,现有特征级增强方法则难以控制变换语义。本文提出HistAug,一种高效生成式模型,可在隐空间实现可控的病理图像增强。通过显式条件化于补丁级变换(如色调、腐蚀),HistAug生成保持原始语义信息的真实增强嵌入。该方法可在单次前向传播中处理大量补丁,显著提升MIL模型性能。在多个切片级任务与不同器官的数据上验证,其在低数据条件下优于现有方法,尤其在数据稀缺时表现更优。消融实验表明,学习到的变换优于噪声扰动,且切片级一致增强至关重要。代码已公开于https://github.com/MICS-Lab/HistAug。

原文摘要 · Abstract (English)

Whole slide image (WSI) analysis in digital pathology presents unique challenges due to the gigapixel resolution of WSIs and the scarcity of dense supervision signals. While Multiple Instance Learning (MIL) is a natural fit for slide-level tasks, training robust models requires large and diverse datasets. Even though image augmentation techniques could be utilized to increase data variability and reduce overfitting, implementing them effectively is not a trivial task. Traditional patch-level augmentation is prohibitively expensive due to the large number of patches extracted from each WSI, and existing feature-level augmentation methods lack control over transformation semantics. We introduce HistAug, a fast and efficient generative model for controllable augmentations in the latent space for digital pathology. By conditioning on explicit patch-level transformations (e.g., hue, erosion), HistAug generates realistic augmented embeddings while preserving initial semantic information. Our method allows the processing of a large number of patches in a single forward pass efficiently, while at the same time consistently improving MIL model performance. Experiments across multiple slide-level tasks and diverse organs show that HistAug outperforms existing methods, particularly in low-data regimes. Ablation studies confirm the benefits of learned transformations over noise-based perturbations and highlight the importance of uniform WSI-wise augmentation. Code is available at https://github.com/MICS-Lab/HistAug.

数字病理图像增强小样本学习隐空间

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