Accelerating Real-Time Imaging for Radiotherapy: Leveraging Multi-GPU Training with PyTorch
dc.contributor.author | Obe, Ruth | |
dc.contributor.author | Kaufmann, Brandt | |
dc.contributor.author | Baird, Kaelen | |
dc.contributor.author | Kadel, Sam | |
dc.contributor.author | Soltani, Yasmin | |
dc.contributor.author | Cham, Mostafa | |
dc.contributor.author | Gobbert, Matthias | |
dc.contributor.author | Barajas, Carlos A. | |
dc.contributor.author | Jiang, Zhuoran | |
dc.contributor.author | Sharma, Vijay R. | |
dc.contributor.author | Ren, Lei | |
dc.contributor.author | Peterson, Stephen W. | |
dc.contributor.author | Polf, Jerimy C. | |
dc.date.accessioned | 2023-11-27T19:56:34Z | |
dc.date.available | 2023-11-27T19:56:34Z | |
dc.date.issued | 2023-10-02 | |
dc.description | 2023 Symposium for Undergraduate Research in Data Science, Systems, and Security (REU Symposium 2023); Jacksonville, Florida, USA; December 15-17, 2023 | |
dc.description.abstract | Proton beam therapy is an advanced form of cancer radiotherapy that uses high-energy proton beams to deliver precise and targeted radiation to tumors. This helps to mitigate unnecessary radiation exposure in healthy tissues. Realtime imaging of prompt gamma rays with Compton cameras has been suggested to improve therapy efficacy. However, the camera’s non-zero time resolution leads to incorrect interaction classifications and noisy images that are insufficient for accurately assessing proton delivery in patients. To address the challenges posed by the Compton camera’s image quality, machine learning techniques are employed to classify and refine the generated data. These machine-learning techniques include recurrent and feedforward neural networks. A PyTorch model was designed to improve the data captured by the Compton camera. This decision was driven by PyTorch’s flexibility, powerful capabilities in handling sequential data, and enhanced GPU usage. This accelerates the model’s computations on large-scale radiotherapy data. Through hyperparameter tuning, the validation accuracy of our PyTorch model has been improved from an initial 7% to over 60%. Moreover, the PyTorch Distributed Data Parallelism strategy was used to train the RNN models on multiple GPUs, which significantly reduced the training time with a minor impact on model accuracy. | |
dc.description.sponsorship | This work is supported by the NSF-grant “REU Site: Online Interdisciplinary Big Data Analytics in Science and Engineering” from the National Science Foundation (grant no. OAC–2050943). Co-author Ren acknowledges support from the NIH-grant R01–CA279013. Co-author Cham additionally acknowledges support as HPCF RA. The hardware used in the computational studies is part of the UMBC High Performance Computing Facility (HPCF). The facility is supported by the U.S. National Science Foundation through the MRI program (grant nos. CNS–0821258, CNS–1228778, OAC–1726023, and CNS–1920079) and the SCREMS program (grant no. DMS–0821311), with additional substantial support from the University of Maryland, Baltimore County (UMBC). See hpcf.umbc.edu for more information on HPCF and the projects using its resources. | |
dc.description.uri | https://hpcf-files.umbc.edu/research/papers/BigDataREU2023Team2REUSymposium.pdf | |
dc.format.extent | 8 pages | |
dc.genre | conference papers and proceedings | |
dc.genre | postprints | |
dc.identifier.uri | http://hdl.handle.net/11603/30856 | |
dc.language.iso | en_US | |
dc.relation.isAvailableAt | The University of Maryland, Baltimore County (UMBC) | |
dc.relation.ispartof | UMBC Information Systems Department Collection | |
dc.relation.ispartof | UMBC Faculty Collection | |
dc.relation.ispartof | UMBC Mathematics and Statistics Department | |
dc.relation.ispartof | UMBC Student Collection | |
dc.rights | This item is likely protected under Title 17 of the U.S. Copyright Law. Unless on a Creative Commons license, for uses protected by Copyright Law, contact the copyright holder or the author. | |
dc.subject | UMBC High Performance Computing Facility (HPCF) | |
dc.title | Accelerating Real-Time Imaging for Radiotherapy: Leveraging Multi-GPU Training with PyTorch | |
dc.type | Text | |
dcterms.creator | https://orcid.org/0000-0003-1745-2292 |