OpenMAP-HIE: Pathology Informed Score-based Diffusion Model for Unsupervised Neonatal Hypoxic-Ischemic Encephalopathy Lesion Segmentation

OpenMAP-HIE: Pathology Informed Score-based Diffusion Model for Unsupervised Neonatal Hypoxic-Ischemic Encephalopathy Lesion Segmentation

Pengyu Kan, Craig Jones, Kenichi Oishi

Published in Preprint, 2026

Abstract

Accurate localization of neonatal hypoxic–ischemic encephalopathy (HIE) lesions on apparent diffusion coefficient (ADC) images is clinically important but is limited by scarce expert annotations and substantial variation in lesion appearance. We propose OpenMAP-HIE, a pathology-informed score-based diffusion model for unsupervised HIE lesion localization without voxel-wise lesion supervision. A denoising diffusion model learns the healthy neonatal ADC distribution, while a ControlNet-based conditional branch preserves patient-specific anatomy. At inference, anomaly evidence is derived directly from differences between conditional and unconditioned noise predictions at independently evaluated diffusion noise levels, avoiding dependence on a final pseudo-healthy reconstruction or sequential reverse-diffusion chain. Based on the characteristic ADC reduction associated with acute HIE, only discrepancies consistent with the expected pathological direction are retained and aggregated across noise levels using a geometric mean. Spatial mean filtering further suppresses isolated responses. The framework was evaluated on the BONBID-HIE dataset. The ADC-only method achieved a Dice score of 0.283 ± 0.244, average precision of 0.505, Hausdorff distance of 25.45 ± 15.63 mm, mean average surface distance of 9.40 ± 8.49 mm, and normalized surface distance of 0.355 ± 0.236, providing the best overlap and surface-distance performance among the evaluated unsupervised methods and Dice performance comparable to supervised ADC-only models. Optional population-derived ADC Z-map refinement further increased Dice to 0.602 ± 0.269. These findings support pathology-informed score discrepancies as a useful signal for neonatal HIE lesion localization, while highlighting remaining challenges in detecting small and low-contrast lesions.

Figure 2a. Inference on a normal slice.

Figure 2b. Inference on an HIE lesion slice.

Figure 2. Inference using pathology-informed noise-prediction discrepancy aggregation for (a) a normal slice and (b) a slice containing HIE lesions. Anomaly evidence comes from differences between the healthy-prior and patient-conditioned noise predictions, kept only when they match the expected ADC reduction.