Cross-view Contrastive Mutual Learning Across Masked Autoencoders for Mammography Diagnosis

被引:2
作者
Wu, Qingxia [1 ,2 ]
Tan, Hongna [3 ,4 ]
Qiao, Zhi [1 ,2 ]
Dong, Pei [1 ,2 ]
Shen, Dinggang
Wang, Meiyun [3 ,4 ]
Xue, Zhong [1 ,5 ]
机构
[1] United Imaging Res Inst Intelligent Imaging, Beijing, Peoples R China
[2] United Imaging Intelligence Beijing Co Ltd, Beijing, Peoples R China
[3] Henan Prov Peoples Hosp, Zhengzhou, Henan, Peoples R China
[4] Zhengzhou Univ, Peoples Hosp, Zhengzhou, Henan, Peoples R China
[5] Shanghai United Imaging Intelligence Co Ltd, Shanghai, Peoples R China
来源
MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2023, PT II | 2024年 / 14349卷
关键词
Mammography diagnosis; Cross-view masked autoencoder; Contrastive learning; Classification; BREAST-CANCER;
D O I
10.1007/978-3-031-45676-3_8
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Mammography is a widely used screening tool for breast cancer, and accurate diagnosis is critical for the effective management of breast cancer. In this study, we propose a novel cross-view mutual learning method that leverages a Cross-view Masked Autoencoder (CMAE) and a Dual-View Affinity Matrix (DAM) to extract cross-view features and facilitate malignancy classification in mammography. CMAE aims to extract the underlying features from multi-view mammography data without relying on lesion labeling information or multi-view registration. DAM helps overcome the limitations of single-view models and identifies unique patterns and features in each view, thereby improving the accuracy and robustness of breast tissue representations. We evaluate our approach on a large-scale in-house mammography dataset and demonstrate promising results compared to existing methods. Additionally, we perform an ablation analysis to investigate the influence of different loss functions on the performance of our method. The results show that all the proposed components contribute positively to the final performance. In summary, the proposed cross-view mutual learning method shows great potential for assisting malignant classification.
引用
收藏
页码:74 / 83
页数:10
相关论文
共 23 条
[1]   Supervised Contrastive Pre-training for Mammographic Triage Screening Models [J].
Cao, Zhenjie ;
Yang, Zhicheng ;
Tang, Yuxing ;
Zhang, Yanbo ;
Han, Mei ;
Xiao, Jing ;
Ma, Jie ;
Chang, Peng .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT VII, 2021, 12907 :129-139
[2]   Automated Analysis of Unregistered Multi-View Mammograms With Deep Learning [J].
Carneiro, Gustavo ;
Nascimento, Jacinto ;
Bradley, Andrew P. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2017, 36 (11) :2355-2365
[3]  
Chen T., 2020, ADV NEURAL INF PROCE, V33, P22243
[4]  
Chen T, 2020, PR MACH LEARN RES, V119
[5]   Multi-modal Masked Autoencoders for Medical Vision-and-Language Pre-training [J].
Chen, Zhihong ;
Du, Yuhao ;
Hu, Jinpeng ;
Liu, Yang ;
Li, Guanbin ;
Wan, Xiang ;
Chang, Tsung-Hui .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT V, 2022, 13435 :679-689
[6]   Breast Cancer Statistics, 2022 [J].
Giaquinto, Angela N. ;
Sung, Hyuna ;
Miller, Kimberly D. ;
Kramer, Joan L. ;
Newman, Lisa A. ;
Minihan, Adair ;
Jemal, Ahmedin ;
Siegel, Rebecca L. .
CA-A CANCER JOURNAL FOR CLINICIANS, 2022, 72 (06) :524-541
[7]   Masked Autoencoders Are Scalable Vision Learners [J].
He, Kaiming ;
Chen, Xinlei ;
Xie, Saining ;
Li, Yanghao ;
Dollar, Piotr ;
Girshick, Ross .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :15979-15988
[8]  
Kyono T, 2019, PR MACH LEARN RES, V106
[9]   Dual Convolutional Neural Networks for Breast Mass Segmentation and Diagnosis in Mammography [J].
Li, Heyi ;
Chen, Dongdong ;
Nailon, William H. ;
Davies, Mike E. ;
Laurenson, David I. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2022, 41 (01) :3-13
[10]   Domain Generalization for Mammography Detection via Multi-style and Multi-view Contrastive Learning [J].
Li, Zheren ;
Cui, Zhiming ;
Wang, Sheng ;
Qi, Yuji ;
Ouyang, Xi ;
Chen, Qitian ;
Yang, Yuezhi ;
Xue, Zhong ;
Shen, Dinggang ;
Cheng, Jie-Zhi .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT VII, 2021, 12907 :98-108