Classification of Volumetric Images Using Multi-Instance Learning and Extreme Value Theorem
Volumetric imaging is an essential diagnostic tool for medical practitioners. The use of popular techniques such as convolutional neural networks (CNN) for analysis of volumetric images is constrained by the availability of detailed (with local annotations) training data and GPU memory. In this paper, the volumetric image classification problem is posed as a multi-instance classification problem and a novel method is proposed to adaptively select positive instances from positive bags during the training phase. This method uses the extreme value theory to model the feature distribution of the images without a pathology and use it to identify positive instances of an imaged pathology. The experimental results, on three separate image classification tasks (i.e. classify retinal OCT images according to the presence or absence of fluid build-ups, emphysema detection in pulmonary 3D-CT images and detection of cancerous regions in 2D histopathology images) show that the proposed method produces classifiers that have similar performance to fully supervised methods and achieves the state of the art performance in all examined test cases.
|Keywords||CNN, COPD, CT, emphysema, Macular Edema, Multiple instance learning, OCT, weakly supervised learning|
|Persistent URL||dx.doi.org/10.1109/TMI.2019.2936244, hdl.handle.net/1765/126151|
|Journal||IEEE Transactions on Medical Imaging|
Tennakoon, R.B, Bortsova, G, Orting, S. (Silas), Gostar, A.K. (Amirali K.), Wille, M.M.W, Saghir, Z, … Bab-Hadiashar, A. (2020). Classification of Volumetric Images Using Multi-Instance Learning and Extreme Value Theorem. IEEE Transactions on Medical Imaging, 39(4), 854–865. doi:10.1109/TMI.2019.2936244