Enhanced Graph Cuts for Brain Tumor Segmentation Using Bayesian Optimization
Brain tumor segmentation is a difficult task, due to the shape variability that malignancy brain structures exhibit between patients. The main problem in this process is that the tumor contour is usually computed from parametric models that need to be well-tuned to perform an accurate segmentation. In this paper, we propose an enhanced Graph cut on which the model parameters are selected through a probabilistic approach. Here, we use Bayesian optimization to find the optimal hyperparameters that segment the tumor volume accurately. The experimental results show that by using Bayesian optimization, the graph cut model performs an accurate segmentation over brain volumes in comparison with common segmentation methods in the state-of-the-art.
KeywordsBayesian optimization Graph cuts Gaussian processes Brain tumor segmentation
This research is developed under the project financed by COLCIENCIAS with code 111074455860. H.F. García is funded by Colciencias under the program: Formación de alto nivel para la ciencia, la tecnología y la innovación - Convocatoria 617 de 2013.
- 1.Beers, A., et al.: Sequential 3D U-Nets for biologically-informed brain tumor segmentation. CoRR abs/1709.02967 (2017)Google Scholar
- 2.Bharath, H.N., Colleman, S., Sima, D.M., Van Huffel, S.: Tumor segmentation from multi-parametric MRI using random forest with superpixel and tensor based feature extraction. In: Brats Challenge 2017 (2017)Google Scholar
- 4.Castillo, L.S., Daza, L.A., Rivera, L.C., Arbeláez, P.: Volumetric multimodality neural network for brain tumor segmentation (2017)Google Scholar
- 7.Gonzalez, J., Longworth, J., James, D., Lawrence, N.: Bayesian optimisation for synthetic gene design. In: NIPS Workshop on Bayesian Optimization in Academia and Industry (2014)Google Scholar
- 9.Isin, A., Direkoglu, C., Sah, M.: Review of MRI-based brain tumor image segmentation using deep learning methods. Proc. Comput. Sci. 102, 317–324 (2016). 12th International Conference on Application of Fuzzy Systems and Soft Computing, ICAFS 2016, 29–30 August 2016, Vienna, AustriaCrossRefGoogle Scholar
- 10.Jiřík, M., Lukes, V., Svobodova, M., Železný, M.: Image segmentation in medical imaging via graph-cuts (2013)Google Scholar
- 13.Rasmussen, C.E., Williams, C.K.I.: Gaussian Processes for Machine Learning (Adaptive Computation and Machine Learning). The MIT Press (2005)Google Scholar
- 14.Shengcong Chen, C.D., Zhou, C.: Brain tumor segmentation with label distribution learning and multi-level feature representation. In: Brats Challenge 2017 (2017)Google Scholar
- 15.Snoek, J., Larochelle, H., Adams, R.P.: Practical Bayesian optimization of machine learning algorithms. In: Pereira, F., Burges, C.J.C., Bottou, L., Weinberger, K.Q. (eds.) Advances in Neural Information Processing Systems, vol. 25, pp. 2951–2959. Curran Associates, Inc. (2012)Google Scholar