arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.01705cs.CVcs.AI
收藏

采用加速计算策略优化面向多类别疾病诊断的X射线图像分类学习

Optimized Learning for X-Ray Image Classification for Multi-Class Disease Diagnoses with Accelerated Computing Strategies

University of Puerto Rico at Mayagüez(波多黎各大学马亚圭斯分校)

Sebastian A. Cruz Romero, Ivanelyz Rivera de Jesus, Dariana J. Troche Quinones, Wilson Rivera Gallego

AI总结:

针对X射线多类别疾病诊断中的误报与漏报问题,提出改进预训练ResNet并结合CUDA、混合精度等加速策略,显著缩短训练与推理时间。

摘要:

基于X射线图像的疾病诊断,关键在于确保识别样本中病症的精确度,而这项任务因假阳性和假阴性的出现而充满挑战。假阳性会带来错误识别不存在病症的风险,导致误诊并降低患者护理质量。相反,假阴性则构成遗漏真实异常的威胁,可能造成治疗和干预延迟,从而给患者带来不良后果。克服这些挑战的紧迫性促使人们持续努力,在计算框架内提升X射线图像分析算法的精确度和可靠性。本研究介绍了为X射线图像多类别疾病诊断定制的改进型预训练ResNet模型,并融入先进优化策略以减少训练和推理任务的执行运行时间。主要目标是通过PyTorch、CUDA、混合精度训练和学习率调度器的加速实现,取得切实的性能提升。虽然结果表明普通训练与CUDA加速训练之间的执行运行时间有显著改善,但各种训练优化模式之间的差异微乎其微。本研究标志着在优化计算方法以减少更大模型训练执行时间方面取得重大进展。此外,我们探索了使用MPI4Py进行有效并行数据处理、将梯度下降优化分布到多个节点的潜力,并利用多进程加速更大数据集的数据预处理。

英文摘要:

X-ray image-based disease diagnosis lies in ensuring the precision of identifying afflictions within the sample, a task fraught with challenges stemming from the occurrence of false positives and false negatives. False positives introduce the risk of erroneously identifying non-existent conditions, leading to misdiagnosis and a decline in patient care quality. Conversely, false negatives pose the threat of overlooking genuine abnormalities, potentially causing delays in treatment and interventions, thereby resulting in adverse patient outcomes. The urgency to overcome these challenges compels ongoing efforts to elevate the precision and reliability of X-ray image analysis algorithms within the computational framework. This study introduces modified pre-trained ResNet models tailored for multi-class disease diagnosis of X-ray images, incorporating advanced optimization strategies to reduce the execution runtime of training and inference tasks. The primary objective is to achieve tangible performance improvements through accelerated implementations of PyTorch, CUDA, Mixed- Precision Training, and Learning Rate Scheduler. While outcomes demonstrate substantial improvements in execution runtimes between normal training and CUDA-accelerated training, negligible differences emerge between various training optimization modalities. This research marks a significant advancement in optimizing computational approaches to reduce training execution time for larger models. Additionally, we explore the potential of effective parallel data processing using MPI4Py for the distribution of gradient descent optimization across multiple nodes and leverage multiprocessing to expedite data preprocessing for larger datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑