JanusNet: Hierarchical Slice-Block Shuffle and Displacement for Semi-Supervised 3D Multi-Organ Segmentation
2508.03997v1
cs.CV
2025-08-09
Авторы:
Zheng Zhang, Tianzhuzi Tan, Guanchun Yin, Bo Zhang, Xiuzhuang Zhou
Резюме на русском
**Резюме**
С WEAKLY SUPERVISED MEDICAL IMAGE SEGMENTATION сталкивается с трудностями из-за недостатка тренировочных данных и аннотаций. Обычно для увеличения диверсии используется дата аугментация, включающая случайное перемешивание трехмерных блоков. Однако этот подход нарушает анатомическую непрерывность в областях с тесной связью, что приводит к сокращению точности, особенно в сложных регионах, например, мелких органах. Мы предлагаем **JanusNet**, новую архитектуру, которая сохраняет анатомическую непрерывность в трехмерных изображениях. Она включает два этапа: **Slice-Block Shuffle** (алгоритм, сохраняющий контекст за пределами разъединенных блоков) и **Confidence-Guided Displacement** (замена блоков на основе достоверности предсказаний). Эти технологии работают вместе для улучшения обучения в сложных областях. На Synapse и AMOS демонстрируется превосходство JanusNet над состоянием технологии, где достигнут прирост DSC на 4% при использовании лишь 20% отмеченных данных. Это решение гарантирует более точное и анатомически согласованное обучение в медицинской сегментации.
Abstract
Limited by the scarcity of training samples and annotations, weakly
supervised medical image segmentation often employs data augmentation to
increase data diversity, while randomly mixing volumetric blocks has
demonstrated strong performance. However, this approach disrupts the inherent
anatomical continuity of 3D medical images along orthogonal axes, leading to
severe structural inconsistencies and insufficient training in challenging
regions, such as small-sized organs, etc. To better comply with and utilize
human anatomical information, we propose JanusNet}, a data augmentation
framework for 3D medical data that globally models anatomical continuity while
locally focusing on hard-to-segment regions. Specifically, our Slice-Block
Shuffle step performs aligned shuffling of same-index slice blocks across
volumes along a random axis, while preserving the anatomical context on planes
perpendicular to the perturbation axis. Concurrently, the Confidence-Guided
Displacement step uses prediction reliability to replace blocks within each
slice, amplifying signals from difficult areas. This dual-stage, axis-aligned
framework is plug-and-play, requiring minimal code changes for most
teacher-student schemes. Extensive experiments on the Synapse and AMOS datasets
demonstrate that JanusNet significantly surpasses state-of-the-art methods,
achieving, for instance, a 4% DSC gain on the Synapse dataset with only 20%
labeled data.
Ссылки и действия
Дополнительные ресурсы: