arXiv reaDer
TransMIL:スライド全体の画像分類のためのTransformerベースの相関マルチインスタンス学習
TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classification
マルチインスタンス学習(MIL)は、全スライド画像(WSI)ベースの病理診断における弱く教師あり分類を解決するための強力なツールです。ただし、現在のMILメソッドは通常、独立した同一の分布仮説に基づいているため、異なるインスタンス間の相関を無視します。この問題に対処するために、相関MILと呼ばれる新しいフレームワークを提案し、収束の証拠を提供しました。このフレームワークに基づいて、形態学的情報と空間情報の両方を調査するTransformerベースのMIL(TransMIL)を考案しました。提案されたTransMILは、優れた視覚化と解釈可能性を備えた、不均衡/均衡およびバイナリ/複数の分類を効果的に処理できます。 3つの異なる計算病理学問題についてさまざまな実験を実施し、最先端の方法と比較して、より優れたパフォーマンスとより高速な収束を実現しました。バイナリ腫瘍分類のテストAUCは、CAMELYON16データセットに対して最大93.09%になる可能性があります。また、がんサブタイプ分類に対するAUCは、TCGA-NSCLCデータセットおよびTCGA-RCCデータセットに対してそれぞれ最大96.03%および98.82%になる可能性があります。実装はhttps://github.com/szc19990412/TransMILで入手できます。
Multiple instance learning (MIL) is a powerful tool to solve the weakly supervised classification in whole slide image (WSI) based pathology diagnosis. However, the current MIL methods are usually based on independent and identical distribution hypothesis, thus neglect the correlation among different instances. To address this problem, we proposed a new framework, called correlated MIL, and provided a proof for convergence. Based on this framework, we devised a Transformer based MIL (TransMIL), which explored both morphological and spatial information. The proposed TransMIL can effectively deal with unbalanced/balanced and binary/multiple classification with great visualization and interpretability. We conducted various experiments for three different computational pathology problems and achieved better performance and faster convergence compared with state-of-the-art methods. The test AUC for the binary tumor classification can be up to 93.09% over CAMELYON16 dataset. And the AUC over the cancer subtypes classification can be up to 96.03% and 98.82% over TCGA-NSCLC dataset and TCGA-RCC dataset, respectively. Implementation is available at: https://github.com/szc19990412/TransMIL.
updated: Sun Oct 31 2021 14:08:35 GMT+0000 (UTC)
published: Wed Jun 02 2021 02:57:54 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アソシエイト