Spatial Masked-Set Learning for Sparse Multi-Shell Diffusion MRI Signal Synthesis

Abstract

Dense multi-shell diffusion MRI provides rich q-space information but requires long acquisition times. We propose a spatial masked-set framework for sparse multi-shell diffusion MRI signal synthesis. The model treats observed measurements as an unordered set, uses a local $3 \times 3 \times 3$ neighborhood for spatial context, and predicts radial-order-6 SHORE coefficients for the center voxel. The coefficients can then be decoded analytically to synthesize signals at arbitrary q-space locations. Training combines shell-wise gradient dropping, dense signal supervision, and rotation-consistent SHORE targets so that sparse input signals remain aligned with their coefficient supervision under augmentation. We evaluate on held-out HCP100 white-matter voxels by retaining limited subsets of measured diffusion-weighted signals from the reference acquisition. The proposed method achieves lower signal NMSE than both analytical q-space models and a state-of-the-art continuous dMRI signal synthesis model designed for arbitrary input and output q-space sampling. In the $b=1000$ setting with 10 input gradients, it achieves $2.70\%$ NMSE, a $22.4\%$ relative reduction over this continuous model. Fractional anisotropy on reconstructed $b=1000$ signals provides a complementary tensor-derived endpoint, with analytical models remaining competitive for FA despite higher dense-signal NMSE across the evaluated q-space. The implementation is available on \href{https://github.com/xmindflow/SHOREPred}{https://github.com/xmindflow/SHOREPred}.