arXiv reaDer
医用画像の分類とセグメンテーションのためのマスクされたオートエンコーダーによる自己事前トレーニング
Self Pre-training with Masked Autoencoders for Medical Image Classification and Segmentation
マスク オートエンコーダー (MAE) は、自然画像解析のためのビジョン トランスフォーマー (ViT) の事前トレーニングに効果的であることが最近示されました。部分的にマスクされた入力から完全な画像を再構築することにより、ViT エンコーダーはコンテキスト情報を集約して、マスクされた画像領域を推測します。このコンテキスト集約機能は、各解剖学的構造が機能的および機械的に他の構造や領域に接続されている医用画像ドメインに特に不可欠であると考えています。事前トレーニング用の ImageNet 規模の医用画像データセットがないため、医用画像分析タスク用に MAE を使用した自己事前トレーニング パラダイムを調査します。私たちの方法は、別のデータセットではなく、ターゲット データのトレーニング セットで ViT を事前トレーニングします。したがって、自己事前トレーニングは、事前トレーニング データを取得するのが難しいより多くのシナリオに役立ちます。私たちの実験結果は、MAE 自己事前トレーニングが、胸部 X 線疾患分類、腹部 CT 多臓器セグメンテーション、MRI 脳腫瘍セグメンテーションなどの多様な医用画像タスクを著しく改善することを示しています。コードは https://github.com/cvlab-stonybrook/SelfMedMAE で入手できます
Masked Autoencoder (MAE) has recently been shown to be effective in pre-training Vision Transformers (ViT) for natural image analysis. By reconstructing full images from partially masked inputs, a ViT encoder aggregates contextual information to infer masked image regions. We believe that this context aggregation ability is particularly essential to the medical image domain where each anatomical structure is functionally and mechanically connected to other structures and regions. Because there is no ImageNet-scale medical image dataset for pre-training, we investigate a self pre-training paradigm with MAE for medical image analysis tasks. Our method pre-trains a ViT on the training set of the target data instead of another dataset. Thus, self pre-training can benefit more scenarios where pre-training data is hard to acquire. Our experimental results show that MAE self pre-training markedly improves diverse medical image tasks including chest X-ray disease classification, abdominal CT multi-organ segmentation, and MRI brain tumor segmentation. Code is available at https://github.com/cvlab-stonybrook/SelfMedMAE
updated: Fri Apr 21 2023 12:40:08 GMT+0000 (UTC)
published: Thu Mar 10 2022 16:22:38 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アソシエイト