Brain3D: Brain Report Automation via Inflated Vision Transformers in 3D
Organizations: University of Naples Federico II, Department of Electrical Engineering and Information Technology, Italy · Northwestern University, Dept. of Computer Science, McCormick School of Engineering and Applied Science, United States · Pegaso University, Department of Information Science and Technology, Italy
Abstract
Current medical vision-language models (VLMs) process volumetric brain MRI using 2D slice-based approximations, fragmenting the spatial context required for accurate neuroradiological interpretation. We developed Brain3D, a staged vision-language framework for automated radiology report generation from 3D brain tumor MRI. Our approach inflates a pretrained 2D medical encoder into a native 3D architecture and progressively aligns it with a causal language model through three stages: contrastive grounding, supervised projector warmup, and LoRA-based linguistic specialization. Unlike generalist 3D medical VLMs, Brain3D is tailored to neuroradiology, where hemispheric laterality, tumor infiltration patterns, and anatomical localization are critical. Evaluated on 468 subjects (BraTS pathological cases plus healthy controls), our model achieves a Clinical Pathology F1 of 0.951 versus 0.413 for a strong 2D baseline while maintaining perfect specificity on healthy scans. The staged alignment proves essential: contrastive grounding establishes visual-textual correspondence, projector warmup stabilizes conditioning, and LoRA adaptation shifts output from verbose captions to structured clinical reports\footnote{Our code is publicly available for transparency and reproducibility
Figures & tables
| Model | Natural Language Generation (NLG) | Semantics | Clinical Efficacy (F1) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| B-1 | B-4 | R-1 | R-2 | R-L | MTR | BERT | CIDEr | Lat | Anat | Path | |
| State-of-the-Art Baselines | |||||||||||
| Med3DVLM [ 6 ] | 0.051 | 0.005 | 0.108 | 0.018 | 0.083 | 0.055 | 0.836 | 0.007 | 0.300 | 0.225 | 0.119 |
| (3D Generalist) | [0.04, 0.06] | [0.00, 0.01] | [0.09, 0.13] | [0.01, 0.02] | [0.07, 0.10] | [0.05, 0.06] | [0.83, 0.84] | [0.00, 0.00] | [0.22, 0.40] | [0.15, 0.30] | [0.08, 0.18] |
| MedGemma 1.5 [ 3 ] | 0.245 | 0.024 | 0.326 | 0.077 | 0.189 | 0.190 | 0.859 | 0.029 | 0.526 | 0.461 | 0.413 |
| (2D Slice-based) | [0.22, 0.27] | [0.02, 0.03] | [0.30, 0.35] | [0.07, 0.09] | [0.17, 0.20] | [0.18, 0.20] | [0.86, 0.86] | [0.02, 0.03] | [0.45, 0.60] | [0.39, 0.53] | [0.36, 0.46] |
| Qualitative Comparison of Generated Reports | |||
|---|---|---|---|
| Ground Truth : The lesion area is in the frontal and parietal regions of the left hemisphere with mixed signal intensities of varying levels and speckled high-intensity signal areas. Edema is primarily concentrated in the peripheral regions of the left frontal and parietal lobes, indicating moderate to severe swelling of the tissues surrounding the lesions. Necrosis is observed in some lesion areas displaying low signal intensity interspersed within regions of high and isointense signals, showing a certain mottled appearance. Ventricular compression is mild deformation of the ventricles, suggesting pressure effects of the lesions on these regions. | |||
| Brain3D (Ours) : The lesion area is in the left parietal and frontal lobes with mixed high-signal areas. Edema signals are mainly observed around these lesions, indicating significant edema presence affecting parts of both frontal and temporal regions as well as some portions within the parietal lobe. Necrosis may be present at low signal intensity or scattered throughout certain sections of the brain tissue affected by edema but shows no clear concentration. Ventricular compression effects on adjacent ventricles can occur due to pressure from surrounding tissues near the ventricular system. | |||
| MedGemma Baseline : There is a large, irregular, mixed signal intensity mass centered in the right frontal lobe , extending into the right temporal lobe and involving the right lateral ventricle . There is significant mass effect on the right lateral ventricle , causing mild to moderate effacement of the adjacent sulci. | |||
| Med3D-VLM : No intracranial hemorrhage. No evidence of acute infarction. No mass effect or midline shift. Normal grey-white matter differentiation . | |||