arXiv:2501.00520cs.CVcs.LG2025-01被引 2

用新数据集和图变压器模型,提升矽肺与肺炎的精准分类。

Innovative Silicosis and Pneumonia Classification: Leveraging Graph Transformer Post-hoc Modeling and Ensemble Techniques

  • 融合图变压器与传统神经网络,捕捉肺部炎症空间关系。
  • 集成模型宏F1达0.9749,各类AUC超0.99。
  • 适合医学影像分析与职业病诊断研究者参考。

本文针对矽肺相关肺部炎症的分类与检测开展全面研究。主要贡献包括:1)构建了名为SVBCX的新标注胸片(CXR)数据集,专用于区分不同致病源引起的肺部炎症,为矽肺与肺炎研究提供宝贵资源;2)提出一种新型深度学习架构,将图变压器网络与传统深度神经网络模块结合,有效识别矽肺与肺炎。同时采用平衡交叉熵(BalCE)作为损失函数,促进各类别间更均衡的学习,增强模型对细微肺部病变的辨别能力。此外,研究探索了融合多种模型架构的集成方法。在自建数据集上的实验结果表明,该方法显著优于基线模型。集成模型取得0.9749的宏F1分数,且每类AUC ROC均超过0.99,验证了其在肺部炎症精准、鲁棒分类中的有效性。

原文摘要 · Abstract (English)

This paper presents a comprehensive study on the classification and detection of Silicosis-related lung inflammation. Our main contributions include 1) the creation of a newly curated chest X-ray (CXR) image dataset named SVBCX that is tailored to the nuances of lung inflammation caused by distinct agents, providing a valuable resource for silicosis and pneumonia research community; and 2) we propose a novel deep-learning architecture that integrates graph transformer networks alongside a traditional deep neural network module for the effective classification of silicosis and pneumonia. Additionally, we employ the Balanced Cross-Entropy (BalCE) as a loss function to ensure more uniform learning across different classes, enhancing the model's ability to discern subtle differences in lung conditions. The proposed model architecture and loss function selection aim to improve the accuracy and reliability of inflammation detection, particularly in the context of Silicosis. Furthermore, our research explores the efficacy of an ensemble approach that combines the strengths of diverse model architectures. Experimental results on the constructed dataset demonstrate promising outcomes, showcasing substantial enhancements compared to baseline models. The ensemble of models achieves a macro-F1 score of 0.9749 and AUC ROC scores exceeding 0.99 for each class, underscoring the effectiveness of our approach in accurate and robust lung inflammation classification.

肺部疾病图像分类图神经网络医疗AI

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