arXiv reaDer
臨床レポートからの自動ラベル抽出による深層強化学習は、3DMRI脳ボリュームを正確に分類します
Deep reinforcement learning with automated label extraction from clinical reports accurately classifies 3D MRI brain volumes
目的:画像分類は、AIの画像化におけるおそらく最も基本的なタスクです。ただし、画像のラベル付けは時間と手間がかかります。最近、強化学習(RL)がMRI脳画像の2Dスライスを高精度で分類できることを実証しました。ここでは、画像分類を高速化するための2つの重要なステップを実行します。最初に、臨床レポートからクラスラベルを自動的に抽出します。次に、以前の2D分類作業を、私たちの機関からの完全な3D画像ボリュームに拡張します。したがって、次のように進めます。パート1では、SBERT自然言語処理アプローチを使用してレポートからラベルを自動的に抽出します。次に、パート2では、これらのラベルをRLとともに使用して、3D画像ボリュームの分類Deep-Qネットワーク(DQN)をトレーニングします。方法:パート1では、90の放射線レポートのインプレッションでSBERTをトレーニングしました。次に、トレーニング済みのSBERTを使用して、パート2で使用するクラスラベルを予測しました。パート2では、マルチステップ画像分類を適用して、3D畳み込みとTD(0)Q学習を使用したDeep-Q学習を組み合わせました。 90枚の画像のセットでトレーニングしました。パート1のトレーニング済みSBERTによる患者レポートから予測されたクラスを使用して、61枚の画像の別のセットでテストしました。比較のために、同じセットのトレーニングおよびテスト画像で教師あり深層学習分類ネットワークのトレーニングとテストも行いました。同じラベル。結果:パート1:放射線レポートのコーパスを使用してトレーニングすると、SBERTモデルは通常のスキャンと転移を含むスキャンの両方で100%の精度を示しました。パート2:次に、これらのラベルを使用すると、教師ありアプローチはトレーニングデータにすぐに適合し、テストセットでは期待どおりにパフォーマンスが低下しましたが(66%の精度、ランダムな推測をわずかに上回りました)、強化学習アプローチは92%の精度を達成しました。結果は統計的に有意であり、p値は3.1 x 10 ^ -5でした。
Purpose: Image classification is perhaps the most fundamental task in imaging AI. However, labeling images is time-consuming and tedious. We have recently demonstrated that reinforcement learning (RL) can classify 2D slices of MRI brain images with high accuracy. Here we make two important steps toward speeding image classification: Firstly, we automatically extract class labels from the clinical reports. Secondly, we extend our prior 2D classification work to fully 3D image volumes from our institution. Hence, we proceed as follows: in Part 1, we extract labels from reports automatically using the SBERT natural language processing approach. Then, in Part 2, we use these labels with RL to train a classification Deep-Q Network (DQN) for 3D image volumes. Methods: For Part 1, we trained SBERT with 90 radiology report impressions. We then used the trained SBERT to predict class labels for use in Part 2. In Part 2, we applied multi-step image classification to allow for combined Deep-Q learning using 3D convolutions and TD(0) Q learning. We trained on a set of 90 images. We tested on a separate set of 61 images, again using the classes predicted from patient reports by the trained SBERT in Part 1. For comparison, we also trained and tested a supervised deep learning classification network on the same set of training and testing images using the same labels. Results: Part 1: Upon training with the corpus of radiology reports, the SBERT model had 100% accuracy for both normal and metastasis-containing scans. Part 2: Then, using these labels, whereas the supervised approach quickly overfit the training data and as expected performed poorly on the testing set (66% accuracy, just over random guessing), the reinforcement learning approach achieved an accuracy of 92%. The results were found to be statistically significant, with a p-value of 3.1 x 10^-5.
updated: Thu Jun 17 2021 20:53:41 GMT+0000 (UTC)
published: Thu Jun 17 2021 20:53:41 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アソシエイト