Methodology for hyperparameter tuning of deep neural networks for efficient and accurate molecular property prediction

被引:0
作者
Nguyen, Xuan Dung James [1 ]
Liu, Y. A. [1 ]
机构
[1] Virginia Polytech Inst & State Univ, Aspen Tech Ctr Excellence Proc Syst Engn, Dept Chem Engn, Blacksburg, VA 24061 USA
关键词
Hyperparameter optimization; Molecular property prediction; Deep neural networks; Machine learning; Melt index; Glass transition temperature of polymers;
D O I
10.1016/j.compchemeng.2024.108928
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
This paper presents a methodology of hyperparameter optimization (HPO) of deep neural networks for molecular property prediction (MPP). Most prior applications of deep learning to MPP have paid only limited attention to HPO, thus resulting in suboptimal values of predicted properties. To improve the efficiency and accuracy of deep learning models for MPP, we must optimize as many hyperparameters as possible and choose a software platform to enable the parallel execution of HPO. We compare the random search, Bayesian optimization, and hyperband algorithms, together with the Bayesian-hyperband combination within the software packages of KerasTuner and Optuna for HPO. We conclude that the hyperband algorithm, which has not been used in previous MPP studies, is most computationally efficient; it gives MPP results that are optimal or nearly optimal in terms of prediction accuracy. Based on our case studies, we recommend the use of the Python library KerasTuner for HPO.
引用
收藏
页数:17
相关论文
共 27 条
  • [1] Anggoro D. A., 2021, Int. J. Intell. Eng. Syst., V14, P198, DOI DOI 10.22266/IJIES2021.1231.19
  • [2] Bergstra J, 2012, J MACH LEARN RES, V13, P281
  • [3] Boldini D, 2023, J CHEMINFORMATICS, V15, DOI 10.1186/s13321-023-00743-7
  • [4] Predicting Polymers' Glass Transition Temperature by a Chemical Language Processing Model
    Chen, Guang
    Tao, Lei
    Li, Ying
    [J]. POLYMERS, 2021, 13 (11)
  • [5] A general optimization protocol for molecular property prediction using a deep learning network
    Chen, Jen-Hao
    Tseng, Yufeng Jane
    [J]. BRIEFINGS IN BIOINFORMATICS, 2022, 23 (01)
  • [6] Deep learning based approach for prediction of glass transition temperature in polymers
    Goswami, Subhasish
    Ghosh, Rajdeep
    Neog, Arohan
    Das, Bitopan
    [J]. MATERIALS TODAY-PROCEEDINGS, 2021, 46 : 5838 - 5843
  • [7] Chemprop: A Machine Learning Package for Chemical Property Prediction
    Heid, Esther
    Greenman, Kevin P.
    Chung, Yunsie
    Li, Shih-Cheng
    Graff, David E.
    Vermeire, Florence H.
    Wu, Haoyang
    Green, William H.
    Mcgill, Charles J.
    [J]. JOURNAL OF CHEMICAL INFORMATION AND MODELING, 2023, 64 (01) : 9 - 17
  • [8] Accurate Physical Property Predictions via Deep Learning
    Hou, Yuanyuan
    Wang, Shiyu
    Bai, Bing
    Chan, H. C. Stephen
    Yuan, Shuguang
    [J]. MOLECULES, 2022, 27 (05):
  • [9] Hyperparameter optimization for machine learning models based on Bayesian optimization
    Wu J.
    Chen X.-Y.
    Zhang H.
    Xiong L.-D.
    Lei H.
    Deng S.-H.
    [J]. Journal of Electronic Science and Technology, 2019, 17 (01) : 26 - 40
  • [10] Electrical sensitivity behavior of a hydrogel composed of polymethacrylic acid/poly(vinyl alcohol)
    Kim, SJ
    Yoon, SG
    Lee, SM
    Lee, SH
    Kim, SI
    [J]. JOURNAL OF APPLIED POLYMER SCIENCE, 2004, 91 (06) : 3613 - 3617