病理組織学的画像解析は、がんなどの疾患の発見に不可欠なプロセスです。ただし、使用可能なメモリ容量を考慮すると、ギガピクセル解像度のスライドイメージ(WSI)全体でCNNをトレーニングすることは困難です。以前の作品のほとんどは、高解像度のWSIを小さな画像パッチに分割し、それらを個別にモデルに入力して、それを腫瘍または正常組織として分類します。ただし、パッチベースの分類では、パッチスケールのローカル情報のみが使用されますが、隣接するパッチ間の関係は無視されます。隣接するパッチとグローバルフィーチャの関係を考慮すると、分類パフォーマンスを改善できます。本論文では、自動病理診断の予測性能を向上させるために、パッチベースの分類モデルとスライドスケールセグメンテーションモデル全体を組み合わせた新しいモデル構造を提案します。分類モデルからパッチの特徴を抽出し、セグメンテーションモデルに入力して、スライド腫瘍確率ヒートマップ全体を取得します。分類モデルはパッチスケールのローカルフィーチャを考慮し、セグメンテーションモデルはグローバル情報を考慮することができます。また、勾配情報を保持し、GPUメモリ容量が制限されたエンドツーエンド学習用にモデルを部分的にトレーニングする新しい最適化方法を提案します。我々の方法をWSIの腫瘍/正常予測に適用し、分類性能は従来のパッチベースの方法と比較して改善されています。
Histopathological image analysis is an essential process for the discovery of diseases such as cancer. However, it is challenging to train CNN on whole slide images (WSIs) of gigapixel resolution considering the available memory capacity. Most of the previous works divide high resolution WSIs into small image patches and separately input them into the model to classify it as a tumor or a normal tissue. However, patch-based classification uses only patch-scale local information but ignores the relationship between neighboring patches. If we consider the relationship of neighboring patches and global features, we can improve the classification performance. In this paper, we propose a new model structure combining the patch-based classification model and whole slide-scale segmentation model in order to improve the prediction performance of automatic pathological diagnosis. We extract patch features from the classification model and input them into the segmentation model to obtain a whole slide tumor probability heatmap. The classification model considers patch-scale local features, and the segmentation model can take global information into account. We also propose a new optimization method that retains gradient information and trains the model partially for end-to-end learning with limited GPU memory capacity. We apply our method to the tumor/normal prediction on WSIs and the classification performance is improved compared with the conventional patch-based method.