Deep Reinforcement Learning for Fractionated Radiotherapy in Non-Small Cell Lung Carcinoma

Artif Intell Med. 2021 Sep:119:102137. doi: 10.1016/j.artmed.2021.102137. Epub 2021 Aug 15.

Abstract

Lung cancer is by far the leading cause of cancer death among both men and women. Radiation therapy is one of the main approaches to lung cancer treatment, and its planning is crucial for the therapy outcome. However, the current practice that uniformly delivers the dose does not take into account the patient-specific tumour features that may affect treatment success. Since radiation therapy is by its very nature a sequential procedure, Deep Reinforcement Learning (DRL) is a well-suited methodology to overcome this limitation. In this respect, in this work we present a DRL controller optimizing the daily dose fraction delivered to the patient on the basis of CT scans collected over time during the therapy, offering a personalized treatment not only for volume adaptation, as currently intended, but also for daily fractionation. Furthermore, this contribution introduces a virtual radiotherapy environment based on a set of ordinary differential equations modelling the tissue radiosensitivity by combining both the effect of the radiotherapy treatment and cell growth. Their parameters are estimated from CT scans routinely collected using the Particle Swarm Optimization algorithm. This permits the DRL to learn the optimal behaviour through an iterative trial and error process with the environment. We performed several experiments considering three rewards functions modelling treatment strategies with different tissue aggressiveness and two exploration strategies for the exploration-exploitation dilemma. The results show that our DRL approach can adapt to radiation therapy treatment, optimizing its behaviour according to the different reward functions and outperforming the current clinical practice.

Keywords: D3QN; Deep reinforcement learning; NSCLC; Particle swarm optimization; Radiation therapy; Tumour treatment optimization.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Carcinoma, Non-Small-Cell Lung* / radiotherapy
  • Dose Fractionation, Radiation
  • Female
  • Humans
  • Lung Neoplasms* / radiotherapy
  • Male
  • Radiotherapy Dosage
  • Radiotherapy Planning, Computer-Assisted