Automated Machine Learning in Medical Image Segmentation

Automated Machine Learning in Medical Image Segmentation

van Gruijthuijsen, Coen (TU Delft Mechanical, Maritime and Materials Engineering)

Pan, W. (mentor)
Starmans, Martijn (mentor)
van Garderen, Karin (mentor)
Klein, Stefan (graduation committee)
Wisse, M. (graduation committee)

Delft University of Technology

Mechanical Engineering

2021-11-05

Semantic Segmentation of medical images are used to improve diagnosis and treatment. In recent years, the application of machine learning methods are increasingly used. However, the design of these models is difficult and time-consuming. In this thesis, we investigated the automation of this process using an Automated Machine Learning (AutoML) algorithm called Bayesian Optimization with HyperBand (BOHB). We found several hyperparameters significantly influencing the performance and confirmed that BOHB is applicable to these kind of problems. However, we did not find a significant difference between BOHB and Random Search on a small search, stressing the need of matching the search space to the search strategy. Finally, our approach was only slightly worse than the current state-of-the-art, which shows that AutoML has potential in this field.

Automated Machine Learning
Hyperparameter Optimization
Convolutional Neural Networks (CNNs)
Semantic Segmentation

http://resolver.tudelft.nl/uuid:cd200c93-3185-4c5b-a3ef-4113b997cee1

Student theses

master thesis

© 2021 Coen van Gruijthuijsen