Correcting Faulty Segmentation Labels

Abstract

In this study, a convolutional neural network is developed to correct errors in a database containing some label errors. The correction of errors is tested by comparing the output of the network to the output of a previous algorithm in order to investigate the use of neural networks for error correction problems. The database used was a medical database in which the labels were segmentations of pills wrapped in bags that were generated by the previous algorithm. The results suggested a convolutional network containing ve residual layers was unable to successfully correct errors generated by the previous algorithm because the network made errors that the original algorithm did not make. These results however are doubtful since the network did show signs of error correction. In further research that concerns how neural networks handle error correction in image segmentations, focus on other network structures such as U-net is advised.