Scheduling Techniques for Liver Segmentation: ReduceLRonPlateau Vs OneCycleLR

02/13/2022
by   Ayman Al-Kababji, et al.
0

Machine learning and computer vision techniques have influenced many fields including the biomedical one. The aim of this paper is to investigate the important concept of schedulers in manipulating the learning rate (LR), for the liver segmentation task, throughout the training process, focusing on the newly devised OneCycleLR against the ReduceLRonPlateau. A dataset, published in 2018 and produced by the Medical Segmentation Decathlon Challenge organizers, called Task 8 Hepatic Vessel (MSDC-T8) has been used for testing and validation. The reported results that have the same number of maximum epochs (75), and are the average of 5-fold cross-validation, indicate that ReduceLRonPlateau converges faster while maintaining a similar or even better loss score on the validation set when compared to OneCycleLR. The epoch at which the peak LR occurs perhaps should be made early for the OneCycleLR such that the super-convergence feature can be observed. Moreover, the overall results outperform the state-of-the-art results from the researchers who published the liver masks for this dataset. To conclude, both schedulers are suitable for medical segmentation challenges, especially the MSDC-T8 dataset, and can be used confidently in rapidly converging the validation loss with a minimal number of epochs.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
10/16/2019

End-to-End Cascaded U-Nets with a Localization Network for Kidney Tumor Segmentation

Kidney tumor segmentation emerges as a new frontier of computer vision i...
research
09/22/2020

Improving Medical Annotation Quality to Decrease Labeling Burden Using Stratified Noisy Cross-Validation

As machine learning has become increasingly applied to medical imaging d...
research
04/27/2020

OR-UNet: an Optimized Robust Residual U-Net for Instrument Segmentation in Endoscopic Images

Segmentation of endoscopic images is an essential processing step for co...
research
01/29/2017

The HASYv2 dataset

This paper describes the HASYv2 dataset. HASY is a publicly available, f...
research
08/04/2019

Automatic segmentation of kidney and liver tumors in CT images

Automatic segmentation of hepatic lesions in computed tomography (CT) im...
research
04/25/2023

Performance Evaluation of Regression Models in Predicting the Cost of Medical Insurance

The study aimed to evaluate the regression models' performance in predic...
research
11/05/2016

Comparing learning algorithms in neural network for diagnosing cardiovascular disease

Today data mining techniques are exploited in medical science for diagno...

Please sign up or login with your details

Forgot password? Click here to reset