arXiv reaDer
LE-NAS:線量予測のためのNASを備えた学習ベースのアンサンブル
LE-NAS: Learning-based Ensemble with NAS for Dose Prediction
目標線量処方と正常組織温存は相反する目的であるため、放射線療法の治療計画は複雑なプロセスです。放射線治療計画のための自動化された正確な線量予測が強く求められています。この研究では、LE-NASという名前の新しい学習ベースのアンサンブルアプローチを提案します。これは、ニューラルアーキテクチャ検索(NAS)と3D放射線治療線量予測のための知識蒸留を統合したものです。具体的には、予測ネットワークはまず、巨大なアーキテクチャ空間から各ブロックを徹底的に検索します。次に、有望なパフォーマンスと多様性を備えた複数のアーキテクチャが選択されます。推論時間を短縮するために、複数の検索されたネットワークからの多様な出力の組み合わせを監督として扱い、学生ネットワークのトレーニングをガイドすることにより、教師と学生のパラダイムを採用しています。さらに、敵対的学習を適用して学生ネットワークを最適化し、教師ネットワークの知識を回復します。私たちの知る限りでは、NASと知識蒸留の組み合わせを調査したのは私たちが初めてです。提案された方法は、公開されているOpenKBPデータセットで評価されており、実験結果は、私たちの方法の有効性と、最先端の方法よりも優れたパフォーマンスを示しています。
Radiation therapy treatment planning is a complex process, as the target dose prescription and normal tissue sparing are conflicting objectives. Automated and accurate dose prediction for radiation therapy planning is in high demand. In this study, we propose a novel learning-based ensemble approach, named LE-NAS, which integrates neural architecture search (NAS) with knowledge distillation for 3D radiotherapy dose prediction. Specifically, the prediction network first exhaustively searches each block from enormous architecture space. Then, multiple architectures are selected with promising performance and diversity. To reduce the inference time, we adopt the teacher-student paradigm by treating the combination of diverse outputs from multiple searched networks as supervisions to guide the student network training. In addition, we apply adversarial learning to optimize the student network to recover the knowledge in teacher networks. To the best of our knowledge, we are the first to investigate the combination of NAS and knowledge distillation. The proposed method has been evaluated on the public OpenKBP dataset, and experimental results demonstrate the effectiveness of our method and its superior performance to the state-of-the-art method.
updated: Sun Oct 10 2021 09:25:42 GMT+0000 (UTC)
published: Sat Jun 12 2021 10:08:52 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アソシエイト