SingleStrip: learning skull-stripping from a single labeled example
2508.10464v1
cs.CV, cs.LG
2025-08-16
Авторы:
Bella Specktor-Fadida, Malte Hoffmann
Резюме на русском
#### Контекст
Получение точных меток для медицинских изображений, таких как МРТ скелета, является бесценным ресурсом для обучения алгоритмов сегментации. Однако традиционная работа с этими метками требует огромных усилий по ручному пометке данных, что может занять много времени и требовать специализированных знаний. Добавление к этому момент, что тренировочные данные для сетей, основанных на глубоком обучении, часто требуются в больших количествах, чтобы достичь высокой точности. Это делает процесс сам по себе трудоемким и дорогостоящим. В данном контексте, SingleStrip предлагает решение, которое сводит к минимуму требования к ручной работе с метками, позволяя обучать сети судя по одной метке.
#### Метод
SingleStrip объединяет семиуровное обучение с самостоятельным тренированием, чтобы обрабатывать изображения с максимально небольшим количеством вручную отмеченных образов. Авторы предлагают автоматическую сегментацию с помощью генерации меток с помощью доменной хаотизации. Это позволяет создавать метки для обучения, даже когда метки отсутствуют. Далее, вторичная сеть, на основе конволюционного автокодировщика, оценивает качество псевдо-меток, выбирая лучшие для последующего отображения. Таким образом, SingleStrip может быть обучен с очень небольшим объемом меток, чтобы достичь высокого качества распознавания скелета.
#### Результаты
Подробно сравнивая различные стратегии оценки качества, такие как групповая самостоятельная оценка и консистентность с тестовым временем, SingleStrip показывает значительную улучшенную точность сравнения с другими сетями, обученными на небольших масках. Результаты показывают, что сеть SingleStrip может выполнять сегментацию головоломок на новых данных с близким к идеальному качеству, даже при обучении с одной маской. Это позволяет решать проблему отсутствия меток и обеспечивать более широкое применение методик семиуровного обучения в медицине.
#### Значимость
SingleStrip может примениться в различных медицинских импровении, где недостаток засечек медицинских изображений ограничивает развитие алгоритмов сегментации. Также, это решение может помочь в технических областях, где необходимо создание меток для новых видов образов. Это существенно снижает трудозатраты для обучения и обработки данных, что может повысить производительность и облегчить развитие новых изображений в сегментации.
#### Выводы
SingleStrip показывает возможность обучения сетей с судя по одной метке, что может значительно упростить процессы тренировки и работы с медицинскими изображениями. Оно открывает новые пути для развития методик семиуровного обучения в области медицины. В бу
Abstract
Deep learning segmentation relies heavily on labeled data, but manual
labeling is laborious and time-consuming, especially for volumetric images such
as brain magnetic resonance imaging (MRI). While recent domain-randomization
techniques alleviate the dependency on labeled data by synthesizing diverse
training images from label maps, they offer limited anatomical variability when
very few label maps are available. Semi-supervised self-training addresses
label scarcity by iteratively incorporating model predictions into the training
set, enabling networks to learn from unlabeled data. In this work, we combine
domain randomization with self-training to train three-dimensional
skull-stripping networks using as little as a single labeled example. First, we
automatically bin voxel intensities, yielding labels we use to synthesize
images for training an initial skull-stripping model. Second, we train a
convolutional autoencoder (AE) on the labeled example and use its
reconstruction error to assess the quality of brain masks predicted for
unlabeled data. Third, we select the top-ranking pseudo-labels to fine-tune the
network, achieving skull-stripping performance on out-of-distribution data that
approaches models trained with more labeled images. We compare AE-based ranking
to consistency-based ranking under test-time augmentation, finding that the AE
approach yields a stronger correlation with segmentation accuracy. Our results
highlight the potential of combining domain randomization and AE-based quality
control to enable effective semi-supervised segmentation from extremely limited
labeled data. This strategy may ease the labeling burden that slows progress in
studies involving new anatomical structures or emerging imaging techniques.
Ссылки и действия
Дополнительные ресурсы: