Yapay Zeka Hızlı Bakış & Çıkarımlar
AI InsightGradCAM++ kullanılarak gerçekleştirilen yorumlanabilirlik analizi, her iki ağın da dikkatini meyvelerin ilgili bölgelerine yoğunlaştırdığını ortaya koymuştur.
📌 3 Maddede Bu Makale
Özet
Farklı çilek türlerinin sınıflandırılması, tarım ve kalite kontrolünde kritik bir problem olmakla birlikte, büyük ölçekli etiketli veri setlerine duyulan ihtiyaç nedeniyle geleneksel derin öğrenme çözümleri için önemli bir engel oluşturmaktadır. Bu çalışmada, yalnızca 103 etiketli eğitim görüntüsü içeren Mendeley Strawberry Grading Dataset kullanılarak dört farklı ticari çilek çeşidini (A, B, C ve Jumbo) sınıflandırmak amacıyla Prototipik Ağlara dayalı bir Az Örnekli Öğrenme (FSL) modeli uygulanmıştır. Isınma ön-eğitimi, diferansiyel öğrenme oranı stratejisi ve 4-yollu 5-örnek episodik eğitimin bir araya getirilmesiyle özgün bir eğitim boru hattı oluşturulmuş; ResNet50 ve EfficientNet-B0 omurgaları 5-katlı tabakalı çapraz doğrulama protokolü altında karşılaştırmalı olarak değerlendirilmiştir. ResNet50 sırasıyla %99.00 test doğruluğu, %99.11 makro F1 skoru ve %98.62 Cohen Kappa değeri elde ederken, EfficientNet-B0 %98.00, %98.33 ve %97.30 değerlerine ulaşmıştır. Her iki model de %99.80 AUC değeri üretmiştir. GradCAM++ kullanılarak gerçekleştirilen yorumlanabilirlik analizi, her iki ağın da dikkatini meyvelerin ilgili bölgelerine yoğunlaştırdığını ortaya koymuştur. Bu çalışma, metrik tabanlı az örnekli öğrenmenin, tarımsal görüntülere ait veri setlerinin yetersiz olduğu durumlarda geleneksel derin öğrenme yöntemlerine başarılı bir alternatif olarak kullanılabileceğini kanıtlamıştır.
Abstract
Classification of different types of strawberries is an essential problem in agriculture and quality control; however, it presents a challenge to traditional deep learning solutions because of the need for large-scale labeled datasets. In this research, a few-shot learning (FSL) model based on Prototypical Networks was applied to classify four different commercial types of strawberries (A, B, C, and Jumbo), utilizing the Mendeley Strawberry Grading Dataset that includes only 103 labeled images for training. A unique pipeline for training an FSL model was created through a combination of warm-up pre-training, differential learning rate strategy, and 4-way 5-shot episodic training; two backbones (ResNet50 and EfficientNet-B0) were compared under a 5-fold stratified cross-validation. While ResNet50 obtained test accuracy, macro F1-score, and Cohen's Kappa equal to 99.00%, 99.11%, and 98.62% respectively, EfficientNet-B0 achieved 98.00%, 98.33%, and 97.30%. Both models produced an AUC value of 99.80%. Interpretability analysis utilizing GradCAM++ showed that both networks focused on relevant areas of the fruits. This research proved that metric-based few-shot learning can be used successfully as an alternative solution to traditional deep learning methods in case of scarce datasets of agricultural images.