SIINR: Structurally Informed Implicit Neural Representations for super-resolution with uncertainty quantification of clinical quality diffusion MRI datasets

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Medical Imaging · Depth: Expert, extended

Summary

SIINR (Structurally Informed Implicit Neural Representations) is a novel framework designed for super-resolution and uncertainty quantification of clinical diffusion Magnetic Resonance Imaging (dMRI) datasets. It addresses the challenge of low out-of-plane resolution in clinical acquisitions by combining a supervised 3D U-net as a prior with a self-supervised implicit neural representation (INR). The U-net is trained on 428 subjects from diverse open-access dMRI datasets, totaling 37,820 volumes, using a 70-20-10 train-validation-test split. The INR component, which uses a four-layer MLP with a 1024 hidden dimension and Fourier feature encoding (n_p 5000, sigma^2 3.5, l_max 8), integrates the high-resolution prior and original low-resolution data, enabling joint spatial and angular modeling, data consistency, and analytic posterior distributions for uncertainty. Validated on test sets and clinical cases, including subjects with multiple sclerosis and brain lesions, SIINR consistently outperforms standard interpolation methods in quantitative error metrics and anatomical fidelity, particularly for super-resolution factors up to 4.

Key takeaway

For Machine Learning Engineers developing dMRI analysis pipelines, SIINR offers a robust solution for enhancing low-resolution clinical data. You should consider integrating SIINR to improve structural detail in thick-slice acquisitions and to provide crucial uncertainty quantification for derived metrics like ADC and FA. This framework allows for more reliable interpretation of neuroimaging results, especially in pathological cases, by flagging uncertain regions.

Key insights

SIINR combines a supervised U-net with a self-supervised INR for dMRI super-resolution and uncertainty quantification.

Principles

Method

SIINR uses a 3D U-net for initial super-resolution, then an INR (MLP with Fourier features) is fit per subject, balancing U-net output and low-resolution data via a combined MSE loss, enabling uncertainty quantification.

In practice

Topics

Code references

Best for: AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.