Detect Glaucoma with Image Segmentation and Transfer Learning

被引:2
作者
Wu, Lianyi [1 ]
Liu, Yiming [1 ]
Shi, Yelin [1 ]
Sheng, Bin [1 ]
Li, Ping [2 ]
Bi, Lei [3 ]
Kim, Jinman [3 ]
机构
[1] Shanghai Jiao Tong Univ, Dept Comp Sci & Engn, Shanghai, Peoples R China
[2] Macau Univ Sci & Technol, Fac Informat Technol, Taipa, Macau, Peoples R China
[3] Univ Sydney, Sch Informat Technol, Sydney, NSW, Australia
来源
PROCEEDINGS OF THE 32ND INTERNATIONAL CONFERENCE ON COMPUTER ANIMATION AND SOCIAL AGENTS (CASA 2019) | 2019年
基金
中国国家自然科学基金;
关键词
glaucoma; segmentation; transfer learning; SegNet; ADDA;
D O I
10.1145/3328756.3328771
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
In this paper, we aim to automatically detect glaucoma via deep learning. To do that, we need to calculate the cup-to-disc ratio (CDR) on fine segmented retina images. To get precise segmentation, we implemented SegNet together with adversarial discriminative domain adaptation (ADDA), the former is a famous artificial neural network with encoder-decoder architecture used in image segmentation area and the latter is a transfer learning method for domain adaptation. We are the first to combine them together to detect glaucoma on test dataset which have different brightness from our training dataset. We thoroughly evaluated the proposed method with various loss functions, normal cross entropy loss, weighted cross entropy loss and dice coefficient loss included. And we show that dice loss is the best for this task. Last but not least, our experiments on transfer learning have shown that our ADDA method reduces the mean square error (MSE) between the CDR of our segmentation and annotations greatly.
引用
收藏
页码:37 / 40
页数:4
相关论文
共 8 条
[1]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[2]   SCN: Switchable Context Network for Semantic Segmentation of RGB-D Images [J].
Lin, Di ;
Zhang, Ruimao ;
Ji, Yuanfeng ;
Li, Ping ;
Huang, Hui .
IEEE TRANSACTIONS ON CYBERNETICS, 2020, 50 (03) :1120-1131
[3]   Compact Video Synopsis via Global Spatiotemporal Optimization [J].
Nie, Yongwei ;
Xiao, Chunxia ;
Sun, Hanqiu ;
Li, Ping .
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS, 2013, 19 (10) :1664-1676
[4]  
Pavithra G., 2017, IEEE INT C CURR TREN
[5]  
Sheng B., 2018, IEEE T VIS COMPUT GR, P1
[6]   Intrinsic Image Decomposition with Step and Drift Shading Separation [J].
Sheng, Bin ;
Li, Ping ;
Jin, Yuxi ;
Tan, Ping ;
Lee, Tong-Yee .
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS, 2020, 26 (02) :1332-1346
[7]   Retinal Vessel Segmentation Using Minimum Spanning Superpixel Tree Detector [J].
Sheng, Bin ;
Li, Ping ;
Mo, Shuangjia ;
Li, Huating ;
Hou, Xuhong ;
Wu, Qiang ;
Qin, Jing ;
Fang, Ruogu ;
Feng, David Dagan .
IEEE TRANSACTIONS ON CYBERNETICS, 2019, 49 (07) :2707-2719
[8]   Adversarial Discriminative Domain Adaptation [J].
Tzeng, Eric ;
Hoffman, Judy ;
Saenko, Kate ;
Darrell, Trevor .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2962-2971