arXiv reaDer
半教師あり医用画像分類のためのクラス固有の分布調整
Class-Specific Distribution Alignment for Semi-Supervised Medical Image Classification
医用画像分類におけるディープ ニューラル ネットワークの成功にもかかわらず、データの注釈には時間がかかり、病気の相対的な希少性によりクラス分布の不均衡が生じるため、この問題は依然として困難です。この問題に対処するために、私たちは、非常に不均衡なデータセットから学習するのに適した自己学習に基づく半教師あり学習フレームワークであるクラス固有分布アライメント (CSDA) を提案します。具体的には、まずプロセスを周辺予測が広がるベクトル空間の基底の変化とみなすことによって分布調整に新しい視点を提供し、次にラベル付きデータとラベルなしデータの両方でクラス依存の周辺予測を捕捉するための CSDA を導出します。多数派層への偏見を避ける。さらに、各クラスのラベルなしサンプルのバランスのとれた数を維持するために、Variable Condition Queue (VCQ) モジュールを提案します。 3 つの公開データセット HAM10000、CheXpert、および Kvasir での実験では、私たちの方法が半教師ありの皮膚疾患、胸部疾患、および内視鏡画像分類タスクで競合するパフォーマンスを提供することが示されています。
Despite the success of deep neural networks in medical image classification, the problem remains challenging as data annotation is time-consuming, and the class distribution is imbalanced due to the relative scarcity of diseases. To address this problem, we propose Class-Specific Distribution Alignment (CSDA), a semi-supervised learning framework based on self-training that is suitable to learn from highly imbalanced datasets. Specifically, we first provide a new perspective to distribution alignment by considering the process as a change of basis in the vector space spanned by marginal predictions, and then derive CSDA to capture class-dependent marginal predictions on both labeled and unlabeled data, in order to avoid the bias towards majority classes. Furthermore, we propose a Variable Condition Queue (VCQ) module to maintain a proportionately balanced number of unlabeled samples for each class. Experiments on three public datasets HAM10000, CheXpert and Kvasir show that our method provides competitive performance on semi-supervised skin disease, thoracic disease, and endoscopic image classification tasks.
updated: Sat Jul 29 2023 13:38:19 GMT+0000 (UTC)
published: Sat Jul 29 2023 13:38: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アソシエイト