Random search as a neural network optimization strategy for Convolutional-Neural-Network (CNN)-based noise reduction in CT

Nathan R. Huber; Andrew D. Missert; Hao Gong; Scott S. Hsieh; Shuai Leng; Lifeng Yu; Cynthia H. McCollough

doi:10.1117/12.2582143

15 February 2021 Random search as a neural network optimization strategy for Convolutional-Neural-Network (CNN)-based noise reduction in CT

Nathan R. Huber, Andrew D. Missert, Hao Gong, Scott S. Hsieh, Shuai Leng, Lifeng Yu, Cynthia H. McCollough

Author Affiliations +

Proceedings Volume 11596, Medical Imaging 2021: Image Processing; 115961U (2021) https://doi.org/10.1117/12.2582143
Event: SPIE Medical Imaging, 2021, Online Only

Abstract

In this study, we describe a systematic approach to optimize deep-learning-based image processing algorithms using random search. The optimization technique is demonstrated on a phantom-based noise reduction training framework; however, the techniques described can be applied generally for other deep learning image processing applications. The parameter space explored included number of convolutional layers, number of filters, kernel size, loss function, and network architecture (either U-Net or ResNet). A total of 100 network models were examined (50 random search, 50 ablation experiments). Following the random search, ablation experiments resulted in a very minor performance improvement indicating near optimal settings were found during the random search. The top performing network architecture was a U-Net with 4 pooling layers, 64 filters, 3x3 kernel size, ELU activation, and a weighted feature reconstruction loss (0.2×VGG + 0.8×MSE). Relative to the low-dose input image, the CNN reduced noise by 90%, reduced RMSE by 34%, and increased SSIM by 76% on six patient exams reserved for testing. The visualization of hepatic and bone lesions was greatly improved following noise reduction.

Conference Presentation

Citation Download Citation

Nathan R. Huber, Andrew D. Missert, Hao Gong, Scott S. Hsieh, Shuai Leng, Lifeng Yu, and Cynthia H. McCollough "Random search as a neural network optimization strategy for Convolutional-Neural-Network (CNN)-based noise reduction in CT", Proc. SPIE 11596, Medical Imaging 2021: Image Processing, 115961U (15 February 2021); https://doi.org/10.1117/12.2582143

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available