arXiv reaDer
不均衡な医用画像分類のための計量学習による識別表現の学習
Learning Discriminative Representation via Metric Learning for Imbalanced Medical Image Classification
モデルトレーニング中の一般的な疾患とまれな疾患の間のデータの不均衡により、インテリジェントな診断システムが一般的な疾患に偏った予測をすることがよくあります。最先端のアプローチでは、2段階の学習フレームワークを適用して、クラスの不均衡の問題を軽減します。最初の段階では一般的な特徴抽出器のトレーニングに焦点を当て、2番目の段階ではクラスの分類器ヘッドの微調整に焦点を当てます。リバランス。ただし、既存の2段階のアプローチでは、さまざまな疾患間のきめ細かい特性が考慮されていないため、多くの場合、最初の段階は、自然画像分類タスクよりも医用画像分類の効果が低くなります。この研究では、特徴抽出器がより識別力のある特徴表現を抽出することを学習できるように、特に2段階フレームワークの第1段階にメトリック学習を組み込むことを提案します。主に3つの医用画像データセットに関する広範な実験は、提案されたアプローチが既存の1段階および2段階のアプローチを一貫して上回っていることを示しており、メトリック学習を2段階のフレームワークの効果的なプラグインコンポーネントとして使用できることを示しています。画像分類タスク。
Data imbalance between common and rare diseases during model training often causes intelligent diagnosis systems to have biased predictions towards common diseases. The state-of-the-art approaches apply a two-stage learning framework to alleviate the class-imbalance issue, where the first stage focuses on training of a general feature extractor and the second stage focuses on fine-tuning the classifier head for class rebalancing. However, existing two-stage approaches do not consider the fine-grained property between different diseases, often causing the first stage less effective for medical image classification than for natural image classification tasks. In this study, we propose embedding metric learning into the first stage of the two-stage framework specially to help the feature extractor learn to extract more discriminative feature representations. Extensive experiments mainly on three medical image datasets show that the proposed approach consistently outperforms existing onestage and two-stage approaches, suggesting that metric learning can be used as an effective plug-in component in the two-stage framework for fine-grained class-imbalanced image classification tasks.
updated: Thu Jul 14 2022 14:57:01 GMT+0000 (UTC)
published: Thu Jul 14 2022 14:57:01 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アソシエイト