Advancing 3D medical image analysis with variable dimension transform based supervised 3D pre-training

被引:2
|
作者
Zhang, Shu [1 ]
Li, Zihao [1 ]
Zhou, Hong-Yu [2 ]
Ma, Jiechao [1 ]
Yu, Yizhou [1 ,2 ]
机构
[1] Deepwise Artificial Intelligence Lab, 8 Haidian Ave, Beijing, Peoples R China
[2] Univ Hong Kong, Dept Comp Sci, Pokfulam, Hong Kong, Peoples R China
基金
中国国家自然科学基金;
关键词
3D medical image; Transfer learning; Variable dimension transform; Supervised pre-training; CT; NETWORK;
D O I
10.1016/j.neucom.2023.01.012
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The difficulties in both data acquisition and annotation substantially restrict the sample sizes of training datasets for 3D medical imaging applications. Therefore, it is non-trivial to build well-performing 3D con-volutional neural networks from scratch. Previous efforts on 3D pre-training have frequently relied on self-supervised approaches, which use either predictive or contrastive learning on unlabeled data to build invariant 3D representations. However, because of the unavailability of large-scale supervision informa-tion, obtaining semantically invariant and discriminative representations from these learning frame-works remains problematic. In this paper, we revisit an innovative yet simple fully-supervised 3D network pre-training framework to take advantage of semantic supervision from large-scale 2D natural image datasets. With a redesigned 3D network architecture, reformulated natural images are used to address the problem of data scarcity and develop powerful 3D representations. Comprehensive experi-ments on five benchmark datasets demonstrate that the proposed pre-trained models can effectively accelerate convergence while also improving accuracy for a variety of 3D medical imaging tasks such as classification, segmentation, and detection. In addition, as compared to training from scratch, it can save up to 60% of annotation efforts. On the NIH DeepLesion dataset, it also achieves state-of-the-art detection performance, outperforming earlier self-supervised and fully-supervised pre-training approaches, as well as methods that do training from scratch. To facilitate further development of 3D medical models, our code and pre-trained model weights are publicly available at https://github.com/u rmagicsmine/CSPR. (c) 2023 Elsevier B.V. All rights reserved.
引用
收藏
页码:11 / 22
页数:12
相关论文
共 50 条
  • [41] Coronary Arteries Segmentation Based on the 3D Discrete Wavelet Transform and 3D Neutrosophic Transform
    Chen, Shuo-Tsung
    Wang, Tzung-Dau
    Lee, Wen-Jeng
    Huang, Tsai-Wei
    Hung, Pei-Kai
    Wei, Cheng-Yu
    Chen, Chung-Ming
    Kung, Woon-Man
    BIOMED RESEARCH INTERNATIONAL, 2015, 2015
  • [42] 3D Shape Recovery by Aggregating 3D Wavelet Transform-Based Image Focus Volumes Through 3D Weighted Least Squares
    Usman Ali
    Muhammad Tariq Mahmood
    Journal of Mathematical Imaging and Vision, 2020, 62 : 54 - 72
  • [43] Conditional GAN with an Attention-Based Generator and a 3D Discriminator for 3D Medical Image Generation
    Jung, Euijin
    Luna, Miguel
    Park, Sang Hyun
    Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 2021, 12906 LNCS : 318 - 328
  • [44] Conditional GAN with an Attention-Based Generator and a 3D Discriminator for 3D Medical Image Generation
    Jung, Euijin
    Luna, Miguel
    Park, Sang Hyun
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT VI, 2021, 12906 : 318 - 328
  • [45] Masked Image Modeling Advances 3D Medical Image Analysis
    Chen, Zekai
    Agarwal, Devansh
    Aggarwal, Kshitij
    Safta, Wiem
    Balan, Mariann Micsinai
    Brown, Kevin
    2023 IEEE/CVF WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2023, : 1969 - 1979
  • [46] Single Image 3D Without a Single 3D Image
    Fouhey, David F.
    Hussain, Wajahat
    Gupta, Abhinav
    Hebert, Martial
    2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1053 - 1061
  • [47] Combining 3D Convolutional Neural Networks with Transfer Learning by Supervised Pre-Training for Facial Micro-Expression Recognition
    Zhi, Ruicong
    Xu, Hairui
    Wan, Ming
    Li, Tingting
    IEICE TRANSACTIONS ON INFORMATION AND SYSTEMS, 2019, E102D (05) : 1054 - 1064
  • [48] Morphogenesis-based deformable models in 3D medical image analysis
    Roux, C
    Ibáñez, L
    Hamitouche, C
    Boniou, M
    MEDICON 2001: PROCEEDINGS OF THE INTERNATIONAL FEDERATION FOR MEDICAL & BIOLOGICAL ENGINEERING, PTS 1 AND 2, 2001, : 452 - 455
  • [49] Integrated 3D medical image processing and analysis system
    Tian, J
    He, HG
    Zhao, MC
    Li, GM
    Ge, XF
    Zhu, FP
    ADVANCED BIOMEDICAL AND CLINICAL DIAGNOSTIC SYSTEMS, 2003, 4958 : 284 - 293
  • [50] Hybrid segmentation framework for 3D medical image analysis
    Chen, T
    Metaxas, D
    MEDICAL IMAGING 2003: IMAGE PROCESSING, PTS 1-3, 2003, 5032 : 1421 - 1432