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.

Ссылки и действия