Multi-task Class Feature Space Fusion Domain Adaptation Network for Thyroid Ultrasound Images: Research on Generalization of Smart Healthcare Systems

Xiang Ying,Zhen Liu,Jie Gao,Ruixuan Zhang,Han Jiang,Xi Wei
DOI: https://doi.org/10.1007/978-3-031-19208-1_12
2022-01-01
Abstract:In recent years, the poor generalizability of deep neural networks in multi-model medical images has attracted widespread attention. Domain adaptation is an approach to alleviate the above problem, which transfers the labeled source domain to the target domain. It can reduce the data labeling workload in the target domain and significantly improve the network's generalizability. However, the differences between foreground areas and background areas of medical images are relatively minor, and it is difficult for existing methods to effectively extract domain invariant features. Further optimization of the feature distribution alignment for each category is also lacking. Therefore, a Multi-task Class feature space Fusion Domain Adaptation Network (MCFDAN) is proposed in this paper. Firstly, a reconstruction branch is added to the baseline network to mitigate feature offset of the target domain during encoding. Secondly, category constraints are added to the fusion of domain feature spaces, improving the generalizability of the source classifier to the target domain. Finally, the network incorporates a recurrent cross-attention module that highlights the feature expression of the lesion region. The evaluation results demonstrate that the proposed network achieves a significant performance improvement, which is important for the application of smart healthcare systems.
What problem does this paper attempt to address?