Pretraining and Distillation Matter More Than Architecture Family for Label-Free Single-Cell Classification

📅 2026-09-09
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
研究通过控制实验比较了不同深度学习架构在无标签单细胞分类中的表现,发现预训练和知识蒸馏比架构选择更重要。
📝 Abstract
Choosing a deep learning architecture for label-free single-cell classification remains an open question, with microscopy benchmarks reporting conflicting conclusions about CNNs versus transformers. We present a controlled benchmark on LIVECell phase-contrast microscopy data using source-image-disjoint train/validation/test splits to prevent parent-image leakage and matched optimisation, augmentation, and evaluation protocols across EfficientNet, Vision Transformer (ViT), and EVA-02 models. This allows the effects of architecture, pretraining, fine-tuning, tokenisation, and distillation to be disentangled. We find that the previously reported CNN advantage is largely explained by pretraining rather than architecture: the smallest pretrained model outperforms the strongest model trained from scratch despite far fewer parameters. Pretraining improves macro-F1 by 3-4 points, while the gap between the best pretrained CNN and transformer is below 0.5 points. Architectural choices nevertheless matter: ViT-S/8 outperforms ViT-S/16 and matches the four-times-larger ViT-B/16 at a quarter of the parameters, showing that finer tokenisation benefits small cell crops. Conversely, layer-wise learning-rate decay, central to the EVA-02 fine-tuning recipe, degrades performance, highlighting that transfer heuristics from natural-image recognition may not generalise to microscopy. Finally, knowledge distillation substantially improves the deployment frontier: compact EfficientNet-B0 students distilled from teacher councils outperform every individually trained backbone, including the EfficientNet-B5 and EVA-02 teachers. Overall, our results show that rigorous control of pretraining and evaluation is essential for interpreting biomedical architecture benchmarks, while distillation may be a more effective route to practical single-cell classification than architecture choice alone.
Problem

Research questions and friction points this paper is trying to address.

label-free single-cell classification
deep learning architecture
microscopy benchmarks
Innovation

Methods, ideas, or system contributions that make the work stand out.

pretraining
knowledge distillation
fine-grained tokenisation
🔎 Similar Papers
2024-08-22Neural Information Processing SystemsCitations: 0
P
Philip Graemer
Centre for the Cellular Microenvironment, Department of Biomedical Engineering, University of Strathclyde, Glasgow, UK; School of Molecular Biosciences, University of Glasgow, Glasgow, UK
G
Giuseppe Di Caprio
Centre for the Cellular Microenvironment, Department of Biomedical Engineering, University of Strathclyde, Glasgow, UK; School of Molecular Biosciences, University of Glasgow, Glasgow, UK