Skip to main content
SHARE
Publication

Model-based Hyperparameter Optimization of Convolutional Neural Networks for Information Extraction from Cancer Pathology Rep...

Publication Type
Conference Paper
Journal Name
IEEE-EMBS International Conference on Biomedical and Health Informatics 2019
Book Title
Proceedings of the 2019 IEEE-EMBS Biomedical and Health Informatics Conference (BHI)
Publication Date
Page Numbers
1 to 4
Issue
1
Conference Name
IEEE-EMBS International Conference on Biomedical and Health Informatics (BHI 2019)
Conference Location
Chicago, Illinois, United States of America
Conference Sponsor
IEEE
Conference Date
-

Finding optimal hyperparameters is necessary to identify the best performing deep learning models but the process is costly. In this paper, we applied model-based optimization, also known as Bayesian optimization, using the CANDLE framework implemented on a High-Performance Computing environment. As a use case we selected information extraction from cancer pathology reports using a multi-task convolutional neural network, and hierarchical convolutional attention network to be optimized. We utilized a synthesized text corpus of 8,000 training cases and 2,000 validation cases with four types of clinical task labels including primary cancer site, laterality, behavior, and histological grade. We demonstrated that hyperparameter optimization using the CANDLE framework is a feasible approach with respect to both scalability and clinical task performance.