Machine Learning Diffuse Optical Tomography Using Extreme Gradient Boosting and Genetic Programming

Ami Hauptman, Ganesh M. Balasubramaniam, Shlomi Arnon

Research output: Contribution to journalArticlepeer-review

Abstract

Diffuse optical tomography (DOT) is a non-invasive method for detecting breast cancer; however, it struggles to produce high-quality images due to the complexity of scattered light and the limitations of traditional image reconstruction algorithms. These algorithms can be affected by boundary conditions and have a low imaging accuracy, a shallow imaging depth, a long computation time, and a high signal-to-noise ratio. However, machine learning can potentially improve the performance of DOT by being better equipped to solve inverse problems, perform regression, classify medical images, and reconstruct biomedical images. In this study, we utilized a machine learning model called “XGBoost” to detect tumors in inhomogeneous breasts and applied a post-processing technique based on genetic programming to improve accuracy. The proposed algorithm was tested using simulated DOT measurements from complex inhomogeneous breasts and evaluated using the cosine similarity metrics and root mean square error loss. The results showed that the use of XGBoost and genetic programming in DOT could lead to more accurate and non-invasive detection of tumors in inhomogeneous breasts compared to traditional methods, with the reconstructed breasts having an average cosine similarity of more than 0.97 ± 0.07 and average root mean square error of around 0.1270 ± 0.0031 compared to the ground truth.

Original languageAmerican English
Article number382
JournalBioengineering
Volume10
Issue number3
DOIs
StatePublished - 1 Mar 2023

Keywords

  • diffuse optical tomography
  • extreme gradient boosting
  • genetic programming
  • inhomogeneous breast
  • inverse problems

All Science Journal Classification (ASJC) codes

  • Bioengineering

Fingerprint

Dive into the research topics of 'Machine Learning Diffuse Optical Tomography Using Extreme Gradient Boosting and Genetic Programming'. Together they form a unique fingerprint.

Cite this