arXiv:2503.10426cs.CVcs.LG2025-03被引 1

用胶囊网络提升医疗垃圾分类准确率,效果优于传统模型。

Improving Medical Waste Classification with Hybrid Capsule Networks

  • 将预训练DenseNet与胶囊网络结合,增强空间特征捕捉能力。
  • 混合模型F1分数达0.92,较基线提升0.03。
  • 适合关注医疗废弃物智能识别的从业者与研究者。

医疗废物处置不当会带来严重的环境和公共健康风险,加剧温室气体排放并传播传染病。高效准确的医疗废物分类对缓解这些风险至关重要。本文探索将胶囊网络与预训练DenseNet模型结合,以提升医疗废物分类性能。据我们所知,这是首次将胶囊网络应用于该任务的研究。采用来自多个公开来源的多样化医疗废物图像数据集,评估三种模型配置:(1) 预训练DenseNet作为基线;(2) 冻结层的预训练DenseNet与胶囊网络结合;(3) 解冻层的预训练DenseNet与胶囊网络结合。实验结果表明,引入胶囊网络显著提升了分类性能,F1分数从基线的0.89提升至0.92(解冻层混合模型)。这凸显了胶囊网络在克服传统卷积模型空间局限性、增强分类鲁棒性方面的潜力。尽管胶囊增强模型表现更优,但由于本研究使用更大、更复杂的数据集,与以往研究的直接比较存在困难。先前研究多依赖规模较小、领域特定的数据集,因而准确率更高。而本研究数据集更具代表性,虽提升泛化能力但也带来更高分类挑战,凸显了数据复杂性与模型性能之间的权衡。

原文摘要 · Abstract (English)

The improper disposal and mismanagement of medical waste pose severe environmental and public health risks, contributing to greenhouse gas emissions and the spread of infectious diseases. Efficient and accurate medical waste classification is crucial for mitigating these risks. We explore the integration of capsule networks with a pretrained DenseNet model to improve medical waste classification. To the best of our knowledge, capsule networks have not yet been applied to this task, making this study the first to assess their effectiveness. A diverse dataset of medical waste images collected from multiple public sources, is used to evaluate three model configurations: (1) a pretrained DenseNet model as a baseline, (2) a pretrained DenseNet with frozen layers combined with a capsule network, and (3) a pretrained DenseNet with unfrozen layers combined with a capsule network. Experimental results demonstrate that incorporating capsule networks improves classification performance, with F1 scores increasing from 0.89 (baseline) to 0.92 (hybrid model with unfrozen layers). This highlights the potential of capsule networks to address the spatial limitations of traditional convolutional models and improve classification robustness. While the capsule-enhanced model demonstrated improved classification performance, direct comparisons with prior studies were challenging due to differences in dataset size and diversity. Previous studies relied on smaller, domain-specific datasets, which inherently yielded higher accuracy. In contrast, our study employs a significantly larger and more diverse dataset, leading to better generalization but introducing additional classification challenges. This highlights the trade-off between dataset complexity and model performance.

医疗废物胶囊网络图像分类

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