ボクセルワイズrs-fMRIデータに非線形表現学習を適用することを提案します。非線形表現の学習は、変分オートエンコーダー(VAE)を使用して行われます。 VAEは、ボクセル単位のrs-fMRIデータでトレーニングされ、意味のある情報を保持する非線形次元削減を実行します。モデルの表現における情報の保持は、下流の年齢回帰および性別分類タスクを使用して評価されます。これらのタスクの結果は非常に有望であり、教師なしモデルの表現でトレーニングされた線形回帰分析は、同じデータセットで年齢回帰のために特別にトレーニングされた教師ありニューラルネットワークとほぼ同じように機能します。モデルはまた、統合失調症診断予測タスクで評価され、神経精神医学データセットの次元削減方法としての実現可能性を評価します。これらの結果は、下流の神経精神医学的課題の結果を改善するために、精神疾患を持たないより多くの個人の事前トレーニングの可能性を浮き彫りにしています。事前トレーニングされたモデルは、統合失調症データセットの可変数のエポックに対して微調整されており、1エポックの微調整が最良の結果をもたらすことがわかります。したがって、この作業は、ボクセルワイズrs-fMRIデータの非線形次元削減を開くだけでなく、ボクセルワイズrs-fMRIデータセットで深層学習モデルを事前トレーニングすると、小さいデータセットでもパフォーマンスが大幅に向上することを示しています。また、将来の研究で統合失調症のような不均一な神経精神障害のVAEの潜在空間におけるrs-fMRI時系列の分布を調べる能力を開きます。これは、モデルの潜在空間から脳空間にポイントを再構築し、VAEが被験者、時点、および被験者の特性の間で学習する関係の理解を深めることができるモデルの生成的側面で補完できます。
We propose to apply non-linear representation learning to voxelwise rs-fMRI data. Learning the non-linear representations is done using a variational autoencoder (VAE). The VAE is trained on voxelwise rs-fMRI data and performs non-linear dimensionality reduction that retains meaningful information. The retention of information in the model's representations is evaluated using downstream age regression and sex classification tasks. The results on these tasks are highly encouraging and a linear regressor trained with the representations of our unsupervised model performs almost as well as a supervised neural network, trained specifically for age regression on the same dataset. The model is also evaluated with a schizophrenia diagnosis prediction task, to assess its feasibility as a dimensionality reduction method for neuropsychiatric datasets. These results highlight the potential for pre-training on a larger set of individuals who do not have mental illness, to improve the downstream neuropsychiatric task results. The pre-trained model is fine-tuned for a variable number of epochs on a schizophrenia dataset and we find that fine-tuning for 1 epoch yields the best results. This work therefore not only opens up non-linear dimensionality reduction for voxelwise rs-fMRI data but also shows that pre-training a deep learning model on voxelwise rs-fMRI datasets greatly increases performance even on smaller datasets. It also opens up the ability to look at the distribution of rs-fMRI time series in the latent space of the VAE for heterogeneous neuropsychiatric disorders like schizophrenia in future work. This can be complemented with the generative aspect of the model that allows us to reconstruct points from the model's latent space back into brain space and obtain an improved understanding of the relation that the VAE learns between subjects, timepoints, and a subject's characteristics.