Global-Local attention network with multi-task uncertainty loss for abnormal lymph node detection in MR images.

TitleGlobal-Local attention network with multi-task uncertainty loss for abnormal lymph node detection in MR images.
Publication TypeJournal Article
Year of Publication2022
AuthorsWang S, Zhu Y, Lee S, Elton DC, Shen TC, Tang Y, Peng Y, Lu Z, Summers RM
JournalMed Image Anal
Volume77
Pagination102345
Date Published2022 Apr
ISSN1361-8423
KeywordsAlgorithms, Humans, Lymph Nodes, Magnetic Resonance Imaging, Uncertainty
Abstract

Accurate and reliable detection of abnormal lymph nodes in magnetic resonance (MR) images is very helpful for the diagnosis and treatment of numerous diseases. However, it is still a challenging task due to similar appearances between abnormal lymph nodes and other tissues. In this paper, we propose a novel network based on an improved Mask R-CNN framework for the detection of abnormal lymph nodes in MR images. Instead of laboriously collecting large-scale pixel-wise annotated training data, pseudo masks generated from RECIST bookmarks on hand are utilized as the supervision. Different from the standard Mask R-CNN architecture, there are two main innovations in our proposed network: 1) global-local attention which encodes the global and local scale context for detection and utilizes the channel attention mechanism to extract more discriminative features and 2) multi-task uncertainty loss which adaptively weights multiple objective loss functions based on the uncertainty of each task to automatically search the optimal solution. For the experiments, we built a new abnormal lymph node dataset with 821 RECIST bookmarks of 41 different types of abnormal abdominal lymph nodes from 584 different patients. The experimental results showed the superior performance of our algorithm over compared state-of-the-art approaches.

DOI10.1016/j.media.2021.102345
Alternate JournalMed Image Anal
PubMed ID35051899
PubMed Central IDPMC8988884
Grant ListR00 LM013001 / LM / NLM NIH HHS / United States
Z01 CL040004 / ImNIH / Intramural NIH HHS / United States