arXiv:2605.03602eess.IV2026-05

Dante让医疗影像分割模型高效适配新医院数据,尤其适合标注少的场景。

Dante: An Open Source Model Pre-Training and Fine-Tuning Tool for the Dafne Federated Framework for Medical Image Segmentation

  • 支持从零训练与参数高效微调,可灵活冻结层并扩展LoRA至多维卷积
  • 在少样本条件下,微调速度比从头训练快63.6%,Dice系数最高达0.957
  • 专为真实临床部署设计,适合医疗影像领域快速迁移与联邦学习应用

将预训练深度学习分割模型适配到新临床领域是医学图像分析中的长期挑战,尤其当目标机构标注数据稀缺时。参数高效微调策略通过有选择地更新部分模型参数,保留已有表征的同时降低小数据集上的过拟合风险。本文提出DAfNe TrainEr(Dante),一个与Dafne联邦分割生态集成的开源训练与微调后端模块。Dante支持从零训练、自动架构配置、可配置的层冻结策略,并将低秩适配(LoRA)扩展至多维卷积层,采用通道级分解。为验证其性能,评估了渐进解冻(GU)和LoRA在涵盖腹部器官与脑白质病变分割的真实跨域MRI迁移场景中,包括全数据与少样本条件。相比从头训练,GU使达到峰值性能85%所需训练轮数减少最多63.6%;在数据充足场景下,LoRA Dice相似性系数最高达0.957。两种策略在所有测试领域均优于基线,且在更丰富预训练数据下提升更显著。结果验证了Dante作为真实临床部署条件下通用医学图像分割微调模块的有效性。Dante代码已开源:https://github.com/dafne-imaging/dafne-torch-trainer,Dafne生态项目见:https://github.com/dafne-imaging。

原文摘要 · Abstract (English)

Adapting pre-trained deep learning segmentation models to new clinical domains is a persistent challenge in medical image analysis, particularly when annotated data at the target site are scarce. Parameter-efficient fine-tuning strategies offer a principled solution by selectively updating a controlled subset of model parameters, preserving previously acquired representations while reducing the risk of overfitting on small datasets. This paper introduces DAfNe TrainEr (Dante), an open-source module integrating with the Dafne federated segmentation ecosystem as a dedicated training and fine-tuning backend. Dante supports training from scratch with automatic architecture configuration, configurable layer freezing schedules, and Low-Rank Adaptation (LoRA) extended to N-dimensional convolutional layers through channel-wise factorization. To validate the module, Gradual Unfreezing (GU) and LoRA are assessed across realistic cross-domain MRI transfer scenarios covering abdominal organ segmentation and brain white matter lesion segmentation, under full-data and few-shot conditions. GU reduced the epochs required to reach 85% of peak performance by up to 63.6% compared to training from scratch, while LoRA achieved Dice Similarity Coefficients up to 0.957 in data-rich scenarios. Both strategies outperformed the baseline across all tested domains, with gains amplified by richer pre-training datasets. These results validate Dante as a domain-agnostic fine-tuning module for medical image segmentation in real clinical deployment conditions. Dante code is available at https://github.com/dafne-imaging/dafne-torch-trainer while Dafne ecosystem project is available at https://github.com/dafne-imaging.

医疗影像微调联邦学习分割

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