弱い教師あり学習の強力な戦略である複数インスタンス学習 (MIL) は、ギガピクセルの全スライド画像 (WSI) でさまざまな予測タスクを実行できます。ただし、WSI の何万ものパッチは、通常、画像拡張に膨大な計算負荷をもたらし、MIL モデルのパフォーマンスの向上を制限します。現在、機能拡張ベースの MIL フレームワークは有望なソリューションですが、Mixup などの既存の方法では非現実的な機能が生成されることがよくあります。より効率的で実用的な拡張方法を探求するために、拡散モデル (DM) を初めて MIL に導入し、AugDiff と呼ばれる機能拡張フレームワークを提案します。具体的には、DM の世代多様性を利用して特徴量拡張の品質を向上させ、段階的生成プロパティを使用して意味情報の保持を制御します。 AugDiff のパフォーマンスを評価するために、3 つの異なるがんデータセット、2 つの異なる特徴抽出器、および 3 つの一般的な MIL アルゴリズムに対して広範な実験を行います。アブレーション研究と視覚化により、有効性がさらに検証されます。さらに、AugDiff の画像拡張よりも高品質な拡張機能と、自己教師あり学習よりも優れていることを強調します。外部データセットに対する一般化は、その幅広い用途を示しています。
Multiple Instance Learning (MIL), a powerful strategy for weakly supervised learning, is able to perform various prediction tasks on gigapixel Whole Slide Images (WSIs). However, the tens of thousands of patches in WSIs usually incur a vast computational burden for image augmentation, limiting the MIL model's improvement in performance. Currently, the feature augmentation-based MIL framework is a promising solution, while existing methods such as Mixup often produce unrealistic features. To explore a more efficient and practical augmentation method, we introduce the Diffusion Model (DM) into MIL for the first time and propose a feature augmentation framework called AugDiff. Specifically, we employ the generation diversity of DM to improve the quality of feature augmentation and the step-by-step generation property to control the retention of semantic information. We conduct extensive experiments over three distinct cancer datasets, two different feature extractors, and three prevalent MIL algorithms to evaluate the performance of AugDiff. Ablation study and visualization further verify the effectiveness. Moreover, we highlight AugDiff's higher-quality augmented feature over image augmentation and its superiority over self-supervised learning. The generalization over external datasets indicates its broader applications.