Dino U-Net: Exploiting High-Fidelity Dense Features from Foundation Models for Medical Image Segmentation
2508.20909v1
cs.CV, eess.IV
2025-08-30
Авторы:
Yifan Gao, Haoyue Li, Feng Yuan, Xiaosong Wang, Xin Gao
Резюме на русском
#### Контекст
Сегодняшние фундаментальные модели, обученные на великомасштабных датасетах с обычными изображениями, показали свою эффективность в задачах медицинской изображейной сегментации. Однако перенос таких моделей в клинические применения сталкивается с рядом проблем. Это включает в себя недостаток точности при работе с низкоуровневыми деталями изображений, что может отрицательно сказаться на качестве диагностики. Недостаток эффективного механизма передачи высокоуровневых семантических признаков в контекст медицинских изображений также является значимой проблемой. Данный исследовательский проект стремится решить эти проблемы, предлагая новую архитектуру, которая может эффективно использовать высококачественные признаки, полученные от предварительно обученных моделей.
#### Метод
Для решения указанных проблем, авторы предложили Dino U-Net, новую архитектуру, основанную на бесплатном DINOv3-модели, которая специально перенастроена для задач медицинской изображейной сегментации. Модель DINOv3, являющаяся мощным общего назначения фундаментальным моделью, была скорректирована для функционирования в составе кодировщика. Для эффективного объединения высокоуровневых семантических признаков с локальными деталями изображений введен адаптер, который отвечает за функциональное слияние этих двух уровней. Чтобы избежать потери качества при проецировании высокоразмерных признаков в пространство декодера, предложен новый модуль — FAPM (Fidelity-Aware Projection Module), который не только сохраняет смысловые признаки, но и улучшает их структуру. Такая конструкция позволяет модели получать улучшенную точность в понимании медицинских изображений.
#### Результаты
На основе предложенной модели проведены тщательные эксперименты на семи различных медицинских изображенческих датасетах. Эти датасеты включали разные виды медицинских изображений, такие как сегментацию тканей, органов и т.д. Результаты показали, что Dino U-Net превосходит ранее предложенные методы в точности сегментации. В частности, модель с 7-миллиардными параметрами демонстрирует существенное улучшение по сравнению с базовой моделью DINOv3. Эта модель также доказала свою скелетопрочность, показав постоянное улучшение качества сегментации при увеличении размера модели.
#### Значимость
Разработанная архитектура имеет широкое клиническое применение, в том числе в работах по сегментации тканей и органов в медицинских изображениях. Основные преимущества в том, что Dino U-Net может эффективно использовать высококачественные признаки, полученные от общего назначения модели DINOv3. Это приводит к не только улучшению точности,
Abstract
Foundation models pre-trained on large-scale natural image datasets offer a
powerful paradigm for medical image segmentation. However, effectively
transferring their learned representations for precise clinical applications
remains a challenge. In this work, we propose Dino U-Net, a novel
encoder-decoder architecture designed to exploit the high-fidelity dense
features of the DINOv3 vision foundation model. Our architecture introduces an
encoder built upon a frozen DINOv3 backbone, which employs a specialized
adapter to fuse the model's rich semantic features with low-level spatial
details. To preserve the quality of these representations during dimensionality
reduction, we design a new fidelity-aware projection module (FAPM) that
effectively refines and projects the features for the decoder. We conducted
extensive experiments on seven diverse public medical image segmentation
datasets. Our results show that Dino U-Net achieves state-of-the-art
performance, consistently outperforming previous methods across various imaging
modalities. Our framework proves to be highly scalable, with segmentation
accuracy consistently improving as the backbone model size increases up to the
7-billion-parameter variant. The findings demonstrate that leveraging the
superior, dense-pretrained features from a general-purpose foundation model
provides a highly effective and parameter-efficient approach to advance the
accuracy of medical image segmentation. The code is available at
https://github.com/yifangao112/DinoUNet.
Ссылки и действия
Дополнительные ресурсы: