arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.04438eess.IVcs.LGq-bio.QM

Improved Brain Age Estimation with Slice-based Set Networks

University of Southern California(南加州大学);Information Sciences Institute(信息科学研究所);Mark and Mary Stevens Institute for Neuroimaging and Informatics(马克与玛丽·史蒂文斯神经影像与信息学研究所);Keck School of Medicine(凯克医学院)

Umang Gupta, Pradeep K. Lam, Greg Ver Steeg, Paul M. Thompson

英文摘要:

Deep Learning for neuroimaging data is a promising but challenging direction. The high dimensionality of 3D MRI scans makes this endeavor compute and data-intensive. Most conventional 3D neuroimaging methods use 3D-CNN-based architectures with a large number of parameters and require more time and data to train. Recently, 2D-slice-based models have received increasing attention as they have fewer parameters and may require fewer samples to achieve comparable performance. In this paper, we propose a new architecture for BrainAGE prediction. The proposed architecture works by encoding each 2D slice in an MRI with a deep 2D-CNN model. Next, it combines the information from these 2D-slice encodings using set networks or permutation invariant layers. Experiments on the BrainAGE prediction problem, using the UK Biobank dataset, showed that the model with the permutation invariant layers trains faster and provides better predictions compared to other state-of-the-art approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑