arXiv reaDer
知識の蒸留を通じて学習された小さな組織学的データセット用のビジョントランスフォーマー
Vision Transformers for Small Histological Datasets Learned through Knowledge Distillation
Computational Pathology (CPATH) システムには、診断タスクを自動化できる可能性があります。ただし、デジタル化された組織学的ガラス スライド上のアーチファクトは、スライド全体画像 (WSI) として知られ、CPATH システムの全体的なパフォーマンスを妨げる可能性があります。ビジョン トランスフォーマー (ViT) などの深層学習 (DL) モデルは、診断アルゴリズムを実行する前にアーティファクトを検出して除外する場合があります。堅牢で一般化された ViT を開発する簡単な方法は、大規模なデータセットでトレーニングすることです。残念ながら、大規模な医療データセットの取得には費用がかかり不便であるため、WSI 向けの一般化されたアーチファクト検出方法の必要性が高まっています。この論文では、気泡検出タスクにおける ViT の分類パフォーマンスを向上させるための学生と教師のレシピを紹介します。 ViT は、生徒と教師の枠組みの下でトレーニングされ、能力の高い教師モデルから既存の知識を抽出することでパフォーマンスを向上させます。当社の最高パフォーマンスの ViT では、F1 スコアと MCC がそれぞれ 0.961 と 0.911 であり、スタンドアロン トレーニングと比較して MCC が 7% 向上していることが観察されています。提案された方法は、CPATH システムにおける効率的な前処理パイプラインのためにカスタマイズされたトランスフォーマーの使用を促進するために、転移学習よりも知識の蒸留を活用するという新しい視点を提示します。
Computational Pathology (CPATH) systems have the potential to automate diagnostic tasks. However, the artifacts on the digitized histological glass slides, known as Whole Slide Images (WSIs), may hamper the overall performance of CPATH systems. Deep Learning (DL) models such as Vision Transformers (ViTs) may detect and exclude artifacts before running the diagnostic algorithm. A simple way to develop robust and generalized ViTs is to train them on massive datasets. Unfortunately, acquiring large medical datasets is expensive and inconvenient, prompting the need for a generalized artifact detection method for WSIs. In this paper, we present a student-teacher recipe to improve the classification performance of ViT for the air bubbles detection task. ViT, trained under the student-teacher framework, boosts its performance by distilling existing knowledge from the high-capacity teacher model. Our best-performing ViT yields 0.961 and 0.911 F1-score and MCC, respectively, observing a 7% gain in MCC against stand-alone training. The proposed method presents a new perspective of leveraging knowledge distillation over transfer learning to encourage the use of customized transformers for efficient preprocessing pipelines in the CPATH systems.
updated: Sat May 27 2023 05:09:03 GMT+0000 (UTC)
published: Sat May 27 2023 05:09:03 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アソシエイト