arXiv · 2208.06917
MTCSNN: Multi-task Clinical Siamese Neural Network for Diabetic Retinopathy Severity Prediction
Abstract
Diabetic Retinopathy (DR) has become one of the leading causes of vision impairment in working-aged people and is a severe problem worldwide. However, most of the works ignored the ordinal information of labels. In this project, we propose a novel design MTCSNN, a Multi-task Clinical Siamese Neural Network for Diabetic Retinopathy severity prediction task. The novelty of this project is to utilize the ordinal information among labels and add a new regression task, which can help the model learn more discriminative feature embedding for fine-grained classification tasks. We perform comprehensive experiments over the RetinaMNIST, comparing MTCSNN with other models like ResNet-18, 34, 50. Our results indicate that MTCSNN outperforms the benchmark models in terms of AUC and accuracy on the test dataset.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Chao Feng, Jui Po Hung, Aishan Li, Jieping Yang, Xinyu Zhang. 2022-08-14. MTCSNN: Multi-task Clinical Siamese Neural Network for Diabetic Retinopathy Severity Prediction. https://arxiv.org/abs/2208.06917
Cite the original work for its findings. Save a collection to share your selection of sources.