[go: up one dir, main page]
More Web Proxy on the site http://driver.im/ skip to main content
10.1145/3233547.3233573acmconferencesArticle/Chapter ViewAbstractPublication PagesbcbConference Proceedingsconference-collections
research-article

Weakly Supervised Deep Learning for Thoracic Disease Classification and Localization on Chest X-rays

Published: 15 August 2018 Publication History

Abstract

Chest X-rays is one of the most commonly available and affordable radiological examinations in clinical practice. While detecting thoracic diseases on chest X-rays is still a challenging task for machine intelligence, due to 1) the highly varied appearance of lesion areas on X-rays from patients of different thoracic disease and 2) the shortage of accurate pixel-level annotations by radiologists for model training. Existing machine learning methods are unable to deal with the challenge that thoracic diseases usually happen in localized disease-specific areas. In this article, we propose a weakly supervised deep learning framework equipped with squeeze-and-excitation blocks, multi-map transfer and max-min pooling for classifying common thoracic diseases as well as localizing suspicious lesion regions on chest X-rays. The comprehensive experiments and discussions are performed on the ChestX-ray14 dataset. Both numerical and visual results have demonstrated the effectiveness of proposed model and its better performance against the state-of-the-art pipelines.

References

[1]
Alan R Aronson and François-Michel Lang. 2010. An overview of MetaMap: historical perspective and recent advances. Journal of the American Medical Informatics Association 17, 3 (2010), 229--236.
[2]
Wendy Webber Chapman, Marcelo Fizman, Brian E Chapman, and Peter J Haug. 2001. A comparison of classification algorithms to automatically identify chest X-ray reports that support pneumonia. Journal of biomedical informatics 34, 1 (2001), 4--14.
[3]
C. Chen, V. C. Kavuri, X. Wang, R. Li, H. Liu, and J. Huang. 2015. Multi-frequency diffuse optical tomography for cancer detection. In 2015 IEEE 12th International Symposium on Biomedical Imaging (ISBI). 67--70.
[4]
David J Crandall and Daniel P Huttenlocher. 2006. Weakly supervised learning of part-based spatial models for visual object recognition. In European conference on computer vision. Springer, 16--29.
[5]
Dina Demner-Fushman, Marc D Kohli, Marc B Rosenman, Sonya E Shooshan, Laritza Rodriguez, Sameer Antani, George R Thoma, and Clement J McDonald. 2015. Preparing a collection of radiology examinations for distribution and retrieval. Journal of the American Medical Informatics Association 23, 2 (2015), 304--310.
[6]
Jia Deng, Wei Dong, Richard Socher, Li-Jia Li, Kai Li, and Li Fei-Fei. 2009. Imagenet: A large-scale hierarchical image database. In Computer Vision and Pattern Recognition, 2009. CVPR 2009. IEEE Conference on. IEEE, 248--255.
[7]
Thibaut Durand, Taylor Mordan, Nicolas Thome, and Matthieu Cord. 2017. Wildcat: Weakly supervised learning of deep convnets for image classification, pointwise localization and segmentation. In IEEE Conference on Computer Vision and Pattern Recognition (CVPR 2017).
[8]
Yu Gordienko, Peng Gang, Jiang Hui, Wei Zeng, Yu Kochura, O Alienin, O Rokovyi, and S Stirenko. 2017. Deep Learning with Lung Segmentation and Bone Shadow Exclusion Techniques for Chest X-Ray Analysis of Lung Cancer. arXiv preprint arXiv:1712.07632 (2017).
[9]
Qingji Guan, Yaping Huang, Zhun Zhong, Zhedong Zheng, Liang Zheng, and Yi Yang. 2018. Diagnose like a radiologist: Attention guided convolutional neural network for thorax disease classification. arXiv preprint arXiv:1801.09927 (2018).
[10]
Sebastian Guendel, Sasa Grbic, Bogdan Georgescu, Kevin Zhou, Ludwig Ritschl, Andreas Meier, and Dorin Comaniciu. 2018. Learning to recognize abnormalities in chest x-rays with location-aware dense networks. arXiv preprint arXiv:1803.04565 (2018).
[11]
Varun Gulshan, Lily Peng, Marc Coram, Martin C Stumpe, Derek Wu, Arunachalam Narayanaswamy, Subhashini Venugopalan, Kasumi Widner, Tom Madams, Jorge Cuadros, et al. 2016. Development and validation of a deep learning algorithm for detection of diabetic retinopathy in retinal fundus photographs. JAMA 316, 22 (2016), 2402--2410.
[12]
Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. 2016. Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition. 770--778.
[13]
Jie Hu, Li Shen, and Gang Sun. 2017. Squeeze-and-excitation networks. arXiv preprint arXiv:1709.01507 (2017).
[14]
Gao Huang, Zhuang Liu, Kilian Q Weinberger, and Laurens van der Maaten. 2017. Densely connected convolutional networks. In Proceedings of the IEEE conference on computer vision and pattern recognition, Vol. 1. 3.
[15]
Mohammad Tariqul Islam, Md Abdul Aowal, Ahmed Tahseen Minhaz, and Khalid Ashraf. 2017. Abnormality Detection and Localization in Chest X-Rays using Deep Convolutional Neural Networks. arXiv preprint arXiv:1705.09850 (2017).
[16]
Simon Jégou, Michal Drozdzal, David Vazquez, Adriana Romero, and Yoshua Bengio. 2017. The one hundred layers tiramisu: Fully convolutional densenets for semantic segmentation. In Computer Vision and Pattern Recognition Workshops (CVPRW), 2017 IEEE Conference on. IEEE, 1175--1183.
[17]
Diederik P Kingma and Jimmy Ba. 2014. Adam: A Method for Stochastic Optimization. Computer Science (2014).
[18]
Alex Krizhevsky, Ilya Sutskever, and Geoffrey E Hinton. 2012. Imagenet classification with deep convolutional neural networks. In Advances in neural information processing systems. 1097--1105.
[19]
Robert Leaman, Ritu Khare, and Zhiyong Lu. 2015. Challenges in clinical natural language processing for automated disorder normalization. Journal of biomedical informatics 57 (2015), 28--37.
[20]
Ruoyu Li and Junzhou Huang. 2015. Fast regions-of-interest detection in whole slide histopathology images. In International Workshop on Patch-based Techniques in Medical Imaging. Springer, 120--127.
[21]
Ruoyu Li, Yeqing Li, Ruogu Fang, Shaoting Zhang, Hao Pan, and Junzhou Huang. 2015. Fast preconditioning for accelerated multi-contrast MRI reconstruction. In International Conference on Medical Image Computing and Computer-Assisted Intervention. Springer, 700--707.
[22]
Zhe Li, Chong Wang, Mei Han, Yuan Xue, Wei Wei, Li-Jia Li, and Fei-Fei Li. 2017. Thoracic Disease Identification and Localization with Limited Supervision. arXiv preprint arXiv:1711.06373 (2017).
[23]
Yun Liu, Krishna Gadepalli, Mohammad Norouzi, George E Dahl, Timo Kohlberger, Aleksey Boyko, Subhashini Venugopalan, Aleksei Timofeev, Philip Q Nelson, Greg S Corrado, et al. 2017. Detecting cancer metastases on gigapixel pathology images. arXiv preprint arXiv:1703.02442 (2017).
[24]
Jonathan Long, Evan Shelhamer, and Trevor Darrell. 2015. Fully convolutional networks for semantic segmentation. In Proceedings of the IEEE conference on computer vision and pattern recognition. 3431--3440.
[25]
Pranav Rajpurkar, Jeremy Irvin, Kaylie Zhu, Brandon Yang, Hershel Mehta, Tony Duan, Daisy Ding, Aarti Bagul, Curtis Langlotz, Katie Shpanskaya, et al. 2017. CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning. arXiv preprint arXiv:1711.05225 (2017).
[26]
Jonathan Rubin, Deepan Sanghavi, Claire Zhao, Kathy Lee, Ashequl Qadir, and Minnan Xu-Wilson. 2018. Large Scale Automated Reading of Frontal and Lateral Chest X-Rays using Dual Convolutional Neural Networks. arXiv preprint arXiv:1804.07839 (2018).
[27]
Arnaud Arindra Adiyoso Setio, Alberto Traverso, Thomas De Bel, Moira SN Berens, Cas van den Bogaard, Piergiorgio Cerello, Hao Chen, Qi Dou, Maria Evelina Fantacci, Bram Geurts, et al. 2017. Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challenge. Medical image analysis 42 (2017), 1--13.
[28]
Junji Shiraishi, Shigehiko Katsuragawa, Junpei Ikezoe, Tsuneo Matsumoto, Takeshi Kobayashi, Ken-ichi Komatsu, Mitate Matsui, Hiroshi Fujita, Yoshie Kodera, and Kunio Doi. 2000. Development of a digital image database for chest radiographs with and without a lung nodule: receiver operating characteristic analysis of radiologists' detection of pulmonary nodules. American Journal of Roentgenology 174, 1 (2000), 71--74.
[29]
Karen Simonyan and Andrew Zisserman. 2014. Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556 (2014).
[30]
Christian Szegedy, Wei Liu, Yangqing Jia, Pierre Sermanet, Scott Reed, Dragomir Anguelov, Dumitru Erhan, Vincent Vanhoucke, Andrew Rabinovich, et al. 2015. Going deeper with convolutions. Cvpr.
[31]
Lorenzo Torresani. 2014. Weakly supervised learning. In Computer Vision. Springer, 883--885.
[32]
Bram Van Ginneken, Mikkel B Stegmann, and Marco Loog. 2006. Segmentation of anatomical structures in chest radiographs using supervised methods: a comparative study on a public database. Medical image analysis 10, 1 (2006), 19--40.
[33]
Xiaosong Wang, Yifan Peng, Le Lu, Zhiyong Lu, Mohammadhadi Bagheri, and Ronald M Summers. 2017. Chestx-ray8: Hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases. In 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 3462--3471.
[34]
Li Yao, Eric Poblenz, Dmitry Dagunts, Ben Covington, Devon Bernard, and Kevin Lyman. 2017. Learning to diagnose from scratch by exploiting dependencies among labels. arXiv preprint arXiv:1710.10501 (2017).
[35]
Li Yao, Jordan Prosky, Eric Poblenz, Ben Covington, and Kevin Lyman. 2018. Weakly Supervised Medical Diagnosis and Localization from Multiple Resolutions. arXiv preprint arXiv:1803.07703 (2018).
[36]
Xujiong Ye, Xinyu Lin, Jamshid Dehmeshki, Greg Slabaugh, and Gareth Beddoe. 2009. Shape-based computer-aided detection of lung nodules in thoracic CT images. IEEE Transactions on Biomedical Engineering 56, 7 (2009), 1810--1820.
[37]
Xinliang Zhu, Jiawen Yao, Feiyun Zhu, and Junzhou Huang. 2017. Wsisa: Making survival prediction from whole slide histopathological images. In IEEE Conference on Computer Vision and Pattern Recognition. 7234--7242.

Cited By

View all
  • (2025)Weakly Supervised Object Detection in Chest X-Rays With Differentiable ROI Proposal Networks and Soft ROI PoolingIEEE Transactions on Medical Imaging10.1109/TMI.2024.343501544:1(221-231)Online publication date: Jan-2025
  • (2025)HydraViT: Adaptive multi-branch transformer for multi-label disease classification from Chest X-ray imagesBiomedical Signal Processing and Control10.1016/j.bspc.2024.106959100(106959)Online publication date: Feb-2025
  • (2025)ItpCtrl-AI: End-to-end interpretable and controllable artificial intelligence by modeling radiologists’ intentionsArtificial Intelligence in Medicine10.1016/j.artmed.2024.103054160(103054)Online publication date: Feb-2025
  • Show More Cited By

Index Terms

  1. Weakly Supervised Deep Learning for Thoracic Disease Classification and Localization on Chest X-rays

      Recommendations

      Comments

      Please enable JavaScript to view thecomments powered by Disqus.

      Information & Contributors

      Information

      Published In

      cover image ACM Conferences
      BCB '18: Proceedings of the 2018 ACM International Conference on Bioinformatics, Computational Biology, and Health Informatics
      August 2018
      727 pages
      ISBN:9781450357944
      DOI:10.1145/3233547
      Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

      Sponsors

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      Published: 15 August 2018

      Permissions

      Request permissions for this article.

      Check for updates

      Author Tags

      1. chest x-ray
      2. computer-aided diagnosis
      3. weakly-supervised learning

      Qualifiers

      • Research-article

      Funding Sources

      • US National Science Foundation
      • NSF CAREER

      Conference

      BCB '18
      Sponsor:

      Acceptance Rates

      BCB '18 Paper Acceptance Rate 46 of 148 submissions, 31%;
      Overall Acceptance Rate 254 of 885 submissions, 29%

      Contributors

      Other Metrics

      Bibliometrics & Citations

      Bibliometrics

      Article Metrics

      • Downloads (Last 12 months)50
      • Downloads (Last 6 weeks)6
      Reflects downloads up to 14 Jan 2025

      Other Metrics

      Citations

      Cited By

      View all
      • (2025)Weakly Supervised Object Detection in Chest X-Rays With Differentiable ROI Proposal Networks and Soft ROI PoolingIEEE Transactions on Medical Imaging10.1109/TMI.2024.343501544:1(221-231)Online publication date: Jan-2025
      • (2025)HydraViT: Adaptive multi-branch transformer for multi-label disease classification from Chest X-ray imagesBiomedical Signal Processing and Control10.1016/j.bspc.2024.106959100(106959)Online publication date: Feb-2025
      • (2025)ItpCtrl-AI: End-to-end interpretable and controllable artificial intelligence by modeling radiologists’ intentionsArtificial Intelligence in Medicine10.1016/j.artmed.2024.103054160(103054)Online publication date: Feb-2025
      • (2024)MULTI-LABEL CLASSIFICATION OF THORACIC DISEASES USING STRUCTURED DEEP LEARNING FRAMEWORKBiomedical Engineering: Applications, Basis and Communications10.4015/S101623722450006636:02Online publication date: 16-Feb-2024
      • (2024)Developing New Fully Connected Layers for Convolutional Neural Networks with Hyperparameter Optimization for Improved Multi-Label Image ClassificationMathematics10.3390/math1206080612:6(806)Online publication date: 8-Mar-2024
      • (2024)Advanced Multi-Label Image Classification Techniques Using Ensemble MethodsMachine Learning and Knowledge Extraction10.3390/make60200606:2(1281-1297)Online publication date: 7-Jun-2024
      • (2024)A Critical Analysis of Deep Semi-Supervised Learning Approaches for Enhanced Medical Image ClassificationInformation10.3390/info1505024615:5(246)Online publication date: 24-Apr-2024
      • (2024)Toward explainable AI in radiology: Ensemble-CAM for effective thoracic disease localization in chest X-ray images using weak supervised learningFrontiers in Big Data10.3389/fdata.2024.13664157Online publication date: 2-May-2024
      • (2024)I-AI: A Controllable & Interpretable AI System for Decoding Radiologists’ Intense Focus for Accurate CXR Diagnoses2024 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)10.1109/WACV57701.2024.00767(7835-7844)Online publication date: 3-Jan-2024
      • (2024)Artificially Generated Visual Scanpath Improves Multilabel Thoracic Disease Classification in Chest X-Ray ImagesIEEE Transactions on Instrumentation and Measurement10.1109/TIM.2024.342859173(1-11)Online publication date: 2024
      • Show More Cited By

      View Options

      Login options

      View options

      PDF

      View or Download as a PDF file.

      PDF

      eReader

      View online with eReader.

      eReader

      Media

      Figures

      Other

      Tables

      Share

      Share

      Share this Publication link

      Share on social media