arXiv reaDer
Incorporating human and learned domain knowledge into training deep neural networks: A differentiable dose volume histogram and adversarial inspired framework for generating Pareto optimal dose distributions in radiation therapy
 線量体積ヒストグラムに基づく微分可能な損失関数である新しいドメイン固有の損失を提案し、パレート最適線量分布を生成するためにディープニューラルネットワークのトレーニングのための敵対的損失と組み合わせます。平均二乗誤差(MSE)損失、線量体積ヒストグラム(DVH)損失、および敵対的(ADV)損失を使用して、ニューラルネットワークモデルの4つのインスタンスをトレーニングしました。1)MSE、2)MSE + ADV、3)MSE + DVH 、および4)MSE + DVH + ADV。 70人の前立腺患者を獲得し、各患者の線量影響配列を計算しました。トレードオフの重みを擬似ランダム化することにより、患者あたり1200パレート表面計画が生成されました(合計84,000計画)。データを54人のトレーニング、6人の検証、10人のテスト患者に分割しました。各モデルは、バッチサイズ2で100,000回の反復でトレーニングされました。各モデルの予測時間は0.052秒です。定量的に、MSE + DVH + ADVモデルの予測誤差は0.038(コンフォメーション)、0.026(均質性)、0.298(R50)、1.65%(D95)、2.14%(D98)、2.43%(D99)でした。 MSEモデルの最悪の予測誤差は、0.134(コンフォメーション)、0.041(均一性)、0.520(R50)、3.91%(D95)、4.33%(D98)、4.60%(D99)でした。平均線量PTVエラーと最大線量PTV、体、膀胱および直腸のエラーの両方について、MSE + DVH + ADVは他のすべてのモデルを上回りました。すべてのモデルの予測の平均平均誤差と最大線量誤差は、それぞれ2.8%と4.2%未満です。専門家の人間ドメイン固有の知識は、パフォーマンス向上の最大の推進力となり、敵対的学習を使用して微妙な特徴をさらにキャプチャすることができます。リアルタイム予測機能により、医師はトレードオフスペースをすばやくナビゲートし、線量計を計画に使用する線量計の具体的なエンドポイントとして作成できます。これにより、治療計画の時間を大幅に短縮でき、臨床医は困難な症例に努力を集中できます。
We propose a novel domain specific loss, which is a differentiable loss function based on the dose volume histogram, and combine it with an adversarial loss for the training of deep neural networks to generate Pareto optimal dose distributions. The mean squared error (MSE) loss, dose volume histogram (DVH) loss, and adversarial (ADV) loss were used to train 4 instances of the neural network model: 1) MSE, 2) MSE+ADV, 3) MSE+DVH, and 4) MSE+DVH+ADV. 70 prostate patients were acquired, and the dose influence arrays were calculated for each patient. 1200 Pareto surface plans per patient were generated by pseudo-randomizing the tradeoff weights (84,000 plans total). We divided the data into 54 training, 6 validation, and 10 testing patients. Each model was trained for 100,000 iterations, with a batch size of 2. The prediction time of each model is 0.052 seconds. Quantitatively, the MSE+DVH+ADV model had the lowest prediction error of 0.038 (conformation), 0.026 (homogeneity), 0.298 (R50), 1.65% (D95), 2.14% (D98), 2.43% (D99). The MSE model had the worst prediction error of 0.134 (conformation), 0.041 (homogeneity), 0.520 (R50), 3.91% (D95), 4.33% (D98), 4.60% (D99). For both the mean dose PTV error and the max dose PTV, Body, Bladder and rectum error, the MSE+DVH+ADV outperformed all other models. All model's predictions have an average mean and max dose error less than 2.8% and 4.2%, respectively. Expert human domain specific knowledge can be the largest driver in the performance improvement, and adversarial learning can be used to further capture nuanced features. The real-time prediction capabilities allow for a physician to quickly navigate the tradeoff space, and produce a dose distribution as a tangible endpoint for the dosimetrist to use for planning. This can considerably reduce the treatment planning time, allowing for clinicians to focus their efforts on challenging cases.
updated: Sat Dec 07 2019 12:19:23 GMT+0000 (UTC)
published: Fri Aug 16 2019 07:45:38 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト