arXiv:2601.00832cs.LGcs.CV2026-01被引 1

用深度学习自动识别虾病,准确率超96%。

ShrimpXNet: A Transfer Learning Framework for Shrimp Disease Classification with Augmented Regularization, Adversarial Training, and Explainable AI

  • 结合对抗训练与混合增强,提升模型泛化能力。
  • ConvNeXt-Tiny模型达96.88%准确率,置信区间[0.953,0.971]。
  • 支持可解释性分析,适合水产养殖智能诊断应用。

虾是全球消费最广泛的水产品之一,兼具营养价值和经济价值。虾类养殖在许多地区是重要收入来源,但疾病爆发严重威胁可持续生产。为应对这一挑战,本研究提出一种基于深度学习的自动化虾病分类方法。使用包含1,149张图像、涵盖四个病害类别的数据集,评估了六种预训练模型:ResNet50、EfficientNet、DenseNet201、MobileNet、ConvNeXt-Tiny 和 Xception。图像背景被移除,并通过Keras图像管道进行标准化预处理。采用Fast Gradient Sign Method(FGSM)进行对抗训练以增强鲁棒性;同时引入CutMix和MixUp等先进数据增强策略,缓解过拟合并提升泛化性能。为增强模型可解释性,应用Grad-CAM、Grad-CAM++和XGrad-CAM等后处理解释方法可视化注意力区域。实验结果表明,ConvNeXt-Tiny表现最优,在测试集上达到96.88%的准确率,经过1000次迭代后,99%置信区间为[0.953, 0.971]。

原文摘要 · Abstract (English)

Shrimp is one of the most widely consumed aquatic species globally, valued for both its nutritional content and economic importance. Shrimp farming represents a significant source of income in many regions; however, like other forms of aquaculture, it is severely impacted by disease outbreaks. These diseases pose a major challenge to sustainable shrimp production. To address this issue, automated disease classification methods can offer timely and accurate detection. This research proposes a deep learning-based approach for the automated classification of shrimp diseases. A dataset comprising 1,149 images across four disease classes was utilized. Six pretrained deep learning models, ResNet50, EfficientNet, DenseNet201, MobileNet, ConvNeXt-Tiny, and Xception were deployed and evaluated for performance. The images background was removed, followed by standardized preprocessing through the Keras image pipeline. Fast Gradient Sign Method (FGSM) was used for enhancing the model robustness through adversarial training. While advanced augmentation strategies, including CutMix and MixUp, were implemented to mitigate overfitting and improve generalization. To support interpretability, and to visualize regions of model attention, post-hoc explanation methods such as Grad-CAM, Grad-CAM++, and XGrad-CAM were applied. Exploratory results demonstrated that ConvNeXt-Tiny achieved the highest performance, attaining a 96.88% accuracy on the test dataset. After 1000 iterations, the 99% confidence interval for the model is [0.953,0.971].

虾病识别深度学习可解释AI对抗训练

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