arXiv reaDer
大腿骨骨折分類を改善するためのカリキュラム学習:事前の知識と不確実性を伴うデータのスケジューリング
Curriculum learning for improved femur fracture classification: scheduling data with prior knowledge and uncertainty
X線画像から近位大腿骨骨折を適切に分類することは、治療法の選択と患者の臨床転帰にとって非常に重要です。骨折の位置と複雑さに応じて画像をタイプとサブタイプに分類する階層的な知識ツリーを記述する、一般的に使用されるAOシステムに依存しています。本論文では、畳み込みニューラルネットワーク(CNN)に基づいて近位大腿骨骨折を3および7AOクラスに自動分類する方法を提案します。知られているように、CNNには、信頼できるラベルが付いた大規模で代表的なデータセットが必要です。これらのデータセットは、手元のアプリケーション用に収集するのが困難です。このホワイトペーパーでは、このような条件下で基本的なCNNのパフォーマンスを向上させるカリキュラム学習(CL)アプローチを設計します。私たちの新しい定式化は、トレーニングサンプルの個別の重み付け、トレーニングセットの並べ替え、データのサブセットのサンプリングという3つのカリキュラム戦略を再統合します。これらの戦略の中核は、トレーニングサンプルをランク付けするスコアリング関数です。 2つの新しいスコアリング関数を定義します。1つはドメイン固有の事前知識からのもので、もう1つは元の自己ペースの不確実性スコアからのものです。近位大腿骨X線写真の臨床データセットで実験を行います。カリキュラムは、経験豊富な外傷外科医のパフォーマンスまで、近位大腿骨骨折の分類を改善します。最良のカリキュラム方法では、事前の知識に基づいてトレーニングセットを並べ替え、分類が15%向上します。公開されているMNISTデータセットを使用して、データ量が限られている、クラスの不均衡がある、ラベルノイズが存在する、という3つの制御された困難な数字認識シナリオに対する統合CL定式化の利点についてさらに説明および実証します。私たちの仕事のコードは、https://github.com/ameliajimenez/curriculum-learning-prior-uncertaintyで入手できます。
An adequate classification of proximal femur fractures from X-ray images is crucial for the treatment choice and the patients' clinical outcome. We rely on the commonly used AO system, which describes a hierarchical knowledge tree classifying the images into types and subtypes according to the fracture's location and complexity. In this paper, we propose a method for the automatic classification of proximal femur fractures into 3 and 7 AO classes based on a Convolutional Neural Network (CNN). As it is known, CNNs need large and representative datasets with reliable labels, which are hard to collect for the application at hand. In this paper, we design a curriculum learning (CL) approach that improves over the basic CNNs performance under such conditions. Our novel formulation reunites three curriculum strategies: individually weighting training samples, reordering the training set, and sampling subsets of data. The core of these strategies is a scoring function ranking the training samples. We define two novel scoring functions: one from domain-specific prior knowledge and an original self-paced uncertainty score. We perform experiments on a clinical dataset of proximal femur radiographs. The curriculum improves proximal femur fracture classification up to the performance of experienced trauma surgeons. The best curriculum method reorders the training set based on prior knowledge resulting into a classification improvement of 15%. Using the publicly available MNIST dataset, we further discuss and demonstrate the benefits of our unified CL formulation for three controlled and challenging digit recognition scenarios: with limited amounts of data, under class-imbalance, and in the presence of label noise. The code of our work is available at: https://github.com/ameliajimenez/curriculum-learning-prior-uncertainty.
updated: Tue Nov 09 2021 16:03:58 GMT+0000 (UTC)
published: Fri Jul 31 2020 14:28:33 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アソシエイト