arXiv reaDer
ルービックキューブの再生による3D医用画像の自己監視機能学習
Self-supervised Feature Learning for 3D Medical Images by Playing a Rubik's Cube
  ディープラーニングの発展を目の当たりにし、3Dボリューム医療データのコンピューター支援診断システムを構築しようとする研究が増えています。ただし、3D医療データの注釈を取得するのは難しいため、注釈付きの3D医療画像の数では、ディープラーニングネットワークを適切にトレーニングするには不十分なことがよくあります。生データの情報を深く活用する自己監視学習は、トレーニングデータの要件を緩和する潜在的なソリューションの1つです。本論文では、体積医用画像の自己学習型学習フレームワークを提案します。新しいプロキシタスク、つまりルービックキューブの回復は、3Dニューラルネットワークを事前にトレーニングするために定式化されています。プロキシタスクには、キューブの再配置とキューブの回転という2つの操作が含まれます。これらの操作により、ネットワークは未加工の3Dデータから平行移動および回転の不変特徴を学習します。スクラッチからのトレーニング戦略と比較して、事前トレーニングされたネットワークからの微調整により、脳出血の分類や脳腫瘍のセグメンテーションなど、さまざまなタスクの精度が向上します。自己監視学習アプローチにより、追加のデータを使用することなく、体積医療データセットの3Dディープラーニングネットワークの精度を大幅に高めることができることを示します。私たちの知る限り、これは3Dニューラルネットワークの自己監視学習に焦点を当てた最初の研究です。
Witnessed the development of deep learning, increasing number of studies try to build computer aided diagnosis systems for 3D volumetric medical data. However, as the annotations of 3D medical data are difficult to acquire, the number of annotated 3D medical images is often not enough to well train the deep learning networks. The self-supervised learning deeply exploiting the information of raw data is one of the potential solutions to loose the requirement of training data. In this paper, we propose a self-supervised learning framework for the volumetric medical images. A novel proxy task, i.e., Rubik's cube recovery, is formulated to pre-train 3D neural networks. The proxy task involves two operations, i.e., cube rearrangement and cube rotation, which enforce networks to learn translational and rotational invariant features from raw 3D data. Compared to the train-from-scratch strategy, fine-tuning from the pre-trained network leads to a better accuracy on various tasks, e.g., brain hemorrhage classification and brain tumor segmentation. We show that our self-supervised learning approach can substantially boost the accuracies of 3D deep learning networks on the volumetric medical datasets without using extra data. To our best knowledge, this is the first work focusing on the self-supervised learning of 3D neural networks.
updated: Sat Oct 05 2019 10:02:13 GMT+0000 (UTC)
published: Sat Oct 05 2019 10:02:13 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アソシエイト