arXiv reaDer
不均衡な医用画像分類のための医療知識に基づく深層学習
Medical Knowledge-Guided Deep Learning for Imbalanced Medical Image Classification
深層学習モデルは、さまざまな画像分類タスクで目覚ましいパフォーマンスを発揮しています。ただし、多くのモデルは、データのバランスが崩れていると、臨床または医療環境でのパフォーマンスが制限されます。この課題に対処するために、分類タスクのドメイン固有の知識を活用してモデルのパフォーマンスを向上させる、医療知識に基づく1クラス分類アプローチを提案します。私たちのアプローチの背後にある理論的根拠は、モデル学習を容易にするために、いくつかの既存の事前の医学的知識をデータ駆動型深層学習に組み込むことができるということです。不均衡な画像分類のための深層学習ベースの1クラス分類パイプラインを設計し、3つのユースケースで、より高い分類パフォーマンスを達成するために追加の中間クラスを生成することにより、特定の各分類タスクの医学的知識を活用する方法を示します。 3つの異なる臨床画像分類タスク(合計8459画像)でアプローチを評価し、6つの最先端の方法と比較して優れたモデルパフォーマンスを示します。この作品のすべてのコードは、論文の受理時に公開されます。
Deep learning models have gained remarkable performance on a variety of image classification tasks. However, many models suffer from limited performance in clinical or medical settings when data are imbalanced. To address this challenge, we propose a medical-knowledge-guided one-class classification approach that leverages domain-specific knowledge of classification tasks to boost the model's performance. The rationale behind our approach is that some existing prior medical knowledge can be incorporated into data-driven deep learning to facilitate model learning. We design a deep learning-based one-class classification pipeline for imbalanced image classification, and demonstrate in three use cases how we take advantage of medical knowledge of each specific classification task by generating additional middle classes to achieve higher classification performances. We evaluate our approach on three different clinical image classification tasks (a total of 8459 images) and show superior model performance when compared to six state-of-the-art methods. All codes of this work will be publicly available upon acceptance of the paper.
updated: Sat Nov 20 2021 16:14:19 GMT+0000 (UTC)
published: Sat Nov 20 2021 16:14:19 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アソシエイト