arXiv:2512.02364cs.CVcs.AI2025-12

用X光片识别结核病,轻量模型表现更优

Tackling Tuberculosis: A Comparative Dive into Machine Learning for Tuberculosis Detection

  • 用ResNet-50和SqueezeNet对比检测结核病
  • SqueezeNet准确率89%、召回率80%,远超ResNet-50
  • 轻量模型适合移动端,助力资源匮乏地区筛查

本研究探讨了预训练ResNet-50与通用SqueezeNet模型在胸部X光片上检测结核病(TB)的应用。针对结核病诊断在资源有限地区效率低的问题,研究使用Kaggle提供的4,200张胸片数据集,通过数据分割、增强与缩放进行预处理。评估指标包括准确率、精确率、召回率与混淆矩阵。结果显示,SqueezeNet模型损失32%,准确率达89%,精确率98%,召回率80%,F1分数87%;而ResNet-50模型损失54%,准确率73%,精确率88%,召回率52%,F1分数65%。研究表明,轻量级模型在结核病检测中更具优势,且具备集成至移动设备的潜力,有助于早期发现与治疗。但未来仍需发展更快、更小、更精准的模型以支持全球抗结核行动。

原文摘要 · Abstract (English)

This study explores the application of machine learning models, specifically a pretrained ResNet-50 model and a general SqueezeNet model, in diagnosing tuberculosis (TB) using chest X-ray images. TB, a persistent infectious disease affecting humanity for millennia, poses challenges in diagnosis, especially in resource-limited settings. Traditional methods, such as sputum smear microscopy and culture, are inefficient, prompting the exploration of advanced technologies like deep learning and computer vision. The study utilized a dataset from Kaggle, consisting of 4,200 chest X-rays, to develop and compare the performance of the two machine learning models. Preprocessing involved data splitting, augmentation, and resizing to enhance training efficiency. Evaluation metrics, including accuracy, precision, recall, and confusion matrix, were employed to assess model performance. Results showcase that the SqueezeNet achieved a loss of 32%, accuracy of 89%, precision of 98%, recall of 80%, and an F1 score of 87%. In contrast, the ResNet-50 model exhibited a loss of 54%, accuracy of 73%, precision of 88%, recall of 52%, and an F1 score of 65%. This study emphasizes the potential of machine learning in TB detection and possible implications for early identification and treatment initiation. The possibility of integrating such models into mobile devices expands their utility in areas lacking TB detection resources. However, despite promising results, the need for continued development of faster, smaller, and more accurate TB detection models remains crucial in contributing to the global efforts in combating TB.

结核病检测深度学习医疗影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。