Multiple instance learning with auxiliary task weighting for multiple myeloma classification
File(s)2107.07805v1.pdf (997.46 KB)
Accepted version
Author(s)
Type
Conference Paper
Abstract
Whole body magnetic resonance imaging (WB-MRI) is the recommended modality for diagnosis of multiple myeloma (MM). WB-MRI is used to detect sites of disease across the entire skeletal system, but it requires significant expertise and is time-consuming to report due to the great number of images. To aid radiological reading, we propose an auxiliary task-based multiple instance learning approach (ATMIL) for MM classification with the ability to localize sites of disease. This approach is appealing as it only requires patient-level annotations where an attention mechanism is used to identify local regions with active disease. We borrow ideas from multi-task learning and define an auxiliary task with adaptive reweighting to support and improve learning efficiency in the presence of data scarcity. We validate our approach on both synthetic and real multi-center clinical data. We show that the MIL attention module provides a mechanism to localize bone regions while the adaptive reweighting of the auxiliary task considerably improves the performance.
Date Issued
2021-09-21
Date Acceptance
2021-06-11
Citation
Lecture Notes in Computer Science, 2021, 12907, pp.786-796
ISSN
0302-9743
Publisher
Springer
Start Page
786
End Page
796
Journal / Book Title
Lecture Notes in Computer Science
Volume
12907
Copyright Statement
© 2021 Springer Nature Switzerland AG. The final publication is available at Springer via https://link.springer.com/chapter/10.1007/978-3-030-87234-2_74
Sponsor
National Institute for Health Research
Identifier
http://arxiv.org/abs/2107.07805v1
Grant Number
MALIMAR : 16/68/34
Source
International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI)
Subjects
cs.CV
cs.CV
Notes
Accepted at MICCAI 2021
Publication Status
Published
Start Date
2021-09-27
Finish Date
2021-10-01
Coverage Spatial
Virtual