Binary and Random Inputs to Rapidly Identify Overfitting of Deep Neural Networks Trained to Output Ultrasound Images

Jiaxin Zhang, Alycen Wiacek, Muyinatu A.Lediju Bell

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

We developed a novel method to detect overfitting of deep neural networks trained to create ultrasound images. This method only requires the network architecture and trained weights, and does not require loss function monitoring during an otherwise time-consuming training process. Specifically, two binary images and an image of Gaussian random noise were used as inputs to three neural networks submitted to the Challenge on Ultrasound Beamforming with Deep Learning (CUBDL). Comparing the network-created images to the ground truth immediately revealed an overfit to the data used to train one of the three networks, indicating the promise of our method to detect overfitting without requiring lengthy network retraining or the collection of additional test data. This approach holds promise for regulatory oversight of DNNs intended to be deployed on patient data.

Original languageEnglish (US)
Title of host publicationIUS 2022 - IEEE International Ultrasonics Symposium
PublisherIEEE Computer Society
ISBN (Electronic)9781665466578
DOIs
StatePublished - 2022
Event2022 IEEE International Ultrasonics Symposium, IUS 2022 - Venice, Italy
Duration: Oct 10 2022Oct 13 2022

Publication series

NameIEEE International Ultrasonics Symposium, IUS
Volume2022-October
ISSN (Print)1948-5719
ISSN (Electronic)1948-5727

Conference

Conference2022 IEEE International Ultrasonics Symposium, IUS 2022
Country/TerritoryItaly
CityVenice
Period10/10/2210/13/22

Keywords

  • beamforming
  • deep learning
  • imaging
  • overfit

ASJC Scopus subject areas

  • Acoustics and Ultrasonics

Fingerprint

Dive into the research topics of 'Binary and Random Inputs to Rapidly Identify Overfitting of Deep Neural Networks Trained to Output Ultrasound Images'. Together they form a unique fingerprint.

Cite this