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.

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