An Improved Medical Image Classification Algorithm Based on Adam Optimizer

被引:2
|
作者
Sun, Haijing [1 ]
Zhou, Wen [2 ]
Yang, Jiapeng [2 ]
Shao, Yichuan [1 ]
Xing, Lei [3 ]
Zhao, Qian [4 ]
Zhang, Le [1 ]
机构
[1] Shenyang Univ, Sch Intelligent Sci & Engn, Shenyang 110044, Peoples R China
[2] Shenyang Univ, Sch Informat Engn, Shenyang 110044, Peoples R China
[3] Univ Surrey, Sch Chem & Chem Engn, Guildford GU2 7XH, England
[4] Shenyang Univ Technol, Sch Sci, Shenyang 110044, Peoples R China
关键词
deep learning; Adam algorithm; gradient cropping; linear interpolation; periodic adjustment of learning rate;
D O I
10.3390/math12162509
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Due to the complexity and illegibility of medical images, it brings inconvenience and difficulty to the diagnosis of medical personnel. To address these issues, an optimization algorithm called GSL(Gradient sine linear) based on Adam algorithm improvement is proposed in this paper, which introduces gradient pruning strategy, periodic adjustment of learning rate, and linear interpolation strategy. The gradient trimming technique can scale the gradient to prevent gradient explosion, while the periodic adjustment of the learning rate and linear interpolation strategy adjusts the learning rate according to the characteristics of the sinusoidal function, accelerating the convergence while reducing the drastic parameter fluctuations, improving the efficiency and stability of training. The experimental results show that compared to the classic Adam algorithm, this algorithm can demonstrate better classification accuracy, the GSL algorithm achieves an accuracy of 78% and 75.2% on the MobileNetV2 network and ShuffleNetV2 network under the Gastroenterology dataset; and on the MobileNetV2 network and ShuffleNetV2 network under the Glaucoma dataset, an accuracy of 84.72% and 83.12%. The GSL optimizer achieved significant performance improvement on various neural network structures and datasets, proving its effectiveness and practicality in the field of deep learning, and also providing new ideas and methods for solving the difficulties in medical image recognition.
引用
收藏
页数:14
相关论文
共 50 条
  • [1] A Gastrointestinal Image Classification Method Based on Improved Adam Algorithm
    Sun, Haijing
    Cui, Jiaqi
    Shao, Yichuan
    Yang, Jiapeng
    Xing, Lei
    Zhao, Qian
    Zhang, Le
    MATHEMATICS, 2024, 12 (16)
  • [2] An Improved Adam's Algorithm for Stomach Image Classification
    Sun, Haijing
    Yu, Hao
    Shao, Yichuan
    Wang, Jiantao
    Xing, Lei
    Zhang, Le
    Zhao, Qian
    ALGORITHMS, 2024, 17 (07)
  • [3] SPGD Algorithm Optimization Based on Adam Optimizer
    Fang, Zhou
    Xu, Xiangxiang
    Li, Xin
    Yang, Huizhen
    Gong, Chenglong
    AOPC 2020: OPTICAL SENSING AND IMAGING TECHNOLOGY, 2020, 11567
  • [4] Image Classification using DNN with an Improved Optimizer
    Saqib, Nazmus
    Rafiquzzaman, G. M.
    2020 IEEE REGION 10 SYMPOSIUM (TENSYMP) - TECHNOLOGY FOR IMPACTFUL SUSTAINABLE DEVELOPMENT, 2020, : 31 - 35
  • [5] Improved CNN Based on Batch Normalization and Adam Optimizer
    Ogundokun, Roseline Oluwaseun
    Maskeliunas, Rytis
    Misra, Sanjay
    Damasevicius, Robertas
    COMPUTATIONAL SCIENCE AND ITS APPLICATIONS - ICCSA 2022 WORKSHOPS, PART V, 2022, 13381 : 593 - 604
  • [6] CNN based Traffic Sign Classification using Adam Optimizer
    Mehta, Smit
    Paunwala, Chirag
    Vaidya, Bhaumik
    PROCEEDINGS OF THE 2019 INTERNATIONAL CONFERENCE ON INTELLIGENT COMPUTING AND CONTROL SYSTEMS (ICCS), 2019, : 1293 - 1298
  • [7] Image Classification Based on Improved Random Forest Algorithm
    Man, Weishi
    Ji, Yuanyuan
    Zhang, Zhiyu
    2018 IEEE 3RD INTERNATIONAL CONFERENCE ON CLOUD COMPUTING AND BIG DATA ANALYSIS (ICCCBDA), 2018, : 346 - 350
  • [8] Medical image segmentation based on improved watershed algorithm
    Shen, Tongping
    Wang, Yuanmao
    PROCEEDINGS OF 2018 IEEE 3RD ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC 2018), 2018, : 1695 - 1698
  • [9] Medical Image Enhancement Algorithm Based on Improved Contourlet
    Guo Qi
    Shen Shu-Ting
    Ren Ping-Chuan
    JOURNAL OF MEDICAL IMAGING AND HEALTH INFORMATICS, 2017, 7 (05) : 962 - 967
  • [10] MEDICAL IMAGE SEGMENTATION BASED ON IMPROVED WATERSHED ALGORITHM
    Li Jing-Yu
    Jin Cheng
    Mu Wei-Bin
    Geng Kui
    Zhang Yan
    2011 INTERNATIONAL CONFERENCE ON COMPUTER AND COMPUTATIONAL INTELLIGENCE (ICCCI 2011), 2012, : 475 - 481