FiberLM: A Transformer-Based Model for Mouse Brain Diffusion MRI Tractography Guided by Viral Tracer Data

Abstract

Diffusion MRI (dMRI) tractography provides a non-invasive method for mapping whole-brain structural connectivity. However, its application is limited by substantial false-positive and false-negative connections. While deep learning based methods have shown promise in improving tractography, most rely on training data derived from conventional dMRI tractography, therefore inheriting the same limitations. Here, we introduce FiberLM, an attention-based Transformer model for mouse brain tractography. The model was trained using a whole-brain streamline dataset based on viral tracer data from the Allen Mouse Brain Connectivity Atlas (AMBCA), allowing the model to learn the properties of both local and long-range axonal trajectories through self-attention. FiberLM was applied to predict anatomically plausible axonal trajectories from ex vivo high-resolution mouse brain dMRI data. Quantitative evaluations demonstrated that FiberLM significantly reduced false-positive and false-negative connections, improved spatial agreement with tracer-defined pathways, and generated whole-brain connectomes that more closely approximated AMBCA results compared to conventional tractography. These findings suggest FiberLM as a potential tool for accurate reconstruction of mouse brain structural connectomics.