用可变帧数模型提升超声视频乳腺结节分类准确率
Variable-frame CNNLSTM for Breast Nodule Classification using Ultrasound Videos
- 将NLP的长短句处理思路引入视频分类,支持不同长度视频输入
- 相比关键帧方法,F1分数提升3-6%,特异性提高1.5%
- 适合处理临床中帧数不一的医学视频数据
医学影像与人工智能的结合已成为智能医疗的重要方向,尤其在深度学习用于临床诊断的医学图像分析方面。现有关键帧分类方法未能提取时序特征,而基于三维卷积的超声视频分类要求患者视频帧数一致,导致特征提取效率低、分类性能差。本研究提出一种基于CNN和LSTM的新型视频分类方法,首次将自然语言处理中的长短期句子处理机制引入视频分类。该方法将CNN提取的图像特征压缩至1×512维,按患者视频帧数排序并填充0值形成可变批次,无效填充部分在LSTM训练前被压缩,以节省计算资源。实验结果表明,所提可变帧数CNNLSTM方法在各项指标上均优于其他方法,相较关键帧方法F1分数提升3-6%,特异性提高1.5%;在准确率和精确率上也优于等帧数CNNLSTM。结果验证了该方法在分类可变帧数超声视频中的有效性,为其他医学影像模态应用提供了可能。
原文摘要 · Abstract (English)
The intersection of medical imaging and artificial intelligence has become an important research direction in intelligent medical treatment, particularly in the analysis of medical images using deep learning for clinical diagnosis. Despite the advances, existing keyframe classification methods lack extraction of time series features, while ultrasonic video classification based on three-dimensional convolution requires uniform frame numbers across patients, resulting in poor feature extraction efficiency and model classification performance. This study proposes a novel video classification method based on CNN and LSTM, introducing NLP's long and short sentence processing scheme into video classification for the first time. The method reduces CNN-extracted image features to 1x512 dimension, followed by sorting and compressing feature vectors for LSTM training. Specifically, feature vectors are sorted by patient video frame numbers and populated with padding value 0 to form variable batches, with invalid padding values compressed before LSTM training to conserve computing resources. Experimental results demonstrate that our variable-frame CNNLSTM method outperforms other approaches across all metrics, showing improvements of 3-6% in F1 score and 1.5% in specificity compared to keyframe methods. The variable-frame CNNLSTM also achieves better accuracy and precision than equal-frame CNNLSTM. These findings validate the effectiveness of our approach in classifying variable-frame ultrasound videos and suggest potential applications in other medical imaging modalities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。