A machine learning toolkit assisted approach for IMRT fluence map optimization: feasibility and advantages.

Abstract

Purpose. Traditional machine learning (ML) and deep learning (DL) applications in treatment planning rely on complex model architectures and large, high-quality training datasets. However, they cannot fully replace the conventional optimization process. This study presents a novel application of ML in treatment planning where established ML/DL toolkits are directly applied to treatment plan optimization.Materials and Methods. A one-layer network was designed based on the dose deposition matrix and implemented in PyTorch's L-BFGS optimizer with GPU acceleration. The classical steepest descent optimizer was selected as a reference for comparison. Both optimizers utilized identical inputs and objective functions to ensure a fair comparison. DVH- and gEUD-based objectives were implemented in standard quadratic forms. Standard uniform and 1,000 random initializations were used to test optimizer's search ability under different starting conditions for prostate and head-and-neck cases.Results. The MLT-assisted framework demonstrated comparable or superior plan quality to classical optimization by achieving lower objective values, improved DVHs and capturing finer modulation details in fluence maps. For gEUD-based optimization, it effectively explored beam weight elevations that classical optimization could only reach with stricter convergence criteria and many more iterations. The quality differences primarily stemmed from convergence speed. The MLT-assisted framework required significantly fewer evaluations and iterations to achieve similar or better results. Optimization on random initial maps further demonstrated that it was more robust and less likely to be trapped. It does not require stricter convergence criteria or extended runs to reach high-quality optima, making it more efficient and reliable.Conclusion. This framework leverages ML toolkits in a novel way, enabling faster convergence, greater robustness and handling of complex constraints. As the first study of its kind, it establishes MLT-assisted optimization as a viable and effective alternative to classical methods.