Adaptive batch-fusion self-supervised learning for ultrasound image pretraining.

Journal: Computerized medical imaging and graphics : the official journal of the Computerized Medical Imaging Society
Published Date:

Abstract

Medical self-supervised learning eliminates the reliance on labels, making feature extraction simple and efficient. The intricate design of pretext tasks in single-modal self-supervised analysis presents challenges, however, compounded by an excessive dependency on data augmentation, leading to a bottleneck in medical self-supervised learning research. Consequently, this paper reanalyzes the feature learnability introduced by data augmentation strategies in medical image self-supervised learning. We introduce an adaptive self-supervised learning data augmentation method from the perspective of batch fusion. Moreover, we propose a conv embedding block for learning the incremental representation between these batches. We tested 5 fused data tasks proposed by previous researchers and it achieved a linear classification protocol accuracy of 94.25% with only 150 self-supervised feature training in Vision Transformer(ViT), which is the best among the same methods. With a detailed ablation study on previous augmentation strategies, the results indicate that the proposed medical data augmentation strategy in this paper effectively represents ultrasound data features in the self-supervised learning process. The code and weights could be found at here.

Authors

  • Jiansong Zhang
    College of Medicine, Huaqiao University, Quanzhou, Fujian Province, China.
  • Xiuming Wu
    Department of Ultrasound, Quanzhou First Hospital Affiliated to Fujian Medical University, Quanzhou, Fujian Province, China.
  • Shunlan Liu
    Department of Ultrasonics, Second Affiliated Hospital of Fujian Medical University, Quanzhou 362000, China.
  • Yuling Fan
    College of Engineering, Huaqiao University, No. 269, Chenghua North Road, Quanzhou, 362021, Fujian, China.
  • Yongjian Chen
    Dermatology and Venereology Division, Department of Medicine Solna, Center for Molecular Medicine, Karolinska Institutet, Stockholm, Sweden.
  • Guorong Lyu
    Department of Ultrasound, The Second Affiliated Hospital of Fujian Medical University, Quanzhou 362000, China.
  • Peizhong Liu
    College of Engineering, Huaqiao University, No. 269, Chenghua North Road, Quanzhou, 362021, Fujian, China. pzliu@hqu.edu.cn.
  • Zhonghua Liu
    The National and Local Joint Engineering Laboratory of Animal Peptide Drug Development, College of Life Sciences, Hunan Normal University, Changsha 410081, People's Republic of China. Electronic address: Liuzh@hunnu.edu.cn.
  • Shaozheng He
    Department of Ultrasound, The Second Affiliated Hospital of Fujian Medical University, Quanzhou 362000, China.