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.

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