Publication: Contrastive Learning Embedded Siamese Neural Network for the Assessment of Fatty Liver
Loading...
Date
2023
Journal Title
IEEE Region 10 Annual International Conference, Proceedings/TENCON
Journal ISSN
Volume Title
Publisher
Institute of Electrical and Electronics Engineers Inc.
Abstract
This paper presents an self-supervised Siamese neural network (SNN) for identification and classification of fatty liver severity. SNN is used for self-supervision tasks for being influenced from model optimization property of supervised and manual annotation property of unsupervised learning. This technique is based on contrastive learning of the joint embedding network which can learn more subtle representations from the medical images for classification task, with just one or few number of labelled images required from each class for training. The efficiency of the proposed model is validated on our dataset of liver ultrasound to classify them into three stages of the fatty liver disease and normal liver. A two-class classifier (normal/grade-I, normal/grade-II and normal/grade-III fatty liver) and four-class classifier (normal, grade-I, grade-II, grade-III fatty liver disease) were trained by minimizing contrastive loss to obtain classification accuracy of 98.91% and 96.84% respectively. � 2023 IEEE.
Description
Keywords
Contrastive learning, Deep learning, Fatty liver, ResNet, Siamese neural network, Ultrasound images