arXiv reaDer
説明可能でコスト重視の教師あり対照学習によるディープ ニューラル ネットワークによる消化器疾患の分類
Gastrointestinal Disease Classification through Explainable and Cost-Sensitive Deep Neural Networks with Supervised Contrastive Learning
消化器疾患はさまざまな形で現れ、潜在的な合併症を引き起こす可能性があるため、医療に重大な課題をもたらします。これらの疾患を正確かつタイムリーに分類することは、治療の選択を導き、患者の転帰を向上させる上で極めて重要です。この論文では、教師あり対比学習を備えたコスト重視の事前トレーニング済みディープ畳み込みニューラル ネットワーク (CNN) アーキテクチャを活用して、消化器疾患を分類する新しいアプローチを紹介します。私たちのアプローチにより、ネットワークは、サンプル間の類似性の関係も考慮しながら、重要な疾患関連の特徴を捉えた表現を学習することができます。不均衡なデータセットと医療における誤分類エラーのコスト重視の性質によってもたらされる課題に取り組むために、私たちはコスト重視の学習を組み込みます。疾患クラスに基づいて誤分類に個別のコストを割り当てることで、重篤な状態の正確な分類を優先します。さらに、説明可能な人工知能 (AI) の勾配ベースの手法を統合することで、モデルの解釈可能性を強化します。この包含により、ネットワークの意思決定プロセスに関する貴重な洞察が得られ、疾患の分類に寄与する特徴を理解するのに役立ちます。私たちが提案したアプローチの有効性を評価するために、Hyper-Kvasir データセットなどの包括的な消化器疾患データセットに対して広範な実験を実行します。既存の研究との徹底的な比較を通じて、私たちのモデルの強力な分類精度、堅牢性、解釈可能性を実証します。私たちは、提案したアプローチの実装を https://github.com/dibya404/Gastrointestinal-Disease-Classification-through-Explainable-and-Cost-Sensitive-DNN-with-SCL で公開しました。
Gastrointestinal diseases pose significant healthcare chall-enges as they manifest in diverse ways and can lead to potential complications. Ensuring precise and timely classification of these diseases is pivotal in guiding treatment choices and enhancing patient outcomes. This paper introduces a novel approach on classifying gastrointestinal diseases by leveraging cost-sensitive pre-trained deep convolutional neural network (CNN) architectures with supervised contrastive learning. Our approach enables the network to learn representations that capture vital disease-related features, while also considering the relationships of similarity between samples. To tackle the challenges posed by imbalanced datasets and the cost-sensitive nature of misclassification errors in healthcare, we incorporate cost-sensitive learning. By assigning distinct costs to misclassifications based on the disease class, we prioritize accurate classification of critical conditions. Furthermore, we enhance the interpretability of our model by integrating gradient-based techniques from explainable artificial intelligence (AI). This inclusion provides valuable insights into the decision-making process of the network, aiding in understanding the features that contribute to disease classification. To assess the effectiveness of our proposed approach, we perform extensive experiments on a comprehensive gastrointestinal disease dataset, such as the Hyper-Kvasir dataset. Through thorough comparisons with existing works, we demonstrate the strong classification accuracy, robustness and interpretability of our model. We have made the implementation of our proposed approach publicly available at https://github.com/dibya404/Gastrointestinal-Disease-Classification-through-Explainable-and-Cost-Sensitive-DNN-with-SCL
updated: Fri Jul 14 2023 19:56:30 GMT+0000 (UTC)
published: Fri Jul 14 2023 19:56:30 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アソシエイト