Mohammed Mudassir Uddin

BlogJanuary 2026paperedge MLfew-shot learning

Pruning for the disease, not for ImageNet

Our first preprint of the year keeps the channels that separate one leaf disease from another, so a few-shot model fits on a Raspberry Pi 4.

On 5 January we posted Meta-Learning Guided Pruning for Few-Shot Plant Pathology on Edge Devices. I am the first author, with Shahnawaz Alam, Mohammed Kaif Pasha, Dr. Tasneem Bano Rehman, Dr. Fahmina Taranum and Afroze Begum.

The setting: a farmer finds an unfamiliar blight, has no lab nearby and no reliable connection. The model has to run on a cheap board and learn a new disease from one to ten photos.

What we changed

Standard pruning keeps a channel because of its weight magnitude, or because the pruned layer still reconstructs its output. Neither asks whether the channel helps tell early blight from late blight, and with a handful of examples per class that is the property worth protecting.

DACIS (disease-aware channel importance scoring) adds Fisher's discriminant ratio across disease classes to a gradient term and an activation-variance term. The PMP pipeline prunes conservatively, meta-learns, then prunes again using the meta-gradients. Pruning once from ImageNet weights would keep the channels ImageNet cared about.

PMP-DACIS: disease-aware channel scoring guides a prune, meta-learn, prune pipeline that shrinks ResNet-18 from 11.2M to 2.5M parameters for real-time diagnosis on a Raspberry Pi 4.
The pipeline. DACIS scores drive both pruning passes; the second pass also uses the gradients collected while meta-learning.

What the paper reports

ResNet-18 goes from 11.2M to 2.5M parameters and runs at 142 ms per image on a Raspberry Pi 4, 3.6 times faster than the full model. At 30% of the parameters it is 3.8 points ahead of Meta-Prune in 5-way 1-shot and 5-shot accuracy.

Where it stops

More detail, tables and the device measurements are on the project page.