Restricted Boltzmann machines based oversampling and semi-supervised learning for false positive reduction in breast CAD.

Journal: Bio-medical materials and engineering
Published Date:

Abstract

The false-positive reduction (FPR) is a crucial step in the computer aided detection system for the breast. The issues of imbalanced data distribution and the limitation of labeled samples complicate the classification procedure. To overcome these challenges, we propose oversampling and semi-supervised learning methods based on the restricted Boltzmann machines (RBMs) to solve the classification of imbalanced data with a few labeled samples. To evaluate the proposed method, we conducted a comprehensive performance study and compared its results with the commonly used techniques. Experiments on benchmark dataset of DDSM demonstrate the effectiveness of the RBMs based oversampling and semi-supervised learning method in terms of geometric mean (G-mean) for false positive reduction in Breast CAD.

Authors

  • Peng Cao
    Medical Image Computing Laboratory of Ministry of Education, Northeastern University, 110819, Shenyang, China.
  • Xiaoli Liu
    Neurology Department, Zhejiang Hospital, Zhejiang 310013, China.
  • Hang Bao
    Medical Image Computing Laboratory of Ministry of Education, Northeastern University, 110819, Shenyang, China.
  • Jinzhu Yang
    College of Information Science and Engineering, Northeastern University, 110819, Shenyang, China.
  • Dazhe Zhao
    Medical Image Computing Laboratory of Ministry of Education, Northeastern University, 110819, Shenyang, China.