Classification of neck tissues in OCT images by using convolutional neural network

Abstract

Identification and classification of surrounding neck tissues are very important in thyroid surgery. The advantages of optical coherence tomography (OCT), high resolution, non-invasion, and non-destruction make it have great potential in identifying different neck tissues during thyroidectomy. We studied the automatic classification for neck tissues in OCT images based on convolutional neural network in this paper. OCT images of five kinds of neck tissues were collected firstly by our home-made swept source (SS-OCT) system, and a dataset was built for neural network training. Three image classification neural networks: LeNet, VGGNet, and ResNet, were used to train and test the dataset. The impact of transfer learning on the classification of neck tissue OCT images was also studied. Through the comparison of accuracy, it was found that ResNet has the best classification accuracy among the three networks. In addition, transfer learning did not significantly improve the accuracy, but it can somewhat accelerate the convergence of the network and shorten the network training time.

Keywords: Classification; Deep learning; Neck tissue; Optical coherence tomography.