arXiv reaDer
深層学習ニューラルネットワークを使用した放射線療法の線量予測における不確実性推定のためのモンテカルロドロップアウトとブートストラップ集約の使用
Using Monte Carlo dropout and bootstrap aggregation for uncertainty estimation in radiation therapy dose prediction with deep learning neural networks
最近、人工知能技術とアルゴリズムは、放射線療法の治療計画の進歩の主要な焦点となっています。これらが臨床ワークフローに組み込まれ始めているため、臨床医の主な関心事は、モデルが正確であるかどうかではなく、答えが正しいかどうかわからないときにモデルが人間のオペレーターに表現できるかどうかです。モンテカルロドロップアウト(MCDO)とブートストラップ集約(バギング)手法を深層学習モデルで使用して、放射線療法の線量予測の不確実性を推定することを提案します。両方のモデルが妥当な不確実性マップを生成でき、提案されたスケーリング手法を使用して、予測および関連するメトリックに解釈可能な不確実性と限界を作成できることを示します。パフォーマンス面では、バギングは、この調査で調査されたほとんどのメトリックで統計的に有意な損失値とエラーの削減を提供します。バギングを追加すると、ベースラインフレームワークと比較して、平均してDmeanで0.34%、Dmaxで0.19%エラーをさらに減らすことができました。全体として、バギングフレームワークは、ベースラインフレームワークのMAE 2.87とは対照的に、大幅に低いMAE2.62を提供しました。バギングの有用性は、パフォーマンスの観点からのみ、問題と許容可能な予測誤差に大きく依存します。トレーニング中のバギングの高い先行計算コストは、バギングを使用することが有利かどうかを判断する際に考慮に入れる必要があります。不確実性の見積もりをオンにした展開に関しては、両方のフレームワークが約12秒の同じパフォーマンス時間を提供します。アンサンブルベースのメタヒューリスティックとして、バギングを既存の機械学習アーキテクチャで使用して安定性とパフォーマンスを向上させることができます。MCDOは、アーキテクチャの一部としてドロップアウトがあるディープラーニングモデルに適用できます。
Recently, artificial intelligence technologies and algorithms have become a major focus for advancements in treatment planning for radiation therapy. As these are starting to become incorporated into the clinical workflow, a major concern from clinicians is not whether the model is accurate, but whether the model can express to a human operator when it does not know if its answer is correct. We propose to use Monte Carlo dropout (MCDO) and the bootstrap aggregation (bagging) technique on deep learning models to produce uncertainty estimations for radiation therapy dose prediction. We show that both models are capable of generating a reasonable uncertainty map, and, with our proposed scaling technique, creating interpretable uncertainties and bounds on the prediction and any relevant metrics. Performance-wise, bagging provides statistically significant reduced loss value and errors in most of the metrics investigated in this study. The addition of bagging was able to further reduce errors by another 0.34% for Dmean and 0.19% for Dmax, on average, when compared to the baseline framework. Overall, the bagging framework provided significantly lower MAE of 2.62, as opposed to the baseline framework's MAE of 2.87. The usefulness of bagging, from solely a performance standpoint, does highly depend on the problem and the acceptable predictive error, and its high upfront computational cost during training should be factored in to deciding whether it is advantageous to use it. In terms of deployment with uncertainty estimations turned on, both frameworks offer the same performance time of about 12 seconds. As an ensemble-based metaheuristic, bagging can be used with existing machine learning architectures to improve stability and performance, and MCDO can be applied to any deep learning models that have dropout as part of their architecture.
updated: Sun Nov 01 2020 00:24:43 GMT+0000 (UTC)
published: Sun Nov 01 2020 00:24:43 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト