eess.IVNov 17, 2025

BrainNormalizer: Anatomy-Informed Pseudo-Healthy Brain Reconstruction from Tumor MRI via Edge-Guided ControlNet

Authors: Min Gu Kwak, Yeonju Lee, Hairong Wang, Kristin R. Swanson, Jing Li

Organizations: Department of Health and Rehabilitation Sciences, University of Pittsburgh, Pittsburgh, PA, USA · H. Milton Stewart School of Industrial and Systems Engineering, Georgia Institute of Technology, Atlanta, GA,2026 USA · Operations Research and Industrial Engineering, Cockrell School of Engineering, University of Texas at Austin, Austin, TX, USAJul · MOSAIC Center, Cedars-Sinai Medical Center, Los Angeles, CA, USA · Computational Biomedicine and Pathology/Lab Medicine, Department of Neurosurgery, Mayo Clinic Arizona, Phoenix, AZ, USA

Abstract

Brain tumors induce complex structural deformations that obscure the patient' s original neuroanatomy, making it difficult to distinguish tumor-induced changes from inherent anatomical variability. Reconstructing a subject-specific pseudo-healthy brain can provide a critical reference for such analysis, but this task is inherently counterfactual, as paired pre-tumor scans and explicit healthy guidance are unavailable. We propose BrainNormalizer, a diffusion-based framework for subject-specific pseudo-healthy brain MRI reconstruction that enables anatomy-informed reconstruction without requiring paired data or explicit healthy references. The framework learns anatomical priors and edge-based structural conditioning through a two-stage training strategy consisting of inpainting-based diffusion fine-tuning and ControlNet-based edge conditioning. At inference, counterfactual pseudo-healthy reconstruction is achieved through a deliberate misalignment strategy, where tumorous inputs are paired with non-tumorous prompts and mirrored contralateral edge maps. This allows subject-specific anatomical guidance to be constructed from the patient's own anatomy, enabling anatomically consistent pseudo-healthy reconstruction that preserves individual structural characteristics. Experiments on the BraTS2020 dataset demonstrate that BrainNormalizer achieves improved distributional realism, symmetry-based structural consistency, and reduced false positive detection compared to existing methods. These results indicate that the proposed framework provides a principled approach for subject-specific counterfactual reconstruction and supports downstream analysis of tumor-induced deformation.

Explore similar work

CardsList
  1. RARF: Region-Aware Rectified Flows for 3D Brain MRI Inpainting

    Sep 3, 2026Tomas Guija-Valiente, Blanca Rodriguez-Gonzalez, Norberto Malpica +1InpaintingBrain Magnetic Resonance Imaging