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 条
  • [21] An improved synergetic algorithm for image classification
    Electrical Eng. and Computer Science, University of Tasmania, GPO Box 252C 7001, Australia
    Pattern Recogn., 12 (1893-1903):
  • [22] A new algorithm for image recognition and classification based on improved Bag of Features algorithm
    Li, Kangshun
    Wang, Fubin
    Zhang, Lixia
    OPTIK, 2016, 127 (11): : 4736 - 4740
  • [23] An Improved Medical Image Registration Algorithm Based on Mutual Information
    Lan, Tian
    Jiang, Hongbo
    Ding, Yi
    Qin, Zhiguang
    2017 INTERNATIONAL CONFERENCE ON VISION, IMAGE AND SIGNAL PROCESSING (ICVISP), 2017, : 88 - 93
  • [24] An Improved Medical Image Segmentation Algorithm Based On Clustering Techniques
    Li, Xiang-Wei
    Kang, Yu-Xiu
    Zhu, Ya-Ling
    Zheng, Gang
    Wang, Jun-Di
    2017 10TH INTERNATIONAL CONGRESS ON IMAGE AND SIGNAL PROCESSING, BIOMEDICAL ENGINEERING AND INFORMATICS (CISP-BMEI), 2017,
  • [25] An Improved FCM Medical Image Segmentation Algorithm Based on MMTD
    Zhou, Ningning
    Yang, Tingting
    Zhang, Shaobai
    COMPUTATIONAL AND MATHEMATICAL METHODS IN MEDICINE, 2014, 2014
  • [26] Medical image segmentation based on improved Ostu algorithm and regional growth algorithm
    Computing Center, Northeastern University, Shenyang 110004, China
    Dongbei Daxue Xuebao, 2006, 4 (398-401):
  • [27] An Improved Algorithm for Medical Image Segmentation
    Huang, Ting-lei
    Bai, Xue
    SECOND INTERNATIONAL CONFERENCE ON GENETIC AND EVOLUTIONARY COMPUTING: WGEC 2008, PROCEEDINGS, 2008, : 289 - 292
  • [28] An improved segmentation algorithm for medical image
    Zhang, Hao
    Yang, Jiongzhao
    Zheng, Jiangyan
    Lou, Guohuan
    Han, Yanfei
    International Journal of Applied Mathematics and Statistics, 2013, 46 (16): : 142 - 150
  • [29] An Improved Algorithm for Medical Image Compression
    Moorthi, M.
    Amutha, R.
    GLOBAL TRENDS IN INFORMATION SYSTEMS AND SOFTWARE APPLICATIONS, PT 2, 2012, 270 : 451 - +
  • [30] Real Time Image Classification and Detection of Potholes using Adam's Optimizer with CNN Framework
    Sathya, R.
    Nikhil, Muppala
    Srinath, K. S. S. Shanmukha
    Reddy, Mula Ranpal
    INTERNATIONAL JOURNAL OF EARLY CHILDHOOD SPECIAL EDUCATION, 2022, 14 (02) : 3714 - 3721