用多头注意力增强Inception V3,提升心脏增大X光片自动检测准确率
Multi Head Attention Enhanced Inception v3 for Cardiomegaly Detection
- 在Inception V3中嵌入多头注意力机制,自动聚焦关键区域
- 在X光数据集上达到95.6%准确率,AUC达96.0
- 适合医学影像分析、心脏病早期筛查场景
医疗行业因新型成像技术而发生显著变革,不仅推动心血管疾病诊断,也实现了心脏增大等结构异常的可视化。本文提出一种结合深度学习与注意力机制的自动化心脏增大检测方法,基于标注的X光图像数据集进行训练。预处理阶段优化图像质量,确保输入数据最优。模型采用Inception V3作为核心架构,并引入多层注意力机制增强特征表达能力。其关键在于多头注意力可自动选择性关注输入图像中的特定区域,从而敏感识别心脏增大的征象。通过注意力评分的计算、复制与应用,强化了关键特征表示,实现高效诊断。评估阶段严格验证,结果显示模型准确率为95.6%,精确率为95.2%,召回率为96.2%,灵敏度为95.7%,特异性为96.1%,曲线下面积(AUC)为96.0,相关结果以图表形式呈现。
原文摘要 · Abstract (English)
The healthcare industry has been revolutionized significantly by novel imaging technologies, not just in the diagnosis of cardiovascular diseases but also by the visualization of structural abnormalities like cardiomegaly. This article explains an integrated approach to the use of deep learning tools and attention mechanisms for automatic detection of cardiomegaly using X-ray images. The initiation of the project is grounded on a strong Data Collection phase and gathering the data of annotated X-ray images of various types. Then, while the Preprocessing module fine-tunes image quality, it is feasible to utilize the best out of the data quality in the proposed system. In our proposed system, the process is a CNN configuration leveraging the inception V3 model as one of the key blocks. Besides, we also employ a multilayer attention mechanism to enhance the strength. The most important feature of the method is the multi-head attention mechanism that can learn features automatically. By exact selective focusing on only some regions of input, the model can thus identify cardiomegaly in a sensitive manner. Attention rating is calculated, duplicated, and applied to enhance representation of main data, and therefore there is a successful diagnosis. The Evaluation stage will be extremely strict and it will thoroughly evaluate the model based on such measures as accuracy and precision. This will validate that the model can identify cardiomegaly and will also show the clinical significance of this method. The model has accuracy of 95.6, precision of 95.2, recall of 96.2, sensitivity of 95.7, specificity of 96.1 and an Area Under Curve(AUC) of 96.0 and their respective graphs are plotted for visualisation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。