基于ConA-FPN的肝脏肿瘤检测算法

ConA-FPN Based Algorithm for Liver Tumor Detection

深度学习方法在病灶检测任务中被广泛应用,但因肝脏肿瘤较小和样本较少的问题,导致无法达到辅助诊断的准确率要求。针对以上问题,提出基于ConA-FPN的肝脏肿瘤检测算法,具体过程为:使用融合ResNet和注意力机制的特征金字塔替换Faster R-CNN中的特征提取网络;使用融合特征解决特征金字塔中的高层模块通道信息损失问题,通过添加CAG注意力机制解决了特征融合带来的特征混叠问题,增强模型对小肿瘤的检测能力;使用迁移学习和数据增强提升模型在小数据集上的检测能力和泛化能力。实验结果表明,ConA-FPN在LITS2017和3D-IRCADB上的平均精度达到87.43%,明显优于主流检测模型。

Deep learning methods are widely used in lesion detection tasks,but the problems of small liver tumors and small samples lead to the inability to achieve the accuracy requirements for assisted diagnosis.To address the above problems,this paper proposes a liver tumor detection algorithm based on ConA-FPN,and the specific process is as follows:first,the feature extraction network in Faster R-CNN is replaced by using a feature pyramid fused with ResNet and attention mechanism;then,the fused features are used to solve the problem of information loss of higher-level module channels in the feature pyramid,the CAG attention mechanism is added to solve the problem that feature fusion brings the feature conflation,and enhances the detection ability of the model for small tumors;finally,transfer learning and data augmentation are used to enhance the detection ability and generalization ability of the model on small data sets.The experimental results show that the average accuracy of ConA-FPN on LITS2017 and 3D-IRCADB reaches 87.43%,which is significantly better than the mainstream detection models.