arXiv:2604.00243cs.CVq-bio.QM2026-04

小模型UCell在细胞分割任务上媲美大模型,无需预训练

UCell: rethinking generalizability and scaling of bio-medical vision models

  • 采用递归结构设计,参数量仅10-30M仍高效
  • 在多个基准上性能媲美10-20倍大的模型
  • 可仅用显微图像从零训练,适合资源有限场景

当前深度学习以规模为驱动,大模型普遍优于小模型。但在生物医学领域,数据量有限且标注成本高,制约了模型扩展。现有研究多聚焦构建超大基础模型,而提升小模型性能的研究被忽视。本文探索参数量仅为10-30M的小模型在单细胞分割任务中的表现。通过在前向计算图中引入递归结构,实现更高效的参数利用。实验表明,UCell在多个基准上性能达到10-20倍于自身规模模型的水平,且对域外数据具有相似泛化能力。更重要的是,UCell仅需显微成像数据即可从零训练,无需依赖自然图像上的大规模预训练,从而摆脱商业数据依赖。此外,通过大量少样本与零样本微调实验,验证了其强适应性。代码已开源。

原文摘要 · Abstract (English)

The modern deep learning field is a scale-centric one. Larger models have been shown to consistently perform better than smaller models of similar architecture. In many sub-domains of biomedical research, however, the model scaling is bottlenecked by the amount of available training data, and the high cost associated with generating and validating additional high quality data. Despite the practical hurdle, the majority of the ongoing research still focuses on building bigger foundation models, whereas the alternative of improving the ability of small models has been under-explored. Here we experiment with building models with 10-30M parameters, tiny by modern standards, to perform the single-cell segmentation task. An important design choice is the incorporation of a recursive structure into the model's forward computation graph, leading to a more parameter-efficient architecture. We found that for the single-cell segmentation, on multiple benchmarks, our small model, UCell, matches the performance of models 10-20 times its size, and with a similar generalizability to unseen out-of-domain data. More importantly, we found that ucell can be trained from scratch using only a set of microscopy imaging data, without relying on massive pretraining on natural images, and therefore decouples the model building from any external commercial interests. Finally, we examined and confirmed the adaptability of ucell by performing a wide range of one-shot and few-shot fine tuning experiments on a diverse set of small datasets. Implementation is available at https://github.com/jiyuuchc/ucell

小模型细胞分割自监督生物医学

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