Jointly Learning Convolutional Representations to Compress Radiological Images and Classify Thoracic Diseases in the Compressed Domain

被引:12
作者
Ranjan, Ekagra [1 ]
Paul, Soumava [2 ]
Kapoor, Siddharth [3 ]
Kar, Aupendu [2 ]
Sethuraman, Ramanathan [4 ]
Sheet, Debdoot [2 ]
机构
[1] Indian Inst Technol, Gauhati, India
[2] Indian Inst Technol, Kharagpur, W Bengal, India
[3] Natl Inst Technol, Mangalore, Karnataka, India
[4] Intel Technol India Pvt Ltd, Bangalore, Karnataka, India
来源
ELEVENTH INDIAN CONFERENCE ON COMPUTER VISION, GRAPHICS AND IMAGE PROCESSING (ICVGIP 2018) | 2018年
关键词
Convolutional autoencoder; compression; X-Ray classification; CLASSIFICATION; CANCER;
D O I
10.1145/3293353.3293408
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning models trained in natural images are commonly used for different classification tasks in the medical domain. Generally, very high dimensional medical images are down-sampled by using interpolation techniques before feeding them to deep learning models that are ImageNet compliant and accept only low-resolution images of size 224 x 224 px. This popular technique may lead to the loss of key information thus hampering the classification. Significant pathological features in medical images typically being small sized and highly affected. To combat this problem, we introduce a convolutional neural network (CNN) based classification approach which learns to reduce the resolution of the image using an autoencoder and at the same time classify it using another network, while both the tasks are trained jointly. This algorithm guides the model to learn essential representations from high-resolution images for classification along with reconstruction. We have used the publicly available dataset of chest x-rays to evaluate this approach and have outperformed state-of-the-art on test data. Besides, we have experimented with the effects of different augmentation approaches in this dataset and report baselines using some well known ImageNet class of CNNs.
引用
收藏
页数:8
相关论文
共 33 条
[1]  
[Anonymous], 2010, P ADV NEUR INF PROC
[2]  
[Anonymous], 2017, Radiologist-level pneumonia detection on chest x-rays with deep learning
[3]  
[Anonymous], 2006, 23 INT C MACH LEARN, DOI [10.1145/1143844.1143874, DOI 10.1145/1143844.1143874]
[4]  
[Anonymous], 2018, PROC IBEROAMER C PAT
[5]   Comparison of Deep Learning Approaches for Multi-Label Chest X-Ray Classification [J].
Baltruschat, Ivo M. ;
Nickisch, Hannes ;
Grass, Michael ;
Knopp, Tobias ;
Saalbach, Axel .
SCIENTIFIC REPORTS, 2019, 9 (1)
[6]  
Bengio Y., 2006, ADV NEURAL INFORM PR, V19
[7]  
Cai Jinzheng, 2018, ARXIV180700958
[8]  
Clevert D.-A., 2015, ICLR
[9]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[10]   Dermatologist-level classification of skin cancer with deep neural networks [J].
Esteva, Andre ;
Kuprel, Brett ;
Novoa, Roberto A. ;
Ko, Justin ;
Swetter, Susan M. ;
Blau, Helen M. ;
Thrun, Sebastian .
NATURE, 2017, 542 (7639) :115-+