Enhanced Graph Cuts for Brain Tumor Segmentation Using Bayesian Optimization

  • Mauricio CastañoEmail author
  • Hernán F. GarcíaEmail author
  • Gloria L. Porras-HurtadoEmail author
  • Álvaro A. OrozcoEmail author
  • Jorge I. Marin-HurtadoEmail author
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11401)


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.


Bayesian 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. 1.
    Beers, A., et al.: Sequential 3D U-Nets for biologically-informed brain tumor segmentation. CoRR abs/1709.02967 (2017)Google Scholar
  2. 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
  3. 3.
    Boykov, Y., Funka-Lea, G.: Graph cuts and efficient N-D image segmentation. Int. J. Comput. Vis. 70, 109–131 (2006)CrossRefGoogle Scholar
  4. 4.
    Castillo, L.S., Daza, L.A., Rivera, L.C., Arbeláez, P.: Volumetric multimodality neural network for brain tumor segmentation (2017)Google Scholar
  5. 5.
    Delong, A., Osokin, A., Isack, H.N., Boykov, Y.: Fast approximate energy minimization with label costs. Int. J. Comput. Vis. 96(1), 1–27 (2012)MathSciNetCrossRefGoogle Scholar
  6. 6.
    Dogra, J., Jain, S., Sood, M.: Segmentation of MR images using hybrid kmean-graph cut technique. Proc. Comput. Sci. 132, 775–784 (2018). International Conference on Computational Intelligence and Data ScienceCrossRefGoogle Scholar
  7. 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
  8. 8.
    Havaei, M., et al.: Brain tumor segmentation with deep neural networks. Med. Image Anal. 35, 18–31 (2017)CrossRefGoogle Scholar
  9. 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. 10.
    Jiřík, M., Lukes, V., Svobodova, M., Železný, M.: Image segmentation in medical imaging via graph-cuts (2013)Google Scholar
  11. 11.
    Jones, D.R.: A taxonomy of global optimization methods based on response surfaces. J. Glob. Optim. 21(4), 345–383 (2001)MathSciNetCrossRefGoogle Scholar
  12. 12.
    Menze, B.H., Jakab, A., Bauer, S., Kalpathy-Cramer, J., et al.: The multimodal brain tumor image segmentation benchmark (brats). IEEE Trans. Med. Imaging 34(10), 1993–2024 (2015)CrossRefGoogle Scholar
  13. 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. 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. 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

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.GAMMA Research Group, Universidad del QuindíoArmeniaColombia
  2. 2.Grupo de Investigación Salud Comfamiliar, Comfamiliar RisaraldaPereiraColombia
  3. 3.Grupo de Investigación en Automática, Universidad Tecnológica de PereiraPereiraColombia
  4. 4.GDSPROC Research Group, Universidad del QuindíoArmeniaColombia

Personalised recommendations