Enhanced Breast Lesion Classification Via Knowledge Guided Cross-Modal and Semantic Data Augmentation

Kun Chen,Yuanfan Guo,Canqian Yang,Yi Xu,Rui Zhang,Chunxiao Li,Rong Wu
DOI: https://doi.org/10.1007/978-3-030-87240-3_6
2021-01-01
Abstract:Ultrasound (US) imaging is a fundamental modality for detecting and diagnosing breast lesions, while shear-wave elastography (SWE) serves as a crucial complementary counterpart. Although an automated breast lesion classification system is desired, training of such a system is constrained by data scarcity and modality imbalance problems due to the lack of SWE devices in rural hospitals. To enhance the diagnosis with only US available, in this work, we propose a knowledge-guided data augmentation framework, which consists of a modal translater and a semantic inverter, achieving cross-modal and semantic data augmentation simultaneously. Extensive experiments show a significant improvement of AUC from 84.36% to 86.71% using the same ResNetl8 classifier after leveraging our augmentation framework, further improving classification performance based on conventional data augmentation methods and GAN-based data augmentation methods without knowledge guidance.
What problem does this paper attempt to address?