Weakly supervised fully convolutional network for PET lesion segmentation

S. Afshari; A. BenTaieb; Z. MiriKharaji; G. Hamarneh

doi:10.1117/12.2512274

15 March 2019 Weakly supervised fully convolutional network for PET lesion segmentation

S. Afshari, A. BenTaieb, Z. MiriKharaji, G. Hamarneh

Proceedings Volume 10949, Medical Imaging 2019: Image Processing; 109491K (2019) https://doi.org/10.1117/12.2512274
Event: SPIE Medical Imaging, 2019, San Diego, California, United States

Abstract

The effort involved in creating accurate ground truth segmentation maps hinders advances in machine learning approaches to tumor delineation in clinical positron emission tomography (PET) scans. To address this challenge, we propose a fully convolutional network (FCN) model to delineate tumor volumes from PET scans automatically while relying on weak annotations in the form of bounding boxes (without delineations) around tumor lesions. To achieve this, we propose a novel loss function that dynamically combines a supervised component, designed to leverage the training bounding boxes, with an unsupervised component, inspired by the Mumford-Shah piecewise constant level-set image segmentation model. The model is trained end-to-end with the proposed differentiable loss function and is validated on a public clinical PET dataset of head and neck tumors. Using only bounding box annotations as supervision, the model achieves competitive results with state-of-the-art supervised and semiautomatic segmentation approaches. Our proposed approach improves the Dice similarity by approximately 30% and reduces the unsigned distance error by approximately 7 mm compared to a model trained with only bounding boxes (weak supervision). Also, after the post-processing step (morphological operations), our weak supervision approach differs only 7% in terms of the Dice similarity from the quality of the fully supervised model, for segmentation task.

Conference Presentation

Citation Download Citation

S. Afshari, A. BenTaieb, Z. MiriKharaji, and G. Hamarneh "Weakly supervised fully convolutional network for PET lesion segmentation", Proc. SPIE 10949, Medical Imaging 2019: Image Processing, 109491K (15 March 2019); https://doi.org/10.1117/12.2512274

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available