Effective hyperparameter optimization with proxy data for multi-organ segmentation

Chen Shen; Holger R. Roth; Vishwesh Nath; Yuichiro Hayashi; Masahiro Oda; Kazunari Misawa; Kensaku Mori

doi:10.1117/12.2611422

4 April 2022 Effective hyperparameter optimization with proxy data for multi-organ segmentation

Chen Shen, Holger R. Roth, Vishwesh Nath, Yuichiro Hayashi, Masahiro Oda, Kazunari Misawa, Kensaku Mori

Proceedings Volume 12032, Medical Imaging 2022: Image Processing; 120320T (2022) https://doi.org/10.1117/12.2611422
Event: SPIE Medical Imaging, 2022, San Diego, California, United States

Abstract

The purpose of this paper is to introduce a practical framework of using proxy data in automatic hyperparameter optimization for 3D multi-organ segmentation. The automated segmentation of abdominal organs from CT volumes is a main task in the medical image analysis field. Much research has been investigated to handle this task based on the immense experience of machine learning. Deep learning approaches require enormous experiments to design the optimal configurations for the best performance. Automatic machine learning (AutoML) using hyperparameter optimization to search the optimal training strategy makes it possible to find the appropriate settings without much deep experience. However, biases of training data can be highly related to the AutoML performance and efficiency. In this paper, we propose an AutoML framework that uses pre-selected proxy data to represent the entire dataset which has the potential to reduce the computation time needed for efficient hyperparameter optimization in searching learning. Both quantitative and qualitative results showed that our framework can effectively build more powerful segmentation models than manually designed deep-learning-based methods and AutoML, which use carefully tuned hyperparameters and randomly selected training subsets, respectively. The average Dice score for 10-class abdominal organ segmentation was 85.9%.

Conference Presentation

Citation Download Citation

Chen Shen, Holger R. Roth, Vishwesh Nath, Yuichiro Hayashi, Masahiro Oda, Kazunari Misawa, and Kensaku Mori "Effective hyperparameter optimization with proxy data for multi-organ segmentation", Proc. SPIE 12032, Medical Imaging 2022: Image Processing, 120320T (4 April 2022); https://doi.org/10.1117/12.2611422

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available