arXiv reaDer
脳MRIの年齢推定のためのタスクガイド付き生成的敵対的ネットワークを使用した画像拡張
Image Augmentation Using a Task Guided Generative Adversarial Network for Age Estimation on Brain MRI
磁気共鳴画像法(MRI)に基づく脳年齢推定は、高齢者の神経変性疾患(アルツハイマー病、パーキンソン病、ハンチントン病など)や若いグループの脳の発達不全の早期診断における活発な研究分野です。ディープラーニング手法は、脳年齢の推定を含む多くの医療画像分析タスクで最先端のパフォーマンスを実現しています。ただし、深層学習モデルのパフォーマンスと一般化可能性は、トレーニングデータセットの量と質に大きく依存します。脳のMRIデータの収集と注釈付けは、どちらも非常に時間がかかります。本論文では、データ不足の問題を克服するために、生成的敵対的ネットワーク(GAN)ベースの画像合成法を提案します。既存のGANベースの方法とは異なり、タスクガイドブランチ(年齢推定の回帰モデル)をGANのジェネレーターの最後に統合します。従来のGAN損失にタスク誘導損失を追加することにより、学習された低次元の潜在空間と合成された画像は、よりタスク固有になります。モデルトレーニング用に合成画像と実際の画像を組み合わせることで、ダウンストリームタスクのパフォーマンスを向上させるのに役立ちます。提案された方法は、年齢推定のための公共の脳MRIデータセットで評価されました。私たちの提案した方法は、深い畳み込みニューラルネットワークベースの回帰モデルとタスクガイドブランチのないGANベースの画像合成方法を上回りました(統計的に有意)。さらに重要なことに、それは画像空間における加齢に関連した脳領域の識別を可能にします。コードはGitHub(https://github.com/ruizhe-l/tgb-gan)で入手できます。
Brain age estimation based on magnetic resonance imaging (MRI) is an active research area in early diagnosis of some neurodegenerative diseases (e.g. Alzheimer, Parkinson, Huntington, etc.) for elderly people or brain underdevelopment for the young group. Deep learning methods have achieved the state-of-the-art performance in many medical image analysis tasks, including brain age estimation. However, the performance and generalisability of the deep learning model are highly dependent on the quantity and quality of the training data set. Both collecting and annotating brain MRI data are extremely time-consuming. In this paper, to overcome the data scarcity problem, we propose a generative adversarial network (GAN) based image synthesis method. Different from the existing GAN-based methods, we integrate a task-guided branch (a regression model for age estimation) to the end of the generator in GAN. By adding a task-guided loss to the conventional GAN loss, the learned low-dimensional latent space and the synthesised images are more task-specific. It helps to boost the performance of the down-stream task by combining the synthesised images and real images for model training. The proposed method was evaluated on a public brain MRI data set for age estimation. Our proposed method outperformed (statistically significant) a deep convolutional neural network based regression model and the GAN-based image synthesis method without the task-guided branch. More importantly, it enables the identification of age-related brain regions in the image space. The code is available on GitHub (https://github.com/ruizhe-l/tgb-gan).
updated: Tue Aug 03 2021 17:56:50 GMT+0000 (UTC)
published: Tue Aug 03 2021 17:56:50 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アソシエイト