SearcharxivSearch

arXiv subjects

Gourav Jhanwar

Publications and source records attributed to Gourav Jhanwar.

4 recordsLinked to original sources

Sparse plus low-rank matrix embedding with applications in cancer radiotherapy optimization

Decomposing a matrix into sparse and low-rank components is central to robust principal component analysis and has broad applications in machine learning, signal processing, and computer vision. Classical formulations seek to recover the underlying sparse and low-rank structure. We instead introduce \emph{sparse-plus-low-rank matrix embedding} (SLME), whose goal is to construct a computationally efficient surrogate for a large dense matrix, without requiring its components to be interpretable. Given $A\in\mathbb{R}^{m\times n}$, SLME approximates $A \approx S+HW$, where $S$ is sparse, $H\in\mathbb{R}^{m\times r}$, $W\in\mathbb{R}^{r\times n}$, and $r\ll \min\{m,n\}$. The resulting matrix-vector product can be evaluated as $Sx+H(Wx)$ in $\mathrm{nnz}(S)+r(m+n)$ operations, rather than the $mn$ operations required by $Ax$. Our primary motivation arises from optimization problems in cancer radiotherapy treatment planning, where a large dense \emph{dose-influence matrix} is a major computational bottleneck. We formulate SLME as a bi-objective nonconvex optimization problem that balances approximation error against the computational cost of the downstream tasks. We then develop \emph{R3-Trust}, an efficient trust-region algorithm that approximates the Pareto frontier in a single parameter-free run. Each point on the resulting frontier provides a sparse-plus-low-rank representation with a different balance between accuracy and downstream computational cost. Experiments on clinical radiotherapy matrices show that decompositions obtained from existing recovery-oriented formulations can be suboptimal for the embedding objective. Conversely, experiments on synthetic instances demonstrate that SLME and R3-Trust can also be applied to sparse-plus-low-rank recovery, where they compare favorably with state-of-the-art recovery methods in both reconstruction accuracy and computational time.

math.OC

Compressed radiotherapy treatment planning (CompressRTP): A new paradigm for rapid and high-quality treatment planning optimization

Background: Radiotherapy treatment planning involves solving large-scale optimization problems that are often approximated and solved sub-optimally due to time constraints. Central to these problems is the dose influence matrix which quantifies the radiation dose delivered from each beamlet to each voxel. Our findings demonstrate that this matrix is highly compressible, enabling a compact representation of the optimization problems and allowing them to be solved more efficiently and accurately. Methods: We precompute the primary (S) and scattering (L) dose contributions of the dose influence matrix A separately for photon therapy, expressed as: A = S + L. Our analysis reveals that the singular values of the scattering matrix L exhibit exponential decay, indicating that L is a low-rank matrix. This allows us to compress L into two smaller matrices: L=HW, where r is relatively small (approximately 5 to 10). Since the primary dose matrix S is sparse, this supports the use of the well-established "sparse-plus-low-rank" decomposition technique for the influence matrix A, approximated as: A = S + H * W. We introduce an efficient algorithm for sparse-plus-low-rank matrix decomposition, even without direct access to the scattering matrix. This algorithm is applied to optimize treatment plans for ten lung and ten prostate patients, using both compressed and sparsified versions of matrix A. We then evaluate the dose discrepancy between the optimized and final plans. We also integrate this compression technique with our in-house automated planning system, ECHO, and evaluate the dosimetric quality of the generated plans with and without compression.

physics.med-ph

Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Function

Dose volume histogram (DVH) metrics are widely accepted evaluation criteria in the clinic. However, incorporating these metrics into deep learning dose prediction models is challenging due to their non-convexity and non-differentiability. We propose a novel moment-based loss function for predicting 3D dose distribution for the challenging conventional lung intensity modulated radiation therapy (IMRT) plans. The moment-based loss function is convex and differentiable and can easily incorporate DVH metrics in any deep learning framework without computational overhead. The moments can also be customized to reflect the clinical priorities in 3D dose prediction. For instance, using high-order moments allows better prediction in high-dose areas for serial structures. We used a large dataset of 360 conventional lung patients with 2Gy $\times$ 30 fractions to train the deep learning (DL) model using clinically treated plans. We trained a UNet-like CNN architecture using computed tomography (CT), planning target volume (PTV) and organ-at-risk contours (OAR) as input to infer corresponding voxel-wise 3D dose distribution. We evaluated three different loss functions: (1) Mean Absolute Error (MAE) Loss, (2) MAE + DVH Loss, and (3) the proposed MAE + Moments Loss. The quality of the predictions was compared using different DVH metrics as well as dose-score and DVH-score, recently introduced by the AAPM knowledge-based planning grand challenge. Model with (MAE + Moment) loss function outperformed the model with MAE loss by significantly improving the DVH-score (11%, p$<$0.01) while having similar computational cost. It also outperformed the model trained with (MAE+DVH) by significantly improving the computational cost (48%) and the DVH-score (8%, p$<$0.01). The code, models, docker container, and Google Colab project are available on our DoseRTX GitHub (https://github.com/nadeemlab/DoseRTX).

cs.CV

Deep Learning 3D Dose Prediction for Conventional Lung IMRT Using Consistent/Unbiased Automated Plans

Deep learning (DL) 3D dose prediction has recently gained a lot of attention. However, the variability of plan quality in the training dataset, generated manually by planners with wide range of expertise, can dramatically effect the quality of the final predictions. Moreover, any changes in the clinical criteria requires a new set of manually generated plans by planners to build a new prediction model. In this work, we instead use consistent plans generated by our in-house automated planning system (named ``ECHO'') to train the DL model. ECHO (expedited constrained hierarchical optimization) generates consistent/unbiased plans by solving large-scale constrained optimization problems sequentially. If the clinical criteria changes, a new training data set can be easily generated offline using ECHO, with no or limited human intervention, making the DL-based prediction model easily adaptable to the changes in the clinical practice. We used 120 conventional lung patients (100 for training, 20 for testing) with different beam configurations and trained our DL-model using manually-generated as well as automated ECHO plans. We evaluated different inputs: (1) CT+(PTV/OAR)contours, and (2) CT+contours+beam configurations, and different loss functions: (1) MAE (mean absolute error), and (2) MAE+DVH (dose volume histograms). The quality of the predictions was compared using different DVH metrics as well as dose-score and DVH-score, recently introduced by the AAPM knowledge-based planning grand challenge. The best results were obtained using automated ECHO plans and CT+contours+beam as training inputs and MAE+DVH as loss function.

cs.CV