Search papers, labs, and topics across Lattice.
This paper introduces RARF, a region-aware rectified flow framework designed for 3D brain MRI inpainting, which enhances automated analysis by reconstructing healthy tissue in pathological areas. By employing a three-dimensional neural network that integrates masked flow-matching and reconstruction-consistency objectives, RARF effectively confines the stochastic interpolation process to the inpainting region while preserving anatomical context from observed voxels. Experimental results demonstrate that RARF achieves competitive reconstruction quality while ensuring anatomical consistency, making it a strong contender in the BraTS Inpainting Challenge 2026.
RARF achieves competitive 3D brain MRI inpainting by intelligently focusing on pathological regions while maintaining anatomical integrity, setting a new standard for medical image reconstruction.
Medical image inpainting has the potential to improve automated brain MRI analysis by reconstructing healthy tissue within pathological regions. We introduce RARF, a task-agnostic region-aware rectified flow framework for masked data generation. We instantiate the framework for 3D brain MRI inpainting as our submission to the BraTS Inpainting Challenge 2026. RARF restricts the stochastic interpolation process to the inpainting region, while the observed voxels remain fixed and provide patient-specific anatomical context. A three-dimensional neural network receives the partially voided image, with Gaussian noise filling the missing region, together with the inpainting mask and the corresponding timestep. The model is trained using masked flow-matching and reconstruction-consistency objectives, combined with mask-aware preprocessing and data augmentation. During inference, the learned velocity field transports the initial noise toward a plausible reconstruction of the missing tissue, which is then combined with the unchanged observed anatomy. Experiments under the BraTS evaluation protocol show that the proposed approach produces competitive reconstructions while maintaining anatomical consistency. Source code is available at: https://github.com/TomasGuija/rarf.
