arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1910.03349q-bio.NCstat.ML
收藏

脑预测建模中的自动机器学习方法分析:一种从皮层解剖测量预测脑年龄的数据驱动方法

Analysis of an Automated Machine Learning Approach in Brain Predictive Modelling: A data-driven approach to Predict Brain Age from Cortical Anatomical Measures

Jessica Dafflon, Walter H. L Pinaya, Federico Turkheimer, James H. Cole, Robert Leech, Mathew A. Harris, Simon R. Cox, Heather C. Whalley, Andrew M. McIntosh, Peter J. Hellyer

摘要:

机器学习(ML)算法的使用在神经科学中显著增加。然而,在众多可能的ML算法中,哪一种是预测目标变量的最佳模型?这种模型的超参数是什么?鉴于这些问题的可能答案过多,在过去的几年里,自动化机器学习(autoML)一直受到关注。在这里,我们应用了一个名为TPOT的autoML库,该库使用基于树的机器学习管道表示,并进行基于遗传编程的方法来找到模型及其超参数,更准确地预测受试者的真实年龄。为了探索autoML并评估其在神经成像数据集中的有效性,我们选择了一个以前广泛研究的焦点问题:大脑年龄预测。在没有任何先验知识的情况下,TPOT能够扫描整个模型空间,并创建管道,该管道的性能优于基于Freesurfer的模型的最新精度,仅使用解剖结构的厚度和体积信息。特别是,我们比较了TPOT(平均准确度误差(MAE):$4.612 \pm .124$年)和相关向量回归(MAE $5.474 \pm .140$年)的性能。TPOT还提出了一些有趣的模型组合,这些模型与目前最常用的大脑预测模型不匹配,但可以很好地推广到看不见的数据。AutoML作为一种数据驱动的方法,在寻找神经成像应用的最佳模型方面取得了可喜的成果。

英文摘要:

The use of machine learning (ML) algorithms has significantly increased in neuroscience. However, from the vast extent of possible ML algorithms, which one is the optimal model to predict the target variable? What are the hyperparameters for such a model? Given the plethora of possible answers to these questions, in the last years, automated machine learning (autoML) has been gaining attention. Here, we apply an autoML library called TPOT which uses a tree-based representation of machine learning pipelines and conducts a genetic-programming based approach to find the model and its hyperparameters that more closely predicts the subject's true age. To explore autoML and evaluate its efficacy within neuroimaging datasets, we chose a problem that has been the focus of previous extensive study: brain age prediction. Without any prior knowledge, TPOT was able to scan through the model space and create pipelines that outperformed the state-of-the-art accuracy for Freesurfer-based models using only thickness and volume information for anatomical structure. In particular, we compared the performance of TPOT (mean accuracy error (MAE): $4.612 \pm .124$ years) and a Relevance Vector Regression (MAE $5.474 \pm .140$ years). TPOT also suggested interesting combinations of models that do not match the current most used models for brain prediction but generalise well to unseen data. AutoML showed promising results as a data-driven approach to find optimal models for neuroimaging applications.

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

↑