GraphXNET- Chest X-Ray Classification Under Extreme Minimal Supervision

被引:37
作者
Aviles-Rivero, Angelica I. [1 ,2 ]
Papadakis, Nicolas [3 ]
Li, Ruoteng [4 ]
Sellars, Philip [1 ,2 ]
Fan, Qingnan [5 ]
Tan, Robby T. [4 ,6 ]
Schoenlieb, Carola-Bibiane [1 ,2 ]
机构
[1] Univ Cambridge, Fac Math, DPMMS, Cambridge, England
[2] Univ Cambridge, Fac Math, DAMPT, Cambridge, England
[3] Univ Bordeaux, CNRS, Talence, France
[4] Natl Univ Singapore, Singapore, Singapore
[5] Stanford Univ, Stanford, CA 94305 USA
[6] Yale NUS Coll, Singapore, Singapore
来源
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT VI | 2019年 / 11769卷
基金
欧盟地平线“2020”; 英国工程与自然科学研究理事会;
关键词
Semi-supervised learning; Classification; Chest X-Ray; Graphs; Transductive learning;
D O I
10.1007/978-3-030-32226-7_56
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The task of classifying X-ray data is a problem of both theoretical and clinical interest. Whilst supervised deep learning methods rely upon huge amounts of labelled data, the critical problem of achieving a good classification accuracy when an extremely small amount of labelled data is available has yet to be tackled. In this work, we introduce a novel semi-supervised framework for X-ray classification which is based on a graph-based optimisation model. To the best of our knowledge, this is the first method that exploits graph-based semi-supervised learning for X-ray data classification. Furthermore, we introduce a new multi-class classification functional with carefully selected class priors which allows for a smooth solution that strengthens the synergy between the limited number of labels and the huge amount of unlabelled data. We demonstrate, through a set of numerical and visual experiments, that our method produces highly competitive results on the ChestX-ray14 data set whilst drastically reducing the need for annotated data.
引用
收藏
页码:504 / 512
页数:9
相关论文
共 20 条
[1]  
[Anonymous], 2013, Advances in Neural Information Processing Systems
[2]  
Bar Y, 2015, I S BIOMED IMAGING, P294, DOI 10.1109/ISBI.2015.7163871
[3]   Laplacian eigenmaps for dimensionality reduction and data representation [J].
Belkin, M ;
Niyogi, P .
NEURAL COMPUTATION, 2003, 15 (06) :1373-1396
[4]  
Bresson X., 2013, ADV NEURAL INFORM PR
[5]   Understanding and Confronting Our Mistakes: The Epidemiology of Error in Radiology and Strategies for Error Reduction [J].
Bruno, Michael A. ;
Walker, Eric A. ;
Abujudeh, Hani H. .
RADIOGRAPHICS, 2015, 35 (06) :1668-1676
[6]  
Buhler T., INT C MACH LEARN, P81
[7]   A First-Order Primal-Dual Algorithm for Convex Problems with Applications to Imaging [J].
Chambolle, Antonin ;
Pock, Thomas .
JOURNAL OF MATHEMATICAL IMAGING AND VISION, 2011, 40 (01) :120-145
[8]  
Chen HB, 2012, LECT NOTES COMPUT SC, V7512, P297, DOI [10.1109/TMI.2013.2259248, 10.1007/978-3-642-33454-2_37]
[9]  
Dodero L, 2014, LECT NOTES COMPUT SC, V8674, P708, DOI 10.1007/978-3-319-10470-6_88
[10]   Rayleigh quotient minimization for absolutely one-homogeneous functionals [J].
Feld, Tal ;
Aujol, Jean-Francois ;
Gilboa, Guy ;
Papadakis, Nicolas .
INVERSE PROBLEMS, 2019, 35 (06)