Learning Rate Optimization of U-Net Architecture Using Grasshopper Optimization Algorithm to Enhance Accuracy in CT Image Segmentation of COVID-19 Patients
Code: G-1363
Authors: Alireza Mehravin * ℗, Mostafa Zaare , Reza Mortazavi
Schedule: Not Scheduled!
Download: Download Poster
Abstract:
Abstract
Background and aims: In light of its excellent learning accuracy and rate, rapid data processing, and independence from large databases for network training, the U-Net architecture is a well-known and popular deep learning architecture for image segmentation and feature extraction. Learning rate selection and updating are crucial in network training. As U-Net is a completely nonlinear network, classical mathematical optimization algorithms increase the probability of local optima. When initializing a neural network and selecting its associated hyper-parameters, there are typically two approaches that can be taken. In the first scenario, values are assigned based on empirical knowledge or prior experience with similar models. The second scenario involves using a metaheuristic algorithm to search for optimal solutions within a range of values that we know empirically has a higher probability of network convergence. Method: This analytical research paper used the grasshopper optimization algorithm (GOA) as a metaheuristic approach to optimize the learning rate of U-Net. The network was trained using 256*256 CT images of the lungs of COVID-19 infected and uninfected individuals. A total of 400 CT images were employed as the training dataset, whereas 80 CT images were used as the testing data. Coding was implemented in MATLAB. Results: The optimization of the learning rate enhanced image segmentation accuracy by 2.23%. Conclusion: Iterative metaheuristic algorithms would lead to longer network training times. However, the proposed network optimization method could be very useful when large databases are not available for network training and higher accuracy is preferred over time savings.
Keywords
Covid-19, GOA, Image segmentation, Metaheuristic, U-Net
