Semi-supervised object detection based on single-stage detector for thighbone fracture localization

被引:4
|
作者
Wei, Jinman [1 ]
Yao, Jinkun [2 ]
Zhang, Guoshan [1 ]
Guan, Bin [1 ]
Zhang, Yueming [1 ]
Wang, Shaoquan [1 ]
机构
[1] Tianjin Univ, Sch Elect Informat Engn, Weijin Rd, Tianjin 300072, Peoples R China
[2] Linyi Peoples Hosp, Dept Radiol, Linyi 276000, Shandong, Peoples R China
来源
NEURAL COMPUTING & APPLICATIONS | 2024年 / 36卷 / 07期
基金
中国国家自然科学基金;
关键词
Semi-supervised learning; Object detection; Single-stage; Thighbone fracture detection;
D O I
10.1007/s00521-023-09277-3
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The thighbone is the largest bone supporting the lower body. If the thighbone fracture is not treated in time, it will lead to lifelong inability to walk. Correct diagnosis of thighbone disease is very important in orthopedic medicine. Deep learning is promoting the development of fracture detection technology. However, the existing computer-aided diagnosis methods rely on a large number of manually labeled data, and labeling these data costs a lot of time and energy. Therefore, we develop an object detection method with limited labeled image quantity and apply it to the thighbone fracture localization. In this work, we build a semi-supervised object detection framework based on single-stage detector, which includes three modules: adaptive difficult sample oriented (ADSO) module, Fusion Box and deformable expand encoder (Dex encoder). ADSO module takes the classification score as the label reliability evaluation criterion by weighting, Fusion Box is designed to merge similar pseudo boxes into a reliable box for box regression and Dex encoder is proposed to enhance the adaptability of image augmentation. The experiment is conducted on the thighbone fracture dataset, which includes 3484 training thighbone fracture images and 358 testing thighbone fracture images. The experimental results show that the proposed method achieves the state-of-the-art AP in thighbone fracture detection at different labeled data rates, i.e., 1%, 5% and 10%. Besides, we use full data to achieve knowledge distillation, our method achieves 86.2% AP50 and 52.6% AP75. Finally, the effectiveness of our method has also been evaluated using the publicly available datasets COCO and VOC.
引用
收藏
页码:3447 / 3461
页数:15
相关论文
共 50 条
  • [1] Semi-supervised object detection based on single-stage detector for thighbone fracture localization
    Jinman Wei
    Jinkun Yao
    Guoshan Zhang
    Bin Guan
    Yueming Zhang
    Shaoquan Wang
    Neural Computing and Applications, 2024, 36 : 3447 - 3461
  • [2] Improving Localization for Semi-Supervised Object Detection
    Rossi, Leonardo
    Karimi, Akbar
    Prati, Andrea
    IMAGE ANALYSIS AND PROCESSING, ICIAP 2022, PT II, 2022, 13232 : 516 - 527
  • [3] Semi-Supervised Object Detection Algorithm Based on Localization Confidence Weighting
    Feng, Zeheng
    Wang, Feng
    Computer Engineering and Applications, 2024, 60 (06) : 249 - 258
  • [4] Lesion Localization in OCT by Semi-Supervised Object Detection
    Wu, Yue
    Zhou, Yang
    Zhao, Jianchun
    Yang, Jingyuan
    Yu, Weihong
    Chen, Youxin
    Li, Xirong
    PROCEEDINGS OF THE 2022 INTERNATIONAL CONFERENCE ON MULTIMEDIA RETRIEVAL, ICMR 2022, 2022, : 639 - 646
  • [5] Dense Learning based Semi-Supervised Object Detection
    Chen, Binghui
    Li, Pengyu
    Chen, Xiang
    Wang, Biao
    Zhang, Lei
    Hua, Xian-Sheng
    2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 4805 - 4814
  • [6] IoU-aware single-stage object detector for accurate localization
    Wu, Shengkai
    Li, Xiaoping
    Wang, Xinggang
    IMAGE AND VISION COMPUTING, 2020, 97
  • [7] Semantic-Edge-Supervised Single-Stage Detector for Oriented Object Detection in Remote Sensing Imagery
    Cao, Dujuan
    Zhu, Changming
    Hu, Xinxin
    Zhou, Rigui
    REMOTE SENSING, 2022, 14 (15)
  • [8] Consistency-based Semi-supervised Learning for Object Detection
    Jeong, Jisoo
    Lee, Seungeui
    Kim, Jeesoo
    Kwak, Nojun
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 32 (NIPS 2019), 2019, 32
  • [9] Interpolation-based Semi-supervised Learning for Object Detection
    Jeong, Jisoo
    Verma, Vikas
    Hyun, Minsung
    Kannala, Juho
    Kwak, Nojun
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 11597 - 11606
  • [10] Dense Information Learning Based Semi-Supervised Object Detection
    Yang, Xi
    Li, Penghui
    Zhou, Qiubai
    Wang, Nannan
    Gao, Xinbo
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2025, 34 : 1022 - 1035