BT-CAP: a subcomponent-aware and anatomically constrained data augmentation framework for multi-modal brain tumor MRI...

Data scarcity and class imbalance remain critical challenges in medical image analysis, particularly for brain tumor MRI segmentation, where subcomponents such as enhancing tumor, non-enhancing tumor, cystic component, and peritumoral edema are underrepresented. Existing augmentation strategies, from classical geometric transforms to GAN-based and diffusion model-based synthesis, either lack subcomponent-level control or require extensive generative model training, limiting their practicality in low-data settings.

We propose the Brain Tumor Compositional Augmentation Pipeline (BT-CAP), a subcomponent-aware and anatomically constrained augmentation framework for multi-modal MRI. BT-CAP decomposes tumor subcomponents and recomposes them through a sequence of targeted operations, including isotropic scaling, B-spline deformation, Powell-optimized spatial rearrangement, inpainting, interface smoothing, and constrained edema deformation, applied consistently across all MRI modalities and segmentation masks, producing label-ready augmented volumes without additional annotation. We evaluated augmentation diversity (SSIM, label distribution, intensity variation, centroid displacement) and anatomical plausibility on 50 BraTS-PEDs 2025 cases, yielding 250 augmented volumes, and assessed segmentation performance on the full 256-case dataset using 3-fold cross-validation.

BT-CAP achieved a mean SSIM of 0.956 ± 0.014 with wide subcomponent volume change ranges and realistic intensity heterogeneity, confirming meaningful structural diversity. By architectural design, all augmented segmentation mask voxels were confined within brain boundaries, and zero overlap between deformed edema and the tumor core was observed across all 250 cases. Segmentation experiments showed mean Dice score improvements of 6%-7% for tumor subcomponents and 2%-3% for tumor core and whole tumor compared to training without compositional augmentation, with a computational cost of approximately 2 min per case on CPU.

BT-CAP establishes a new class of compositional augmentation methods that deliver anatomically structured, label-ready, and scalable data generation without generative model training. The framework is applicable to any multi-class segmentation task where data scarcity and structural fidelity are critical, and is openly available at https://github.com/dramintavallaii/BT-CAP.