arXiv reaDer
AIはラベルなしで進化することができます:知識蒸留による胸部X線診断のための自己進化型ビジョントランスフォーマー
AI can evolve without labels: self-evolving vision transformer for chest X-ray diagnosis through knowledge distillation
深層学習ベースのコンピューター支援診断システムは最近専門家レベルのパフォーマンスを達成しましたが、堅牢な深層学習モデルを開発するには、手動注釈付きの大規模で高品質のデータが必要であり、取得に費用がかかります。この状況は、病院で毎年収集される胸部X線が、特に貧困地域で専門家による手動のラベル付けがないために使用できないという問題を引き起こします。これに対処するために、ここでは、自己教師あり学習と自己トレーニングによる知識蒸留を使用する新しい深層学習フレームワークを紹介します。これは、少数のラベルでトレーニングされた元のモデルのパフォーマンスが、ラベルのないデータを増やすことで徐々に改善できることを示しています。 。実験結果は、提案されたフレームワークが実世界の環境に対して印象的な堅牢性を維持し、結核、気胸、COVID-19などのいくつかの診断タスクに一般的に適用できることを示しています。特に、同じ量のラベル付きデータでトレーニングされたモデルよりも、モデルのパフォーマンスがさらに優れていることを示しました。提案されたフレームワークは、毎年大量のデータが蓄積される医用画像処理に大きな可能性を秘めていますが、グラウンドトゥルース注釈の取得には費用がかかります。
Although deep learning-based computer-aided diagnosis systems have recently achieved expert-level performance, developing a robust deep learning model requires large, high-quality data with manual annotation, which is expensive to obtain. This situation poses the problem that the chest x-rays collected annually in hospitals cannot be used due to the lack of manual labeling by experts, especially in deprived areas. To address this, here we present a novel deep learning framework that uses knowledge distillation through self-supervised learning and self-training, which shows that the performance of the original model trained with a small number of labels can be gradually improved with more unlabeled data. Experimental results show that the proposed framework maintains impressive robustness against a real-world environment and has general applicability to several diagnostic tasks such as tuberculosis, pneumothorax, and COVID-19. Notably, we demonstrated that our model performs even better than those trained with the same amount of labeled data. The proposed framework has a great potential for medical imaging, where plenty of data is accumulated every year, but ground truth annotations are expensive to obtain.
updated: Sun Feb 13 2022 22:40:46 GMT+0000 (UTC)
published: Sun Feb 13 2022 22:40:46 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アソシエイト