arXiv reaDer
RoS-KD: ノイズの多い医用画像処理のためのロバストな確率的知識抽出アプローチ
RoS-KD: A Robust Stochastic Knowledge Distillation Approach for Noisy Medical Imaging
AI を利用した医用画像は、ペースの速いヘルスケア診断を提供できるため、最近大きな注目を集めています。ただし、通常は、高い注釈コスト、観察者間のばらつき、ヒューマン アノテーターのエラー、およびコンピューター生成ラベルのエラーにより、高品質のデータセットが不足するという問題があります。ノイズの多いラベル付きデータセットでトレーニングされたディープ ラーニング モデルは、ノイズの種類に敏感であり、目に見えないサンプルの一般化が少なくなります。この課題に対処するために、ノイズの多い情報を学習する際の抑止力を確保するために、複数のソースからトピックを学習するという概念を模倣するRobust Stochastic Knowledge Distillation (RoS-KD) フレームワークを提案します。より具体的には、RoS-KD は、トレーニング データの重複するサブセットでトレーニングされた複数の教師から知識を抽出することにより、スムーズで十分な情報に基づいた堅牢な学生マニホールドを学習します。実世界のデータセットを使用した一般的な医療画像分類タスク (心肺疾患と病変の分類) に関する当社の広範な実験では、多くの一般的な大規模ネットワーク (ResNet-50、DenseNet-121、 MobileNet-V2) を比較的小規模なネットワークで使用できること、および敵対的攻撃 (PGD、FSGM) に対する堅牢性を備えていること。より具体的には、RoS-KD は、基礎となる学生が最近の競合知識蒸留ベースラインに対して ResNet-18 である場合、病変分類および心肺疾患分類タスクの F1 スコアでそれぞれ 2% および 4% の改善を達成します。さらに、心肺疾患分類タスクでは、RoS-KD は、AUC スコアが約 1% 増加することで、ほとんどの SOTA ベースラインよりも優れています。
AI-powered Medical Imaging has recently achieved enormous attention due to its ability to provide fast-paced healthcare diagnoses. However, it usually suffers from a lack of high-quality datasets due to high annotation cost, inter-observer variability, human annotator error, and errors in computer-generated labels. Deep learning models trained on noisy labelled datasets are sensitive to the noise type and lead to less generalization on the unseen samples. To address this challenge, we propose a Robust Stochastic Knowledge Distillation (RoS-KD) framework which mimics the notion of learning a topic from multiple sources to ensure deterrence in learning noisy information. More specifically, RoS-KD learns a smooth, well-informed, and robust student manifold by distilling knowledge from multiple teachers trained on overlapping subsets of training data. Our extensive experiments on popular medical imaging classification tasks (cardiopulmonary disease and lesion classification) using real-world datasets, show the performance benefit of RoS-KD, its ability to distill knowledge from many popular large networks (ResNet-50, DenseNet-121, MobileNet-V2) in a comparatively small network, and its robustness to adversarial attacks (PGD, FSGM). More specifically, RoS-KD achieves >2% and >4% improvement on F1-score for lesion classification and cardiopulmonary disease classification tasks, respectively, when the underlying student is ResNet-18 against recent competitive knowledge distillation baseline. Additionally, on cardiopulmonary disease classification task, RoS-KD outperforms most of the SOTA baselines by ~1% gain in AUC score.
updated: Sat Oct 15 2022 22:32:20 GMT+0000 (UTC)
published: Sat Oct 15 2022 22:32:20 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アソシエイト