arXiv reaDer
BEL: A Bag Embedding Loss for Transformer は、複数インスタンス全体のスライド画像分類を強化します
BEL: A Bag Embedding Loss for Transformer enhances Multiple Instance Whole Slide Image Classification
複数インスタンス学習 (MIL) は、ギガピクセル組織病理学全体スライド画像 (WSI) の分類タスクの主要なアプローチになっています。 MIL フレームワーク内では、単一の WSI (バッグ) がパッチ (インスタンス) に分解され、WSI レベルの注釈のみが使用可能になります。最近の MIL アプローチは、インスタンス間の依存関係をモデル化するトランスフォーマー アーキテクチャの機能を利用して、非常に有益なバッグ レベル表現を生成します。ただし、高倍率のデータセットに適用すると、多数のインスタンスと弱い監視学習信号のために問題が発生します。この問題に対処するために、新しいバッグ埋め込み損失 (BEL) を使用して変圧器を追加でトレーニングすることを提案します。 BEL は、同じクラスのバッグ埋め込み間の距離を最小化し、異なるクラス間の距離を最大化することにより、モデルに差別的なバッグレベル表現を学習させます。公開されている 2 つの組織病理学データセット、BRACS と CAMELYON17 で Transformer アーキテクチャ TransMIL を使用して BEL を評価します。 BEL を使用すると、TransMIL が両方のデータセットでベースライン モデルよりも優れていることが示され、組織学的患者材料の臨床的に関連性の高い AI ベースの腫瘍分類に貢献しています。
Multiple Instance Learning (MIL) has become the predominant approach for classification tasks on gigapixel histopathology whole slide images (WSIs). Within the MIL framework, single WSIs (bags) are decomposed into patches (instances), with only WSI-level annotation available. Recent MIL approaches produce highly informative bag level representations by utilizing the transformer architecture's ability to model the dependencies between instances. However, when applied to high magnification datasets, problems emerge due to the large number of instances and the weak supervisory learning signal. To address this problem, we propose to additionally train transformers with a novel Bag Embedding Loss (BEL). BEL forces the model to learn a discriminative bag-level representation by minimizing the distance between bag embeddings of the same class and maximizing the distance between different classes. We evaluate BEL with the Transformer architecture TransMIL on two publicly available histopathology datasets, BRACS and CAMELYON17. We show that with BEL, TransMIL outperforms the baseline models on both datasets, thus contributing to the clinically highly relevant AI-based tumor classification of histological patient material.
updated: Thu Mar 02 2023 16:02:55 GMT+0000 (UTC)
published: Thu Mar 02 2023 16:02:55 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト