Radiation and Cancer Physics

PD 09 - Physics 5 - Poster Discussion - Imaging for Treatment Planning

1079 - Parameter Tuning in Regularization-Based Iterative CT Reconstruction Via Deep Reinforcement Learning

Tuesday, October 23
1:42 PM - 1:48 PM
Location: Room 217 C/D

Parameter Tuning in Regularization-Based Iterative CT Reconstruction Via Deep Reinforcement Learning
C. Shen, Y. Gonzalez, L. Chen, S. B. Jiang, and X. Jia; Department of Radiation Oncology, University of Texas Southwestern Medical Center, Dallas, TX

Purpose/Objective(s): Regularization-based iterative CT reconstruction has demonstrated its tremendous power. Formulated under an optimization approach, regularization-based iterative CT reconstruction considers an objective function containing a data-fidelity term and a regularization term. With the latter introduced to enforce image quality, high quality images can be reconstructed under clinically desirable conditions, such as reduced imaging dose to patients. Yet there are always parameters in the problem, which govern tradeoffs between the two terms and impact solution image quality. Manual parameter tuning is not only tedious, but become impractical when many parameters exist. Motivated by recent advancements in deep-learning area that can accomplish tasks in a human-like manner, such as playing computer games, we propose to employ deep reinforcement learning (DRL) to develop a system that automatically adjusts parameters with human-level intelligence, such that the intensive efforts of manual parameter tuning can be released.

Materials/Methods: We consider an example problem of CT reconstruction with pixel-wise total-variation (TV) regularization. We set up a parameter tuning policy network (PTPN). It maps an image patch to a decision that specifies the direction and amplitude of parameter adjustment for the pixel at the patch center. To train PTPN, a reward function is defined that favors improvement in image quality, as quantified by the difference between the solution image and the ground truth image, which is known at the training stage. We train PTPN via end-to-end deep DRL with experience replay technique. Once PTPN is trained, we test it in simulation and real experimental cases. During a reconstruction process, we initiate parameters randomly. PTPN continuously observes the intermediate reconstructed image and make decision to adjust the parameters, until the image quality cannot be further improved.

Results: In simulation studies, PTPN can intelligently guide the reconstruction process, yielding images with 3% lower error than that under the randomly initialized parameters. The resulting image is slightly better than that under manually tuned parameter (~0.5% lower error). Similar behavior is observed in experimental data. For the problem of CT reconstruction with pixel-wise TV regularization, closed form solution of the optimal parameter values can be derived. The parameters tuned by PTPN appeared similar to the optimal parameters, demonstrating the effectiveness of PTPN.

Conclusion: Under DRL, we have developed PTPN that can adjust parameters of an iterative CT reconstruction problem in a human-like manner. The resulting image is similar or better than those under manual parameter tuning.

Author Disclosure: C. Shen: None. Y. Gonzalez: None. S.B. Jiang: Patent/License Fees/Copyright; Sun Nuclear. X. Jia: None.

Chenyang Shen, PhD

University of Texas Southwestern Medical Center

Disclosure:
No relationships to disclose.

Presentation(s):

Send Email for Chenyang Shen


Assets

1079 - Parameter Tuning in Regularization-Based Iterative CT Reconstruction Via Deep Reinforcement Learning



Attendees who have favorited this

Please enter your access key

The asset you are trying to access is locked. Please enter your access key to unlock.

Send Email for Parameter Tuning in Regularization-Based Iterative CT Reconstruction Via Deep Reinforcement Learning