arXiv:2504.06039cs.CV2025-04中稿 · the 47th Annual In…被引 5

用小模型集成提升胶囊内镜异常检测,兼顾精度与轻量化。

Enhanced Anomaly Detection for Capsule Endoscopy Using Ensemble Learning Strategies

  • 用不同损失函数训练多个小模型,再集成预测结果
  • 在两个数据集上达76.9%以上AUC,参数量远少于现有方法
  • 适合资源受限的医疗设备部署,尤其适用于胶囊内镜

胶囊内镜可捕捉消化道图像以筛查常规内镜难以发现的疾病。由于视频胶囊体积有限,直接嵌入人工智能模型需严格控制模型大小,给异常检测带来挑战。此外,该领域数据稀缺也制约了有效检测的实现。为此,本文提出一种集成学习策略,仅需少量独立训练的神经网络即可完成训练与推理。通过采用来自异常检测领域的多种损失函数分别训练各子模型,实现预测集成。该方法在目前最大的两个公开胶囊内镜数据集Galar和Kvasir-Capsule上验证,分别取得76.86%和76.98%的AUC成绩。相比现有基线方法,本方案在所有模型中显著减少参数量,是将AI技术应用于胶囊内镜的关键一步。

原文摘要 · Abstract (English)

Capsule endoscopy is a method to capture images of the gastrointestinal tract and screen for diseases which might remain hidden if investigated with standard endoscopes. Due to the limited size of a video capsule, embedding AI models directly into the capsule demands careful consideration of the model size and thus complicates anomaly detection in this field. Furthermore, the scarcity of available data in this domain poses an ongoing challenge to achieving effective anomaly detection. Thus, this work introduces an ensemble strategy to address this challenge in anomaly detection tasks in video capsule endoscopies, requiring only a small number of individual neural networks during both the training and inference phases. Ensemble learning combines the predictions of multiple independently trained neural networks. This has shown to be highly effective in enhancing both the accuracy and robustness of machine learning models. However, this comes at the cost of higher memory usage and increased computational effort, which quickly becomes prohibitive in many real-world applications. Instead of applying the same training algorithm to each individual network, we propose using various loss functions, drawn from the anomaly detection field, to train each network. The methods are validated on the two largest publicly available datasets for video capsule endoscopy images, the Galar and the Kvasir-Capsule dataset. We achieve an AUC score of 76.86% on the Kvasir-Capsule and an AUC score of 76.98% on the Galar dataset. Our approach outperforms current baselines with significantly fewer parameters across all models, which is a crucial step towards incorporating artificial intelligence into capsule endoscopies.

异常检测胶囊内镜集成学习轻量化

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