Weakly Supervised Lesion Localization With Probabilistic-CAM Pooling

05/29/2020
by   Wenwu Ye, et al.
11

Localizing thoracic diseases on chest X-ray plays a critical role in clinical practices such as diagnosis and treatment planning. However, current deep learning based approaches often require strong supervision, e.g. annotated bounding boxes, for training such systems, which is infeasible to harvest in large-scale. We present Probabilistic Class Activation Map (PCAM) pooling, a novel global pooling operation for lesion localization with only image-level supervision. PCAM pooling explicitly leverages the excellent localization ability of CAM during training in a probabilistic fashion. Experiments on the ChestX-ray14 dataset show a ResNet-34 model trained with PCAM pooling outperforms state-of-the-art baselines on both the classification task and the localization task. Visual examination on the probability maps generated by PCAM pooling shows clear and sharp boundaries around lesion regions compared to the localization heatmaps generated by CAM. PCAM pooling is open sourced at https://github.com/jfhealthcare/Chexpert.

READ FULL TEXT
research
09/05/2022

A Benchmark for Weakly Semi-Supervised Abnormality Localization in Chest X-Rays

Accurate abnormality localization in chest X-rays (CXR) can benefit the ...
research
07/16/2018

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

Chest X-rays is one of the most commonly available and affordable radiol...
research
10/22/2020

High resolution weakly supervised localization architectures for medical images

In medical imaging, Class-Activation Map (CAM) serves as the main explai...
research
09/21/2018

Global Weighted Average Pooling Bridges Pixel-level Localization and Image-level Classification

In this work, we first tackle the problem of simultaneous pixel-level lo...
research
07/19/2017

Object-Extent Pooling for Weakly Supervised Single-Shot Localization

In the face of scarcity in detailed training annotations, the ability to...
research
06/09/2020

Rethinking Localization Map: Towards Accurate Object Perception with Self-Enhancement Maps

Recently, remarkable progress has been made in weakly supervised object ...
research
12/14/2015

Learning Deep Features for Discriminative Localization

In this work, we revisit the global average pooling layer proposed in [1...

Please sign up or login with your details

Forgot password? Click here to reset