DAE-GCN: Identifying Disease-Related Features for Disease Prediction

Chu-ran Wang,Xinwei Sun,Fandong Zhang,Yizhou Yu,Yizhou Wang
DOI: https://doi.org/10.1007/978-3-030-87240-3_5
2021-01-01
Abstract:Learning disease-related representations plays a critical role in image-based cancer diagnosis, due to its trustworthy, interpretable and good generalization power. A good representation should not only be disentangled from the disease-irrelevant features, but also incorporate the information of lesion's attributes (e.g., shape, margin) that are often identified first during cancer diagnosis clinically. To learn such a representation, we propose a Disentangle Auto-Encoder with Graph Convolutional Network (DAE-GCN), which adopts a disentangling mechanism with the guidance of a GCN model in the AE-based framework. Specifically, we explicitly separate the encoded features into disease-related features and others. Among such features that all participate in image reconstruction, we only employ the disease-related features for disease prediction. Besides, to account for lesions' attributes, we propose to leverage the attributes and adopt the GCN to learn them during training. Take mammogram mass benign/malignant classification as an example, our DAE-GCN helps improve the performance and the interpretability of cancer prediction, which can be verified by state-of-the-art performance on one public dataset DDSM and three in-house datasets.
What problem does this paper attempt to address?