Mutual information-based disentangled neural networks for classifying unseen categories in different domains: application to fetal ultrasound imaging
File(s)feature_generalization.pdf (3.65 MB)
Accepted version
Author(s)
Type
Journal Article
Abstract
Deep neural networks exhibit limited generalizability across images with different entangled domain features and categorical features. Learning generalizable features that can form universal categorical decision boundaries across domains is an interesting and difficult challenge. This problem occurs frequently in medical imaging applications when attempts are made to deploy and improve deep learning models across different image acquisition devices, across acquisition parameters or if some classes are unavailable in new training databases. To ad-dress this problem, we propose Mutual Information-based Disentangled Neural Networks (MIDNet), which extract generalizable categorical features to transfer knowledge to unseen categories in a target domain. The proposed MID-Net adopts a semi-supervised learning paradigm to alleviate the dependency on labeled data. This is important for real-world applications where data annotation is time-consuming, costly and requires training and expertise. We extensively evaluate the proposed method on fetal ultra-sound datasets for two different image classification tasks where domain features are respectively defined by shadow artifacts and image acquisition devices. Experimental results show that the proposed method outperforms the state-of-the-art on the classification of unseen categories in a target domain with sparsely labeled training data.
Date Issued
2021-02
Date Acceptance
2020-10-28
Citation
IEEE Transactions on Medical Imaging, 2021, 40 (2), pp.722-734
ISSN
0278-0062
Publisher
Institute of Electrical and Electronics Engineers
Start Page
722
End Page
734
Journal / Book Title
IEEE Transactions on Medical Imaging
Volume
40
Issue
2
Copyright Statement
© 2020 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Sponsor
Engineering & Physical Science Research Council (E
Wellcome Trust
Wellcome Trust/EPSRC
Wellcome Trust
Engineering & Physical Science Research Council (E
Identifier
https://ieeexplore.ieee.org/document/9247170
Grant Number
RTJ5557761-1
PO :RTJ5557761-1
NS/A000025/1
RTJ5557761
RTJ5557761-1
Subjects
Nuclear Medicine & Medical Imaging
08 Information and Computing Sciences
09 Engineering
Publication Status
Published
Date Publish Online
2020-11-03