Empirical Study of Overfitting in Deep Learning for Predicting Breast Cancer Metastasis.

ABSTRACT: Overfitting may affect the accuracy of predicting future data because of weakened generalization. In this research, we used an electronic health records (EHR) dataset concerning breast cancer metastasis to study the overfitting of deep feedforward neural networks (FNNs) prediction models. We studied how each hyperparameter and some of the interesting pairs of hyperparameters were interacting to influence the model performance and overfitting. The 11 hyperparameters we studied were activate function, weight initializer, number of hidden layers, learning rate, momentum, decay, dropout rate, batch size, epochs, L1, and L2. Our results show that most of the single hyperparameters are either negatively or positively corrected with model prediction performance and overfitting. In particular, we

SUBMITTER: Xu C 

PROVIDER: S-EPMC10093528 | biostudies-literature | 2023 Mar

REPOSITORIES: biostudies-literature