arXiv:2502.10385cs.CVcs.AI2025-02ICML被引 24

用编码率正则简化DINO训练,提升稳定性和效果

Simplifying DINO via Coding Rate Regularization

  • 在损失函数中加入编码率正则项,替代复杂设计
  • 简化后模型在下游任务上表现更优,且更鲁棒
  • 适合希望快速复现高质量自监督模型的研究者

DINO和DINOv2是广泛用于大规模无标签图像数据表示学习的模型家族,其学习到的表征在图像分类、分割等下游任务中常达到顶尖性能。然而,它们依赖大量经验性设计,训练流程复杂且不稳定,需精细调参以防止表征坍塌,极大增加了改进或适配新领域的难度。本文提出,可通过在损失函数中显式加入编码率正则项,移除大部分经验性设计,仅保留关键机制。由此得到的简化版本分别称为SimDINO和SimDINOv2。结果表明,这些简化模型对网络架构和超参数变化更具鲁棒性,且在下游任务中学习到的表征质量更高,实现了对原版DINO和DINOv2的帕累托改进。该工作揭示了简化设计原则在改善深度学习实践中的潜力。

原文摘要 · Abstract (English)

DINO and DINOv2 are two model families being widely used to learn representations from unlabeled imagery data at large scales. Their learned representations often enable state-of-the-art performance for downstream tasks, such as image classification and segmentation. However, they employ many empirically motivated design choices and their training pipelines are highly complex and unstable -- many hyperparameters need to be carefully tuned to ensure that the representations do not collapse -- which poses considerable difficulty to improving them or adapting them to new domains. In this work, we posit that we can remove most such-motivated idiosyncrasies in the pre-training pipelines, and only need to add an explicit coding rate term in the loss function to avoid collapse of the representations. As a result, we obtain highly simplified variants of the DINO and DINOv2 which we call SimDINO and SimDINOv2, respectively. Remarkably, these simplified models are more robust to different design choices, such as network architecture and hyperparameters, and they learn even higher-quality representations, measured by performance on downstream tasks, offering a Pareto improvement over the corresponding DINO and DINOv2 models. This work highlights the potential of using simplifying design principles to improve the empirical practice of deep learning.

自监督学习表示学习模型简化DINO

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