Search papers, labs, and topics across Lattice.
This paper introduces TensorLDM, a component-wise latent diffusion model designed to reconstruct diffusion tensors from sparse diffusion-weighted images (DWIs) while ensuring anatomical consistency. By employing two group-specific encoders and a shared Cross-Component Attention mechanism, TensorLDM effectively captures inter-component dependencies and enhances the accuracy of tensor reconstruction. The model demonstrates superior performance on the Human Connectome Project dataset, achieving a low SPD-violation rate and improved voxel-wise reconstruction accuracy compared to existing methods.
TensorLDM achieves near-ground-truth diffusion tensor reconstruction with a remarkable SPD-violation rate of just 1.54%, setting a new standard for anatomical consistency in DTI.
Reconstructing diffusion tensors from sparse DWIs is critical for accelerating Diffusion Tensor Imaging (DTI) in clinical settings, yet current deep learning approaches frequently yield anatomically inconsistent or physically implausible tensors. We introduce TensorLDM, a component-wise latent diffusion model that processes the six tensor components through two group-specific encoders (for diagonal and off-diagonal elements) while maintaining anatomical consistency via shared DWI conditioning. TensorLDM uses an Anatomy-Conditioned Autoencoder that encourages the latent to focus on tensor properties rather than re-encoding structural information. A shared Cross-Component Attention (CCA) mechanism, applied in both autoencoder refinement and diffusion fine-tuning, models inter-component dependencies, while a Mixture-of-Experts (MoE) DWI conditioner provides component-adaptive conditioning. On the Human Connectome Project (HCP) dataset under a single-shell, four-volume sparse acquisition, TensorLDM produces the most accurate downstream tractography and tensors with near-ground-truth physical validity (SPD-violation rate 1.54% vs. 1.40%), with the best or comparable voxel-wise reconstruction accuracy. Geodesic tensor error measured by the Log-Euclidean Metric (LEM) corroborates these gains.
