بهینه سازی نرخ یادگیری معماری U-Net با استفاده از الگوریتم بهینه سازی ملخ به منظور افزایش دقت در بخش بندی تصاویر CT بیماران مبتلا به COVID-19
کد: G-1363
نویسندگان: Alireza Mehravin * ℗, Mostafa Zaare , Reza Mortazavi
زمان بندی: زمان بندی نشده!
برچسب: پردازش تصاویر پزشکی
دانلود: دانلود پوستر
خلاصه مقاله:
خلاصه مقاله
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.
کلمات کلیدی
Covid-19, GOA, Image segmentation, Metaheuristic, U-Net
