arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1701.04355cs.LGstat.ML
收藏

使用深度学习和基于高斯过程的模型选择对MRI数据进行分类

Classification of MRI data using Deep Learning and Gaussian Process-based Model Selection

Philips Research(飞利浦研究院)

Hadrien Bertrand, Matthieu Perrot, Roberto Ardon, Isabelle Bloch

摘要:

随着医学图像数据量的不断增加,根据解剖学视野对MRI图像进行分类是一个必须解决的问题。与此同时,深度学习的进步使其成为解决计算机视觉问题的合适工具。使用通用架构(如AlexNet)提供了相当好的结果,但不足以用于临床使用。改进模型并不是一件容易的事情,因为网络的架构和训练都需要大量的超参数,而且对它们的相关性的理解有限。由于穷举搜索不容易处理,我们建议首先通过随机搜索优化网络,然后通过基于高斯过程和改进概率的自适应搜索。应用这种方法在一个大的和不同的MRI数据集,我们表现出基线网络和最终的一个(最困难的类高达20%)之间的实质性改善。

英文摘要:

The classification of MRI images according to the anatomical field of view is a necessary task to solve when faced with the increasing quantity of medical images. In parallel, advances in deep learning makes it a suitable tool for computer vision problems. Using a common architecture (such as AlexNet) provides quite good results, but not sufficient for clinical use. Improving the model is not an easy task, due to the large number of hyper-parameters governing both the architecture and the training of the network, and to the limited understanding of their relevance. Since an exhaustive search is not tractable, we propose to optimize the network first by random search, and then by an adaptive search based on Gaussian Processes and Probability of Improvement. Applying this method on a large and varied MRI dataset, we show a substantial improvement between the baseline network and the final one (up to 20\% for the most difficult classes).

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

↑