arXiv reaDer
スライド全体の画像分類のための多重検出ベースの複数インスタンス学習ネットワーク
Multiplex-detection Based Multiple Instance Learning Network for Whole Slide Image Classification
複数インスタンス学習 (MIL) は、病理診断のためにスライド全体の画像 (WSI) を分類するための強力なアプローチです。 WSI 分類に関する MIL の基本的な課題は、バッグ ラベルをトリガーする重要なインスタンスを発見することです。ただし、以前の方法は、主に独立した同一の分布仮説 (iid) の下で設計されており、インスタンス間の相関関係または腫瘍の不均一性を無視しています。この論文では、上記の問題に取り組むために、新しい多重検出ベースの複数インスタンス学習 (MDMIL) を提案します。具体的には、MDMIL は、内部クエリ生成モジュール (IQGM) と多重検出モジュール (MDM) によって構築され、トレーニング中のメモリベースのコントラスト損失によって支援されます。まず、IQGM はインスタンスの確率を与え、分布分析後に信頼性の高い特徴を集約することにより、次の MDM の内部クエリ (IQ) を生成します。次に、MDM のマルチプレックス検出クロスアテンション (MDCA) とマルチヘッドセルフアテンション (MHSA) が協力して、WSI の最終的な表現を生成します。このプロセスでは、IQ とトレーニング可能な変分クエリ (VQ) がインスタンス間の接続を正常に構築し、異種腫瘍に対するモデルの堅牢性を大幅に向上させます。最後に、特徴空間の制約をさらに強化し、トレーニング プロセスを安定させるために、メモリ ベースのコントラスト ロスを採用します。 CAMELYON16、TCGA-NSCLC、および TCGA-RCC データセットなど、3 つの計算病理学データセットで実験を行います。優れた精度と AUC は、提案された MDMIL が他の最先端の方法よりも優れていることを示しています。
Multiple instance learning (MIL) is a powerful approach to classify whole slide images (WSIs) for diagnostic pathology. A fundamental challenge of MIL on WSI classification is to discover the critical instances that trigger the bag label. However, previous methods are primarily designed under the independent and identical distribution hypothesis (i.i.d), ignoring either the correlations between instances or heterogeneity of tumours. In this paper, we propose a novel multiplex-detection-based multiple instance learning (MDMIL) to tackle the issues above. Specifically, MDMIL is constructed by the internal query generation module (IQGM) and the multiplex detection module (MDM) and assisted by the memory-based contrastive loss during training. Firstly, IQGM gives the probability of instances and generates the internal query (IQ) for the subsequent MDM by aggregating highly reliable features after the distribution analysis. Secondly, the multiplex-detection cross-attention (MDCA) and multi-head self-attention (MHSA) in MDM cooperate to generate the final representations for the WSI. In this process, the IQ and trainable variational query (VQ) successfully build up the connections between instances and significantly improve the model's robustness toward heterogeneous tumours. At last, to further enforce constraints in the feature space and stabilize the training process, we adopt a memory-based contrastive loss, which is practicable for WSI classification even with a single sample as input in each iteration. We conduct experiments on three computational pathology datasets, e.g., CAMELYON16, TCGA-NSCLC, and TCGA-RCC datasets. The superior accuracy and AUC demonstrate the superiority of our proposed MDMIL over other state-of-the-art methods.
updated: Wed Aug 31 2022 12:19:25 GMT+0000 (UTC)
published: Sat Aug 06 2022 14:36:48 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アソシエイト