Extending 2D foundational DINOv3 representations to 3D segmentation of neonatal brain MR images
Organizations: Institute of Business Administration Karachi
Abstract
Precise volumetric delineation of hippocampal structures is essential for quantifying neurodevelopmental trajectories in pre-term and term infants, where subtle morphological variations may carry prognostic significance. While foundation encoders trained on large-scale visual data offer discriminative representations, their 2D formulation is a limitation with respect to the D organization of brain anatomy. We propose a volumetric segmentation strategy that reconciles this tension through a structured window-based disassembly-reassembly mechanism: the global MRI volume is decomposed into non-overlapping 3D windows or sub-cubes, each processed via a separate decoding arm built upon frozen high-fidelity features, and subsequently reassembled prior to a ground-truth correspendence using a dense-prediction head. This architecture preserves constant a decoder memory footprint while forcing predictions to lie within an anatomically consistent geometry. Evaluated on the ALBERT dataset for hippocampal segmentation, the proposed approach achieves a Dice score of 0.65 for a single 3D window. The method demonstrates that volumetric anatomical structure could be recovered from frozen 2D foundation representations through structured compositional decoding, and offers a principled and generalizable extension for foundation models for 3D medical applications.
Figures & tables
| Experiment | ALBERT (n=20) | ||
|---|---|---|---|
| DSC | IoU | Vol. Error (%) | |
| 8 sub-cubes ( ) | 0.3518 | 0.2148 | 30.42 |
| 1 sub-cube ( ) | 0.6514 | 0.4851 | 14.61 |
| Experiment | ALBERT (n=20) | ||
|---|---|---|---|
| DSC | IoU | Vol. Error (%) | |
| Baseline | 0.6514 | 0.4851 | 14.61 |
| No depth embedding | 0.6528 | 0.4872 | 11.71 |
| Single-scale decoding | 0.3585 | 0.2198 | 31.75 |
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 768 papers