• Home
  • Search
  • Applying a Random Projection Algorithm to Optimize Machine Learning Model for Breast Lesion Classification.
  • Open Access IconOpen Access
  • Cite Icon33
  • https://doi.org/10.1109/tbme.2021.3054248Copy DOI Icon

Applying a Random Projection Algorithm to Optimize Machine Learning Model for Breast Lesion Classification.

Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Since computer-aided diagnosis (CAD) schemes of medical images usually computes large number of image features, which creates a challenge of how to identify a small and optimal feature vector to build robust machine learning models, the objective of this study is to investigate feasibility of applying a random projection algorithm (RPA) to build an optimal feature vector from the initially CAD-generated large feature pool and improve performance of machine learning model. We assemble a retrospective dataset involving 1,487 cases of mammograms in which 644 cases have confirmed malignant mass lesions and 843 have benign lesions. A CAD scheme is first applied to segment mass regions and initially compute 181 features. Then, support vector machine (SVM) models embedded with several feature dimensionality reduction methods are built to predict likelihood of lesions being malignant. All SVM models are trained and tested using a leave-one-case-out cross-validation method. SVM generates a likelihood score of each segmented mass region depicting on one-view mammogram. By fusion of two scores of the same mass depicting on two-view mammograms, a case-based likelihood score is also evaluated. Comparing with the principle component analyses, nonnegative matrix factorization, and Chi-squared methods, SVM embedded with RPA yielded a significantly higher case-based lesion classification performance with the area under ROC curve of 0.84 ± 0.01 (p<0.02). The study demonstrates that RPA is a promising method to generate optimal feature vectors and improve SVM performance. This study presents a new method to develop CAD schemes with significantly higher and robust performance.

Similar Papers
  • Research Article
  • Citations5

Comparative study of different machine learning models in landslide susceptibility assessment: A case study of Conghua District, Guangzhou, China

  • Jan 01, 2024
  • China Geology
  • Ao Zhang +10
  • PDF
  • Research Article
  • Citations20

Machine learning prediction of prostate cancer from transrectal ultrasound video clips

  • Aug 26, 2022
  • Frontiers in Oncology
  • Kai Wang +9
  • PDF
  • Research Article
  • Citations20

Tool and Workpiece Condition Classification Using Empirical Mode Decomposition (EMD) with Hilbert–Huang Transform (HHT) of Vibration Signals and Machine Learning Models

  • Feb 09, 2023
  • Applied Sciences
  • Isaac Opeyemi Olalere +1
  • Research Article
  • Citations9

Support Vector Machine Based Prediction of Work-Life Balance Among Women in Information Technology Organizations

  • Jun 01, 2022
  • IEEE Engineering Management Review
  • Chhavi Gupta +2
  • Research Article
  • Citations1

Near infrared spectroscopy for the characterisation of bovine teeth in terms of sex, diet, tooth type and place of origin

  • Nov 22, 2024
  • Journal of Near Infrared Spectroscopy
  • Ne Pretorius +3
  • PDF
  • Research Article
  • Citations32

The Machine Learning Model for Distinguishing Pathological Subtypes of Non-Small Cell Lung Cancer.

  • May 26, 2022
  • Frontiers in Oncology
  • Hongyue Zhao +9
  • PDF
  • Research Article
  • Citations3

Advancing vector-borne disease prediction through functional classifier integration: A novel approach for enhanced modeling

  • Feb 26, 2024
  • Letters In Animal Biology
  • Mokammel Hossain Tito +9
  • Research Article
  • Citations42

Feature Selection Using Parallel Genetic Algorithm for the Prediction of Geometric Mean Diameter of Soil Aggregates by Machine Learning Methods

  • May 01, 2014
  • Arid Land Research and Management
  • A A Besalatpour +4
  • PDF
  • Research Article
  • Citations6

Parkinson’s Disease Classification Using Random Forest Kerb Feature Selection

  • Jan 01, 2023
  • Intelligent Automation &amp; Soft Computing
  • E Bharath +1
  • Research Article
  • Citations10

A new method to classify pathologic grades of astrocytomas based on magnetic resonance imaging appearances

  • Jan 01, 2010
  • Neurology India
  • Min He +6
  • Research Article

Impact of Feature Set Size on the Performance of Machine Learning Models in Cross-Project Defect Prediction

  • Aug 27, 2025
  • International Journal on Advanced Science Engineering and Information Technology
  • Yahaya Zakariyau Bala +3
  • Research Article
  • Citations5

앙상블 SVM 모형을 이용한 기업 부도 예측

  • Nov 30, 2013
  • Journal of the Korean Data and Information Science Society
  • Ha Na Choi +1
  • Research Article
  • Citations15

Improving autocoding performance of rare categories in injury classification: Is more training data or filtering the solution?

  • Nov 08, 2017
  • Accident Analysis &amp; Prevention
  • Gaurav Nanda +2
  • Research Article
  • Citations9

Machine learning models compared with current clinical indices to predict the outcome of high flow nasal cannula therapy in acute hypoxemic respiratory failure

  • Mar 07, 2025
  • Critical Care
  • Hang Yu +10
  • PDF
  • Research Article
  • Citations49

Improving the Accuracy of Landslide Detection in “Off-site” Area by Machine Learning Model Portability Comparison: A Case Study of Jiuzhaigou Earthquake, China

  • Oct 29, 2019
  • Remote Sensing
  • Qiao Hu +4
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.