PRISM: Differentiable Analysis-by-Synthesis for Fixel Recovery in Diffusion MRI
Organizations: Indiana University, Bloomington, IN, USA
Abstract
Diffusion MRI microstructure fitting is nonconvex and often performed voxelwise, which limits fiber peak recovery in narrow crossings. This work introduces PRISM, a differentiable analysis-by-synthesis framework that fits an explicit multi-compartment forward model end-to-end over spatial patches. The model combines cerebrospinal fluid (CSF), gray matter, up to K white-matter fiber compartments (stick-and-zeppelin), and a restricted compartment, with explicit fiber directions and soft model selection via repulsion and sparsity priors. PRISM supports a fast MSE objective and a Rician negative log-likelihood (NLL) that jointly learns sigma without oracle information. A lightweight nuisance calibration module (smooth bias field and per-measurement scale/offset) is included for robustness and regularized to identity in clean-data tests. On synthetic crossing-fiber data (SNR=30; five methods, 16 crossing angles), PRISM achieves 3.5 degrees best-match angular error with 95% recall, which is 1.9x lower than the best baseline (MSMT-CSD, 6.8 degrees, 83% recall); in NLL mode with learned sigma, error drops to 2.3 degrees with 99% recall, resolving crossings down to 20 degrees. On the DiSCo1 phantom (NLL mode), PRISM improves connectivity correlation over CSD baselines at all four tracking angles (best r=.934 at 25 degrees vs. .920 for MSMT-CSD). Whole-brain HCP fitting (~741k voxels, MSE mode) completes in ~12 min on a single GPU with near-identical results across random seeds.
Figures & tables
| Method | 1-fib | Overall | F1 | Rec. | ||||||||
| PRISM MSE | 0.7 | 7.5 | 9.9 | 11.7 | 8.2 | 1.9 | 1.5 | 1.5 | 1.3 | 3.5 | 96 | 95 |
| PRISM NLL | 1.4 | 5.8 | 4.9 | 3.9 | 3.1 | 1.9 | 1.5 | 1.3 | 1.3 | 2.3 | 99 | 99 |
| CSD | 3.1 | 7.9 | 10.3 | 12.7 | 15.2 | 8.8 | 3.9 | 4.0 | 3.5 | 7.7 | 90 | 82 |
| MSMT-CSD | 3.0 | 7.8 | 10.2 | 12.7 | 15.2 | 5.3 | 3.4 | 3.2 | 3.1 | 6.8 | 91 | 83 |
| ODF-FP | 3.1 | 7.4 | 9.7 | 11.9 | 13.7 | 16.9 | 10.9 | 9.4 | 10.9 | 11.6 | 87 | 86 |
| Rprop | Adam | Adam (.05) | AdamW | NAdam | RMSprop | Adagrad | SGD+m | |
|---|---|---|---|---|---|---|---|---|
| MSE ( ) | 21.2 | 24.4 | 21.5 | 24.4 | 24.5 | 21.5 | 45.7 | 85.3 |
| Iter thr | 22 | 200 | 63 | 200 | 200 | 39 | 200 | 200 |
Diffusion MRI
Diffusion MRI measures water motion in tissue to characterize brain microstructure, white-matter pathways, and disease.
See all 51 papers